a45e472358
YOLOv5 Export Benchmarks ( #6613 )
...
* Add benchmarks.py
* Update
* Add requirements
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* dataset autodownload from root
* Update
* Redirect to /dev/null
* sudo --help
* Cleanup
* Add exports pd df
* Updates
* Updates
* Updates
* Cleanup
* dir handling fix
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
* Cleanup2
* Cleanup3
* Cleanup model_type
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-02-12 16:05:43 +01:00
96d8f86085
W&B: don't log media in evolve ( #6617 )
2022-02-12 13:04:50 +01:00
51f7750aad
W&B: Improve resume stability ( #6611 )
...
* log best.pt metrics at train end
* update
* Update __init__.py
* flush callbacks when using evolve
* remember batch size on resuming
* Update train.py
* improve stability of resume
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-02-12 13:03:09 +01:00
3eef80457f
Add YOLOv5n to Reproduce section ( #6619 )
2022-02-11 21:48:01 +01:00
b40bdd6b2d
YouTube dependency fix `youtube_dl==2020.12.2` ( #6612 )
...
Per https://github.com/ultralytics/yolov5/issues/5860#issuecomment-1035320018 by @hdnh2006
2022-02-11 13:46:20 +01:00
cb2ad9f685
Fix `ROOT / data` when running W&B `log_dataset()` ( #6606 )
...
* Fix missing data folder when running log_dataset
* Use ROOT/'data'
* PEP8 whitespace
2022-02-10 16:17:47 +01:00
c21da596f3
Fix `hyp_evolve.yaml` indexing bug ( #6604 )
...
* Fix `hyp_evolve.yaml` indexing bug
Bug caused hyp_evolve.yaml to display latest generation result rather than best generation result.
* Update plots.py
* Update general.py
* Update general.py
* Update general.py
2022-02-10 14:51:29 +01:00
a5c9057dcc
Update train.py
2022-02-10 14:15:18 +01:00
364b7c5111
Fix YouTube dislike button bug in `pafy` package ( #6603 )
...
Per https://github.com/ultralytics/yolov5/issues/6583#issuecomment-1034421945 by @alicera
2022-02-10 13:13:44 +01:00
9cf80b7f60
return `opt` from `train.run()` ( #6581 )
2022-02-09 15:11:52 +01:00
9c513ca629
Add `DATASETS_DIR` global in general.py ( #6578 )
2022-02-08 22:20:44 +01:00
f40854b61b
Allow custom` --evolve` project names ( #6567 )
...
* Update train.py
As see in #6463 , modification on train in evolve process to allow custom save directory.
* fix val
* PEP8
whitespace around operator
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-02-08 16:13:45 +01:00
f627bc53f8
Fix `plot_labels()` colored histogram bug ( #6574 )
...
* Fix `plot_labels()` colored histogram bug
* Cleanup
2022-02-08 14:03:50 +01:00
22da63e30f
Fix zero-export handling with `if any(f):` ( #6569 )
...
* Fix zero-export handling with `if any(f):`
Partial fix for https://github.com/ultralytics/yolov5/issues/6563
* Cleanup
2022-02-08 12:20:39 +01:00
006eb40ab7
Improved AutoBatch DDP error message ( #6568 )
...
* Improved AutoBatch DDP error message
* Cleanup
2022-02-08 11:45:30 +01:00
b57abb17f2
Move trainloader functions to class methods ( #6559 )
...
* Move trainloader functions to class methods
* results = ThreadPool(NUM_THREADS).imap(self.load_image, range(n))
* Cleanup
2022-02-07 18:59:26 +01:00
dc7e093071
Edge TPU TF imports fix ( #6542 )
...
* Edge TPU TF imports fix
Fix for https://github.com/ultralytics/yolov5/issues/6535#issuecomment-1030631526
* Update common.py
2022-02-05 15:45:44 +01:00
cba4303d32
Fix 6 Flake8 issues ( #6541 )
...
* F541
* F821
* F841
* E741
* E302
* E722
* Apply suggestions from code review
* Update general.py
* Update datasets.py
* Update export.py
* Update plots.py
* Update plots.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-02-05 15:22:59 +01:00
e1a6a0b100
flake8: code meanings ( #6481 )
2022-02-04 19:54:34 +01:00
aff0281969
Load checkpoint on CPU instead of on GPU ( #6516 )
...
* Load checkpoint on CPU instead of on GPU
* refactor: simplify code
* Cleanup
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-02-04 18:58:18 +01:00
8fcdf3b60b
Fixing minor multi-streaming issues with TensoRT engine ( #6504 )
...
* Update batch-size in model.warmup() + indentation for logging inference results
* These changes are in response to PR comments
* [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-02-04 18:19:37 +01:00
079b36d72b
Edge TPU `tf.lite.experimental.load_delegate` fix ( #6536 )
...
* Edge TPU `tf.lite.experimental.load_delegate` fix
Fix attempt for #6535
* [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-02-04 09:30:01 +01:00
9bc72a3ac2
Edge TPU export 'list index out of range' fix ( #6533 )
2022-02-04 00:04:06 +01:00
8d05716b66
Edge TPU compiler `sudo` fix ( #6531 )
...
* Edge TPU compiler sudo fix
Allows for auto-install of Edge TPU compiler on non-sudo systems like the YOLOv5 Docker image.
@kalenmike
* Update export.py
* Update export.py
* Update export.py
2022-02-03 20:40:09 +01:00
a82292ec53
Social icons after text ( #6473 )
...
* Social icons after text
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update README.md
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-02-03 19:55:19 +01:00
c3e599cfda
Update val_batch*.jpg for Chinese fonts ( #6526 )
...
* Update plots for Chinese fonts
* make is_chinese() non-str safe
* Add global FONT
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update general.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-02-03 19:09:24 +01:00
cb40c9afda
TODO issues exempt from stale action ( #6530 )
2022-02-03 18:11:28 +01:00
19e0208fc9
Update hyp.scratch-high.yaml ( #6525 )
...
Update `lrf: 0.1`, tested on YOLOv5x6 to 55.0 mAP@0.5:0.95, slightly higher than current.
2022-02-03 12:15:13 +01:00
b73c62ebc5
W&B: Remember batchsize on resuming ( #6512 )
...
* log best.pt metrics at train end
* update
* Update __init__.py
* flush callbacks when using evolve
* remember batch size on resuming
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-02-03 11:29:52 +01:00
4c40933266
Suppress export.run() TracerWarnings ( #6499 )
...
Suppresses warnings when calling export.run() directly, not just CLI python export.py.
Also adds Requirements examples for CPU and GPU backends
2022-02-01 23:52:50 +01:00
842d049e1b
Suppress `torch.jit.TracerWarning` on export ( #6498 )
...
* Suppress torch.jit.TracerWarning
TracerWarnings can be safely ignored.
* Cleanup
2022-02-01 22:59:26 +01:00
77977e0791
CoreML inference fix `list()` -> `sorted()` ( #6496 )
2022-02-01 22:34:15 +01:00
5e4ff195b2
Improved `export.py` usage examples ( #6495 )
...
* Improved `export.py` usage examples
* Cleanup
2022-02-01 22:06:29 +01:00
b884ea36c4
Simplify TF normalized to pixels ( #6494 )
2022-02-01 21:17:56 +01:00
6445a8137e
Resolve dataset paths ( #6489 )
2022-02-01 15:54:51 +01:00
7539cd75c3
Add Product Hunt social media icon ( #6464 )
...
* Social media icons update
* fix URL
* Update README.md
2022-01-28 20:23:17 +01:00
d8b5beb0b0
Fix2 `select_device()` for Multi-GPU ( #6461 )
...
* Fix2 select_device() for Multi-GPU
* Cleanup
* Cleanup
* Simplify error message
* Improve assert
* Update torch_utils.py
2022-01-28 08:18:01 +01:00
856d4e5733
Fix `select_device()` for Multi-GPU ( #6434 )
...
* Fix `select_device()` for Multi-GPU
Possible fix for https://github.com/ultralytics/yolov5/issues/6431
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
2022-01-26 23:47:00 +01:00
fe7de6a82d
Revert "Remove `dataset_stats()` autodownload capability ( #6303 )" ( #6442 )
...
This reverts commit 3119b2f27c
2022-01-26 19:10:59 +01:00
3b7ac28ed1
Add `*.asf` video support ( #6436 )
2022-01-26 14:26:02 +01:00
d5966c93f1
Namespace `VERBOSE` env variable to `YOLOv5_VERBOSE` ( #6428 )
...
* Verbose updates
* Verbose updates
2022-01-25 17:57:27 -05:00
ff8646cdea
Update workflows ( #6427 )
...
* Workflow updates
* quotes fix
* best to weights fix
2022-01-25 17:33:22 -05:00
16563ac5b5
Prefer `tflite_runtime` for TFLite inference if installed ( #6406 )
...
* import tflite_runtime if tensorflow not installed
* rename tflite to tfli
* Attempt tflite_runtime for all TFLite workflows
Also rename tfli to tfl
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-01-25 07:24:24 -08:00
ed9bac8392
Rename logger from 'utils.logger' to 'yolov5' ( #6421 )
...
* Gave a more explicit name to the logger
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-01-25 07:21:06 -08:00
cfecd903a3
Update `greetings.yaml` email address ( #6412 )
...
* Update `greetings.yaml` email address
* Update greetings.yml
2022-01-24 15:28:52 -08:00
482af479c0
Add `detect.py` GIF video inference ( #6410 )
...
* Add detect.py GIF video inference
* Cleanup
2022-01-24 13:11:11 -08:00
8efe97719c
Add `stop_training=False` flag to callbacks ( #6365 )
...
* New flag 'stop_training' in util.callbacks.Callbacks class to prematurely stop training from callback handler
* Removed most of the new checks, leaving only the one after calling 'on_train_batch_end'
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-01-22 16:37:21 -10:00
c43439aa31
Add `albumentations` to Dockerfile ( #6392 )
2022-01-21 21:06:02 -10:00
bd815d48df
FROM nvcr.io/nvidia/pytorch:21.10-py3 ( #6379 )
...
21.12 generates dockerhub errors so rolling back to 21.10 with latest pytorch install. Not sure if this torch install will work on non-GPU dockerhub autobuild so this is an experiment.
2022-01-20 19:15:04 -10:00
1b41a1d059
FROM nvcr.io/nvidia/pytorch:21.12-py3 ( #6377 )
2022-01-20 17:59:41 -10:00
Glenn Jocher
Ayush Chaurasia
or-toledano
Christoph Fürbacher
Matthias Vadcard
Jirka Borovec
bilzard
greg2451
toschi23
Motoki Kimura
Jonathan Samelson
Matthias