84a8099b75
Update torch_utils.py import `LOGGER` ( #5483 )
2021-11-03 17:17:38 +01:00
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
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
042f02ff9b
Fix tf.py `LoadImages()` dataloader return values ( #5455 )
2021-11-02 23:04:15 +01:00
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
19c8760caa
Fixed a small typo in CONTRIBUTING.md ( #5445 )
...
possibel -> possible
2021-11-02 12:05:31 +01:00
4c0982a243
Update README.md ( #5438 )
...
2-line update
2021-11-01 18:28:14 +01:00
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
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
5d4258fac5
Fix MixConv2d() remove shortcut + apply depthwise ( #5410 )
2021-10-30 13:38:51 +02:00
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
620b535f85
Update sweep.py ( #5402 )
2021-10-29 19:21:59 +02:00
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
a4fece8c14
DDP `nl` fix ( #5332 )
2021-10-25 16:03:39 +02:00
9c31a66f5d
Update `AutoShape.forward()` model.classes example ( #5324 )
2021-10-25 14:03:25 +02:00
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
692be757b6
W&B: Media panel fix ( #5317 )
2021-10-25 12:06:12 +02:00
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
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
e0c3f42de7
Uncomment OpenCV 4.5.4 requirement in detect.py ( #5305 )
2021-10-23 13:40:34 +02:00
79d8f1f678
Created using Colaboratory
2021-10-22 21:19:23 +02:00
b760acec11
Created using Colaboratory
2021-10-22 21:11:29 +02:00
441b47c443
More informative `EarlyStopping()` message ( #5303 )
2021-10-22 20:02:19 +02:00
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
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
a18b0c36cd
Update rebase.yml with workflows permissions ( #5255 )
...
Update token with workflows permissions
2021-10-19 12:02:15 +02:00
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
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
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
8e87396705
Add dependabot for GH actions ( #5250 )
2021-10-19 11:23:30 +02:00
4f5e274eee
Update rebase.yml ( #5245 )
2021-10-19 09:46:05 +02:00
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
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
0000334a6f
Update Objects365.yaml val count ( #5212 )
2021-10-15 12:40:40 -07:00
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
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
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
6b31cfd793
Highlight contributors in README ( #5173 )
...
* Highlight contributors in README
* Adjust Contact section
* Update README.md
2021-10-14 12:49:16 -07:00
7ee5aed0b3
Improved check_suffix() robustness to `''` and `""` ( #5192 )
...
* Improved check_suffix() robustness to `''` and `""`
* Cleanup
2021-10-14 12:00:39 -07:00
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
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
e520e820f1
Update autodownload fallbacks to v6.0 assets ( #5177 )
2021-10-13 19:58:38 -07:00
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
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
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
20a809dee5
Update stale.yml ( #5156 )
2021-10-12 17:54:56 -07:00
26784afd99
Use YOLOv5n for CI testing ( #5154 )
2021-10-12 08:42:26 -07:00
34da872ab6
fix `tf` conversion in new v6 models ( #5153 )
...
* fix `tf` conversion in new v6 (#5147 )
* sort imports
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-12 08:38:54 -07:00
956be8e642
YOLOv5 release v6.0 ( #5141 )
...
* Update P5 models
* Update P6 models
* Update with GFLOPs and Params
* Update with GFLOPs and Params
* Update README
* Update
* Update README
* Update
* Update
* Add times
* Update README
* Update results
* Update results
* Update results
* Update hyps
* Update plots
* Update plots
* Update README.md
* Add nano models to hubconf.py
2021-10-11 23:47:18 -07:00
938efe5dcf
update `detect.py` in order to support torch script ( #5109 )
...
* update detect.py in order to support torch script
This change assumes the torchscrip file was previously saved with `export.py`
* update `detect.py` for torchscript support
Simple update for torchscript support. Assumes the torchscript file has been generated with `export.py`
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-10-11 21:56:06 -07:00
Glenn Jocher
Mrinal Jain
Pranath
Ayush Chaurasia
Jirka Borovec
Zhiqiang Wang
Cristi Fati
Jebastin Nadar
jdfr
dependabot[bot]
Zhiqiang Wang
Farley Lai
NauchtanRobotics
Yoni Chechik
Andrei Ionut Damian