mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
2,476 Commits
10 Branches
10 Tags
584 MiB
Commit Graph

2476 Commits (9bf18554c3e4b250ba7063876f0191f573ffb7a4)
 

Author SHA1 Message Date
Glenn Jocher cff9717d73
Comment EMA assert (#9173) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-26 20:06:26 +02:00
Glenn Jocher 5d3d051c9b
Inline `_make_grid()` meshgrid (#9170) 
* Inline _make_grid() meshgrid 

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-26 15:29:31 +02:00
Glenn Jocher 85d7ae2195
Rename onnx_dynamic -> dynamic (#9168) 2022-08-26 14:34:28 +02:00
Glenn Jocher f2b8f3fe3a Created using Colaboratory 2022-08-25 22:17:28 +02:00
Glenn Jocher 30e674b14d
New YOLOv5 v6.2 splash images (#9142) 
* New YOLOv5 v6.2 splash images

@AyushExel @AlanDimmer 

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Created using Colaboratory

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README_cn.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-25 15:06:20 +02:00
Glenn Jocher 729dc169ba
Fixed Classify offsets (#9155) 2022-08-25 15:04:27 +02:00
Glenn Jocher d07ddc6996
New TryExcept decorator (#9154) 
* New TryExcept decorator

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-25 14:34:26 +02:00
Glenn Jocher f0e5a608f5
Add CSV logging to GenericLogger (#9128) 
Enable CSV logging for Classify training.

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-24 12:31:50 +02:00
Ayush Chaurasia f8816f58b7
Infer Loggers project name (#9117) 
* smart project name inference

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update __init__.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-24 12:15:37 +02:00
Glenn Jocher e6f54c5b32
Fix numpy to torch cls streaming bug (#9112) 
* Fix numpy to torch cls streaming bug

Resolves https://github.com/ultralytics/yolov5/issues/9111

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-23 23:54:05 +02:00
Glenn Jocher 51c9f92297
Streaming Classification support (#9106) 
* Streaming Classification support

* Streaming Classification support

* Streaming Classification support
 2022-08-23 17:54:51 +02:00
Glenn Jocher 48e56d3c9b
Add optional `transforms` argument to LoadStreams() (#9105) 
* Add optional `transforms` argument to LoadStreams()

Prepare for streaming classification support

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Cleanup

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* fix

* batch size > 1 fix

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-23 14:37:46 +02:00
Glenn Jocher d0fa0042bd
New `@try_export` decorator (#9096) 
* New export decorator

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* New export decorator

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Cleanup

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* rename fcn to func

* rename to @try_export

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-23 13:06:33 +02:00
Glenn Jocher eab35f66f9
Install `torch` latest stable (#9092) 
Install torch 1.12.1 stable 

GPU assignment issues in 1.13 nightly that comes with image

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-22 22:18:01 +02:00
Glenn Jocher 06831aa9e9
Improved Usage example docstrings (#9075) 
* Updated Usage examples

* Update detect.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update predict.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-22 01:06:29 +02:00
Glenn Jocher 0abae780b3
Remove unused Timeout import (#9073) 
* Remove unused Timeout import

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-22 00:04:30 +02:00
Glenn Jocher 262187e95d
Two dimensional `size=(h,w)` AutoShape support (#9072) 
* Two dimensional `size=(h,w)` AutoShape support

May resolve https://github.com/ultralytics/yolov5/issues/9039

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update hubconf.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-21 23:26:07 +02:00
Glenn Jocher db2ee5fa0c
De-thread TensorBoard graph logging (#9071) 
* De-thread TensorBoard graph logging

Issues with Classification models

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update __init__.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-21 17:07:56 +02:00
Glenn Jocher 5373a28c1b Created using Colaboratory 2022-08-21 17:01:05 +02:00
Glenn Jocher 8665d557c1
Threaded TensorBoard graph logging (#9070) 
* Log TensorBoard graph on pretrain_routine_end

* fix
 2022-08-21 16:51:50 +02:00
Glenn Jocher 0b8639a40a
Rename 'labels' to 'instances' (#9066) 
* Rename labels to instances

* Rename labels to instances

* align val
 2022-08-21 15:50:02 +02:00
0zppd 87e8deadd5
zero-mAP fix remove `torch.empty()` forward pass in `.train()` mode (#9068) 
* Fix Zero Map Issue

Signed-off-by: 0zppd <111682241+0zppd@users.noreply.github.com>

* Update __init__.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: 0zppd <111682241+0zppd@users.noreply.github.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-21 15:40:28 +02:00
Glenn Jocher e6b4bf0bc2 Created using Colaboratory 2022-08-21 14:21:48 +02:00
Glenn Jocher 1499526f56 Created using Colaboratory 2022-08-21 14:20:12 +02:00
Glenn Jocher 794f117f4b Created using Colaboratory 2022-08-21 13:33:13 +02:00
Glenn Jocher 6077bf032a
TensorBoard fix in tutorial.ipynb (#9064) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-21 13:10:02 +02:00
Glenn Jocher e0700cce77
Daemon `plot_labels()` for faster start (#9057) 
* Daemon `plot_labels()` for faster start

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update train.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-21 03:47:37 +02:00
Glenn Jocher 27fb6fd8fc
zero-mAP fix 3 (#9058) 
* zero-mAP fix 3

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update torch_utils.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update torch_utils.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-21 03:22:24 +02:00
Glenn Jocher 841f312f93
zero-mAP fix return `.detach()` to EMA (#9056) 
Resolves https://github.com/ultralytics/hub/issues/82

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-21 02:23:39 +02:00
Glenn Jocher 93f63ee33f
Refactor for simplification 2 (#9055) 
* Refactor for simplification 2

* Update __init__.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-21 02:00:35 +02:00
Glenn Jocher c725511bfc
Refactor for simplification (#9054) 
* Refactor for simplification

* cleanup

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-21 01:34:03 +02:00
Glenn Jocher f258cf8b37
Add `--save-hybrid` mAP warning (#9050) 
* Add `--save-hybrid` mAP warning

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update val.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-20 17:17:35 +02:00
Glenn Jocher fc8758a49b
Temporarily remove `macos-latest` from CI (#9049) 
* Temporarily remove macos-latest from CI

macos-latest causing many failed CI runs that resolve after manually re-running 2 or 3 times. I don't know what the cause is. Will replace at a later date.

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update ci-testing.yml

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-20 16:45:11 +02:00
Glenn Jocher a409ec7953
AutoBatch protect from negative batch sizes (#9048) 
* AutoBatch protect from negative batch sizes

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* cleanup

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-20 16:29:08 +02:00
Glenn Jocher ba1c6773c2 Created using Colaboratory 2022-08-19 18:41:30 +02:00
Glenn Jocher aed88848a2 Simplified notebook 2022-08-19 18:00:38 +02:00
Glenn Jocher fdcb92a938
Update train.py `import val as validate` (#9037) 
* Update train.py `import val as validate`

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-19 16:54:58 +02:00
Glenn Jocher 4a8ab3bc42
VOC `names` dictionary fix (#9034) 
* VOC names dictionary fix

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update dataloaders.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-19 15:07:04 +02:00
Glenn Jocher 781401ec70
Default --data `imagenette160` training (fastest) (#9033) 
* Default --data `imagenette160` training (fastest)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update train.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update train.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update train.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-19 15:06:37 +02:00
Glenn Jocher 1cd3e752de Created using Colaboratory 2022-08-19 02:01:40 +02:00
Glenn Jocher 840b7232db
Attach transforms to model (#9028) 
* Attach transforms to model

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update val.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update train.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-19 01:59:51 +02:00
Glenn Jocher 4bc5520e94 Created using Colaboratory 2022-08-19 01:30:14 +02:00
Glenn Jocher eb359c3a22
Add PyTorch Hub classification CI checks (#9027) 
* Add PyTorch Hub classification CI checks

Add PyTorch Hub loading of official and custom trained classification models to CI checks. 

May help resolve https://github.com/ultralytics/yolov5/issues/8790#issuecomment-1219840718

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update hubconf.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-18 21:45:11 +02:00
Glenn Jocher a5a47c52ca
Remove unused `time_sync` import (#9026) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-18 20:26:18 +02:00
Glenn Jocher de6e6c0110 Created using Colaboratory 2022-08-18 20:23:14 +02:00
Glenn Jocher 61adf017f2
`torch.empty()` for speed improvements (#9025) 
`torch.empty()` for speed improvement

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-18 20:12:33 +02:00
Glenn Jocher d40cd0d454
Improved `Profile()` inference timing (#9024) 
* Improved `Profile()` class

* Update predict.py

* Update val.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update val.py

* Update AutoShape

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-18 19:55:38 +02:00
cher-liang c0e7a776cd
Faster pre-processing for gray image input (#9009) 
* faster 1 channel to 3 channels image conversion

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-18 14:18:02 +02:00
Glenn Jocher 20049be2e7
EMA FP32 assert classification bug fix (#9016) 
* Return EMA float on classification val

* verbose val fix

* EMA check
 2022-08-18 14:06:15 +02:00
Glenn Jocher 529aafd737
Fix TorchScript JSON string key bug (#9015) 
* Fix TorchScript JSON string key bug

Resolves https://github.com/ultralytics/yolov5/issues/9011

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-18 11:50:24 +02:00