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>
3 years ago
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>
3 years ago
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>
3 years ago
Zhiqiang Wang
8e87396705
Add dependabot for GH actions ( #5250 )
3 years ago
Glenn Jocher
4f5e274eee
Update rebase.yml ( #5245 )
3 years ago
Glenn Jocher
752d97df2c
Add `on_fit_epoch_end` callback ( #5232 )
* Add `on_fit_epoch_end` callback
* Add results to train
* Update __init__.py
3 years ago
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
3 years ago
Glenn Jocher
0000334a6f
Update Objects365.yaml val count ( #5212 )
3 years ago
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
3 years ago
Farley Lai
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>
3 years ago
Glenn Jocher
1c9f710b46
Add hyp.scratch-med.yaml ( #5196 )
* Add hyp.scratch-med.yaml
* Update hyp.scratch-med.yaml
3 years ago
Glenn Jocher
6b31cfd793
Highlight contributors in README ( #5173 )
* Highlight contributors in README
* Adjust Contact section
* Update README.md
3 years ago
Glenn Jocher
7ee5aed0b3
Improved check_suffix() robustness to `''` and `""` ( #5192 )
* Improved check_suffix() robustness to `''` and `""`
* Cleanup
3 years ago
NauchtanRobotics
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>
3 years ago
Ayush Chaurasia
1172ac15fb
W&B: DDP fix ( #5176 )
* fix dpp with wandb disabled
* PyCharm reformat
* Fix
* fix
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
e520e820f1
Update autodownload fallbacks to v6.0 assets ( #5177 )
3 years ago
Ayush Chaurasia
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>
3 years ago
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
3 years ago
Glenn Jocher
b754525e99
Check `'onnxruntime-gpu' if torch.has_cuda` ( #5087 )
* Check `'onnxruntime-gpu' if torch.has_cuda`
* fix indent
3 years ago
Glenn Jocher
20a809dee5
Update stale.yml ( #5156 )
3 years ago
Glenn Jocher
26784afd99
Use YOLOv5n for CI testing ( #5154 )
3 years ago
Yoni Chechik
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>
3 years ago
Glenn Jocher
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
3 years ago
Andrei Ionut Damian
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>
3 years ago
Glenn Jocher
0bf24cf641
Add OpenCV DNN option for ONNX inference ( #5136 )
* Add OpenCV DNN option for ONNX inference
Usage:
```bash
python detect.py --weights yolov5s.onnx # ONNX Runtime inference
python detect.py --weights yolov5s.onnx -dnn # OpenCV DNN inference
```
* DNN prediction to tensor
* Update detect.py
3 years ago
Jebastin Nadar
9d75e42f98
Refactor `Detect()` anchors for ONNX <> OpenCV DNN compatibility ( #4833 )
* refactor anchors and anchor_grid in Detect Layer
* fix CI failures by adding compatibility
* fix tf failure
* fix different devices errors
* Cleanup
* fix anchors overwriting issue
* better refactoring
* Remove self.anchor_grid shape check (redundant with self.grid check)
Also PEP8 / 120 line width
* Convert _make_grid() from static to dynamic method
* Remove anchor_grid.to(device)
clone() should already clone to same device as self.anchors
* fix different devices error
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Malte Lorbach
153873e9e4
Fix `ROOT` as relative path ( #5129 )
* use os.path.relpath instead of relative_to
* use os.path.relpath instead of relative_to
* Remove os.path from val.py
* Remove os.path from train.py
* Update detect.py import to os
* Update export.py import to os
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
4cf7d487a6
Update plots.py ( #5127 )
3 years ago
Glenn Jocher
7d37b3c52e
Update val.py `pad = 0.0 if task == speed else 0.5` ( #5121 )
* Update val.py `pad = 0.0 if task == speed else 0.5`
* Cleanup
3 years ago
Glenn Jocher
ba4b79de8b
Update val.py `--speed` and `--study` usages ( #5120 )
3 years ago
Glenn Jocher
4f9718abe6
Pass `--device` for `--task study` ( #5118 )
3 years ago
Jebastin Nadar
a0e15046c3
Fix different devices bug when moving model from GPU to CPU ( #5110 )
* fix different devices bug
* extend _apply() instead of to() for a general fix
* Only apply if Detect() is last layer
Co-authored-by: Jebastin Nadar <njebastin10@gmail.com>
* Indent fix
* Add comment to yolo.py
* Add comment to common.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Nan
4a6dfffdaa
Pass `LOCAL_RANK` to `torch_distributed_zero_first()` ( #5114 )
Co-authored-by: qiningonline <qiningonline@gmail.com>
3 years ago
hamid mohammadi
97b6b14abe
Add `requests` to requirements.txt ( #5112 )
* Requirements updated
1. Requests added to requirements.txt. That might not be included in all docker base images, adding it to the requirements is safer.
2. Added a minimum version to Pandas. It's a good practice to have versions for all dependencies.
* Sort alphabetically
* Update requirements.txt
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
276b674513
Fix SKU-110K HUB: `OSError` ( #5106 )
3 years ago
Egor Orel
48b00dbc0d
Re-open IP-camera videostream if disconnected ( #5074 )
* Re-open IP-camera videostream if disconnected. Provide IP-streaming analysis stability
* Re-open IP-camera videostream if disconnected. Provide IP-streaming analysis stability
* `self.imgs[i] *= 0` bug fix and cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
57ce73cd4f
Update datasets.py comments ( #5088 )
3 years ago
Glenn Jocher
ce8e5dc8bb
Update Dockerfile to `ADD` Arial.ttf ( #5084 )
* Update Dockerfile to `ADD` Arial.ttf
Fix for `Arial.ttf download in Docker` #5078
* Cleanup
* Cleanup2
3 years ago
Glenn Jocher
070af88108
Fix `yaml.safe_load()` ignore emoji errors ( #5060 )
3 years ago
Glenn Jocher
5afc9c25ef
Implement `--save-period` locally ( #5047 )
This PR adds a new training argument `--save-period` to save training checkpoints every `x` epochs. To save training every 50 epochs for example:
```
python train.py --save-period 50 # saves epoch50.pt, epoch100.pt, epoch150.pt, ... etc.
```
This saved checkpoints in addition to existing last.pt and best.pt checkpoints and does not affect their behavior. Default value is -1, i.e. disabled.
3 years ago
Glenn Jocher
d133968b1e
Clip TTA Augmented Tails ( #5028 )
* Clip TTA Augmented Tails
Experimental TTA update.
* Update yolo.py
* Update yolo.py
* Update yolo.py
* Update yolo.py
3 years ago
Zhiqiang Wang
1922ddeac0
Fix pylint: do not use bare 'except' ( #5025 )
* Fix E722, do not use bare 'except'
* Remove used codes
* Add FileNotFoundError in LoadImagesAndLabels
* Remove AssertionError
* Ignore LoadImagesAndLabels
* Ignore downloads.py
* Ignore torch_utils.py
* Ignore train.py
* Ignore datasets.py
* Enable utils/download.py
* Fixing exception in thop
* Remove unused code
* Fixing exception in LoadImagesAndLabels
* Fixing exception in exif_size
* Fixing exception in parse_model
* Ignore exceptions in requests
* Revert the exception as suggested
* Revert the exception as suggested
3 years ago
Glenn Jocher
b0ade48457
Fix missing `opt.device` on `--task study` ( #5031 )
3 years ago
Glenn Jocher
cce7e78fae
Created using Colaboratory
3 years ago
Glenn Jocher
89a6e9ffa7
Compute loss on final val ( #5017 )
3 years ago
Glenn Jocher
b20e381492
Update README.md ( #5015 )
3 years ago
Glenn Jocher
452e84703c
Update train.py ( #5014 )
3 years ago
Kalen Michael
cce313d67d
Add YOLOv5 Survey link to README.md ( #5000 )
* added callbacks
* added back callback to main
* added save_dir to callback output
* merged in upstream
* removed ghost code
* added survey
* Update README.md
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
ba99092304
Update W&B README ( #5006 )
3 years ago
Glenn Jocher
1916226841
Created using Colaboratory
3 years ago