Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
1,737 Commits 18 Branches 0 Tags
4e841b9b16aa60a39cf7c11be58f55fa2fdc34f2
Commit Graph
1737 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocherandpre-commit 34b859a41e Keras CI fix (#5486) 
* Keras CI fix

* pre-commit fixes

* Update ci-testing.yml

Co-authored-by: pre-commit <pre-commit@example.com>
 2021-11-03 19:25:44 +01:00
Glenn Jocher 06bf8ef7e5 Add tf.py verification printout (#5484) 
* Update tf.py with verified confirmation

* Update ci-testing.yml

* Update ci-testing.yml
 2021-11-03 19:04:51 +01:00
Glenn Jocher 62d77a1027 Created using Colaboratory 2021-11-03 18:55:25 +01:00
Glenn Jocher 84a8099b75 Update torch_utils.py import LOGGER (#5483) 2021-11-03 17:17:38 +01:00
Glenn Jocher df30426c03 Improve GPU name (#5478) 
* Improve GPU name

* Update torch_utils.py

* Update torch_utils.py

* Update torch_utils.py

* Update torch_utils.py
 2021-11-03 16:59:53 +01:00
Glenn Jocher 0eb37ad8af Remove check_requirements(('tensorflow>=2.4.1',)) (#5476) 
`check_requirements()` is unreliable for large packages like torch and tensorflow that may have multiple installation routes (i.e. conda, pip, tensorflow-cpu, etc.)
 2021-11-03 14:51:37 +01:00
Glenn Jocher 042f02ff9b Fix tf.py LoadImages() dataloader return values (#5455) 2021-11-02 23:04:15 +01:00
Mrinal JainandGlenn Jocher 7476012a4d Update check_git_status() to run under ROOT working directory (#5441) 
* Updating check_git_status() to switch to the repository root before performing git ops

* More pythonic

* Linting

* Remove redundant Path() call

* Generalizing the context manager

* Fixing error in context manager

* Cleanup

* Change to decorator

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-11-02 22:16:01 +01:00
Pranath 19c8760caa Fixed a small typo in CONTRIBUTING.md (#5445) 
possibel -> possible
 2021-11-02 12:05:31 +01:00
Glenn Jocher 4c0982a243 Update README.md (#5438) 
2-line update
 2021-11-01 18:28:14 +01:00
Glenn Jocherandpre-commit 7b1f7aec46 Update get_loggers() (#4854) 
* Update `set_logging()`

* Update export.py

* pre-commit fixes

* Update LoadImages

* Update LoadStreams

* Update print_args

* Single LOGGER definition

* yolo.py fix

Co-authored-by: pre-commit <pre-commit@example.com>
 2021-11-01 18:22:13 +01:00
Glenn Jocher 8c326a1edf Meshgrid indexing='ij' for PyTorch 1.10 (#5309) 
* Meshgrid `indexing='ij'` for PyTorch 1.10

Will not merge currently as breaks backwards compatibility.

* Meshgrid `indexing='ij'` for PyTorch 1.10

Will not merge currently as breaks backwards compatibility.

* Add check_version hard argument

* Update comment
 2021-11-01 14:33:08 +01:00
Glenn Jocher 5d4258fac5 Fix MixConv2d() remove shortcut + apply depthwise (#5410) 2021-10-30 13:38:51 +02:00
Glenn Jocherandpre-commit 7f9bbf0268 Update GitHub issues templates (#5404) 
* Update GitHub issues templates

* pre-commit fixes

Co-authored-by: pre-commit <pre-commit@example.com>
 2021-10-29 23:16:04 +02:00
Ayush Chaurasia 620b535f85 Update sweep.py (#5402) 2021-10-29 19:21:59 +02:00
Jirka BorovecandGlenn Jocher ed887b5976 Add pre-commit CI actions (#4982) 
* define pre-commit

* add CI code

* configure

* apply pre-commit

* fstring

* apply MD

* pre-commit

* Update torch_utils.py

* Update print strings

* notes

* Cleanup code-format.yml

* Update setup.cfg

* Update .pre-commit-config.yaml

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-28 18:35:01 +02:00
Glenn Jocher a4fece8c14 DDP nl fix (#5332) 2021-10-25 16:03:39 +02:00
Glenn Jocher 9c31a66f5d Update AutoShape.forward() model.classes example (#5324) 2021-10-25 14:03:25 +02:00
Glenn Jocher ca19df5f7f Add autobatch feature for best batch-size estimation (#5092) 
* Autobatch

* fix mem

* fix mem2

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update train.py

* print result

* Cleanup print result

* swap fix in call

* to 64

* use total

* fix

* fix

* fix

* fix

* fix

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Cleanup printing

* Update final printout

* Update autobatch.py

* Update autobatch.py

* Update autobatch.py
 2021-10-25 13:56:13 +02:00
Ayush Chaurasia 692be757b6 W&B: Media panel fix (#5317) 2021-10-25 12:06:12 +02:00
Zhiqiang Wang 3d897986c7 Small fixes to docstrings (#5313) 
* Minor fixes of the git checkout new branch

* Use em dash to quote

* Revert the change of git checkout

* Maybe we should up-to-date with the upstream/master?
 2021-10-24 22:05:34 +02:00
Cristi FatiandGlenn Jocher fee83c1634 Weights download script minor improvements (#5213) 
* Add nano model to the download list

* Add possibility to also download the "*6" models

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-23 14:20:26 +02:00
Jebastin Nadar e0c3f42de7 Uncomment OpenCV 4.5.4 requirement in detect.py (#5305) 2021-10-23 13:40:34 +02:00
Glenn Jocher 79d8f1f678 Created using Colaboratory 2021-10-22 21:19:23 +02:00
Glenn Jocher b760acec11 Created using Colaboratory 2021-10-22 21:11:29 +02:00
Glenn Jocher 441b47c443 More informative EarlyStopping() message (#5303) 2021-10-22 20:02:19 +02:00
jdfrandGlenn Jocher 15e8c4c15b take EXIF orientation tags into account when fixing corrupt images (#5270) 
* take EXIF orientation tags into account when fixing corrupt images

* fit 120 char

* sort imports

* Update local exif_transpose comment

We have a local inplace version that is faster than the official as the image is not copied. AutoShape() uses this for Hub models, but here it is not important as the datasets.py usage is infrequent (AutoShape() it is applied every image).

* Update datasets.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-20 21:35:47 +02:00
jdfrandGlenn Jocher db3bbdde0e autosplit: take image files with uppercase extensions into account (#5269) 
* take image files with uppercase extensions into account in autosplit

* case fix

* Refactor implementation

Removes additional variable (capital variable names are also only for global variables), and uses the same methodology as implemented earlier in datasets.py L409.

* Remove redundant rglob characters

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-20 19:25:33 +02:00
Glenn Jocher a18b0c36cd Update rebase.yml with workflows permissions (#5255) 
Update token with workflows permissions
 2021-10-19 12:02:15 +02:00
dependabot[bot] fd3e2ad8ef Bump actions/stale from 3 to 4 (#5253) 
Bumps [actions/stale](https://github.com/actions/stale) from 3 to 4.
- [Release notes](https://github.com/actions/stale/releases)
- [Changelog](https://github.com/actions/stale/blob/main/CHANGELOG.md)
- [Commits](https://github.com/actions/stale/compare/v3...v4)

---
updated-dependencies:
- dependency-name: actions/stale
  dependency-type: direct:production
  update-type: version-update:semver-major
...

Signed-off-by: dependabot[bot] <support@github.com>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2021-10-19 11:37:54 +02:00
dependabot[bot] aa26504137 Bump actions/cache from 1 to 2.1.6 (#5252) 
Bumps [actions/cache](https://github.com/actions/cache) from 1 to 2.1.6.
- [Release notes](https://github.com/actions/cache/releases)
- [Commits](https://github.com/actions/cache/compare/v1...v2.1.6)

---
updated-dependencies:
- dependency-name: actions/cache
  dependency-type: direct:production
  update-type: version-update:semver-major
...

Signed-off-by: dependabot[bot] <support@github.com>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2021-10-19 11:31:24 +02:00
dependabot[bot] fb12faafa3 Bump cirrus-actions/rebase from 1.4 to 1.5 (#5251) 
Bumps [cirrus-actions/rebase](https://github.com/cirrus-actions/rebase) from 1.4 to 1.5.
- [Release notes](https://github.com/cirrus-actions/rebase/releases)
- [Commits](https://github.com/cirrus-actions/rebase/compare/1.4...1.5)

---
updated-dependencies:
- dependency-name: cirrus-actions/rebase
  dependency-type: direct:production
  update-type: version-update:semver-minor
...

Signed-off-by: dependabot[bot] <support@github.com>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2021-10-19 11:28:52 +02:00
Zhiqiang Wang 8e87396705 Add dependabot for GH actions (#5250) 2021-10-19 11:23:30 +02:00
Glenn Jocher 4f5e274eee Update rebase.yml (#5245) 2021-10-19 09:46:05 +02:00
Glenn Jocher 752d97df2c Add on_fit_epoch_end callback (#5232) 
* Add `on_fit_epoch_end` callback

* Add results to train

* Update __init__.py
 2021-10-18 14:28:22 +02:00
Glenn Jocher 13f7275555 Update/inplace ops (#5233) 
* Clip Objects365 autodownload labels (#5214)

Fixes out of bounds labels that seem to affect ~10% of images in dataset.

* Inplace ops
 2021-10-18 14:24:48 +02:00
Glenn Jocher 0000334a6f Update Objects365.yaml val count (#5212) 2021-10-15 12:40:40 -07:00
Glenn Jocher 991c654e81 Autofix duplicate label handling (#5210) 
* Autofix duplicate labels

PR changes duplicate label handling from report error and ignore image-label pair to report warning and autofix image-label pair. 

This should fix this common issue for users and allow everyone to get started and get a model trained faster and easier than before.

* sign fix

* Cleanup

* Increment cache version

* all to any fix
 2021-10-15 12:32:08 -07:00
Farley LaiandGlenn Jocher fc3606420d Update Objects365.yaml to include the official validation set (#5194) 
* Update Objects365.yaml

Download the official Objects365 validation set and convert the labels

* Enforce 4-space indent, reformat and cleanup

* shorten list comprehension

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-14 20:48:47 -07:00
Glenn Jocher 1c9f710b46 Add hyp.scratch-med.yaml (#5196) 
* Add hyp.scratch-med.yaml

* Update hyp.scratch-med.yaml
 2021-10-14 20:40:30 -07:00
Glenn Jocher 6b31cfd793 Highlight contributors in README (#5173) 
* Highlight contributors in README

* Adjust Contact section

* Update README.md
 2021-10-14 12:49:16 -07:00
Glenn Jocher 7ee5aed0b3 Improved check_suffix() robustness to '' and "" (#5192) 
* Improved check_suffix() robustness to `''` and `""`

* Cleanup
 2021-10-14 12:00:39 -07:00
NauchtanRoboticsandGlenn Jocher 0be58f16f0 Adjust legend labels for classes without instances (#5174) 
* legend-labels Adjust legend labels for classes without instances.

* #5158 Re-indexed series names: only classes containing data.

* #5158 Re-indexed series names: only classes containing data.

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-14 11:57:00 -07:00
Ayush ChaurasiaandGlenn Jocher 1172ac15fb W&B: DDP fix (#5176) 
* fix dpp with wandb disabled

* PyCharm reformat

* Fix

* fix

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-13 20:11:04 -07:00
Glenn Jocher e520e820f1 Update autodownload fallbacks to v6.0 assets (#5177) 2021-10-13 19:58:38 -07:00
Ayush ChaurasiaandGlenn Jocher d0bfeb3980 W&B: fix DDP with wandb disabled (#5163) 
* fix dpp with wandb disabled

* PyCharm reformat

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-13 16:17:15 -07:00
Glenn Jocher a346926996 Add class filtering to LoadImagesAndLabels() dataloader (#5172) 
* Add train class filter feature to datasets.py

Allows for training on a subset of total classes if `include_class` list is defined on datasets.py L448:
```python
        include_class = []  # filter labels to include only these classes (optional)
```

* segments fix
 2021-10-13 15:48:54 -07:00
Glenn Jocher b754525e99 Check 'onnxruntime-gpu' if torch.has_cuda (#5087) 
* Check `'onnxruntime-gpu' if torch.has_cuda`

* fix indent
 2021-10-12 22:25:38 -07:00
Glenn Jocher 20a809dee5 Update stale.yml (#5156) 2021-10-12 17:54:56 -07:00
Glenn Jocher 26784afd99 Use YOLOv5n for CI testing (#5154) 2021-10-12 08:42:26 -07:00