mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
1,792 Commits 10 Branches 10 Tags
Commit Graph

1792 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher
ee6c70ae43
Fix yolov3.yaml remove list (#6655) 
Per https://github.com/ultralytics/yolov3/issues/1887#issuecomment-1041135181
 2022-02-16 10:13:05 +01:00
imyhxy
ca0a00784d
Fixed wandb logger KeyError (#6637) 2022-02-14 17:08:36 +01:00
Glenn Jocher
538930023c
Fix ConfusionMatrix scale vmin=0.0 (#6638) 
Fix attempt for https://github.com/ultralytics/yolov5/issues/6626
 2022-02-14 14:54:55 +01:00
Glenn Jocher
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
Ayush Chaurasia
96d8f86085
W&B: don't log media in evolve (#6617) 2022-02-12 13:04:50 +01:00
Ayush Chaurasia
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
Glenn Jocher
3eef80457f
Add YOLOv5n to Reproduce section (#6619) 2022-02-11 21:48:01 +01:00
Glenn Jocher
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
or-toledano
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
Glenn Jocher
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
Glenn Jocher
a5c9057dcc
Update train.py 2022-02-10 14:15:18 +01:00
Glenn Jocher
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
Christoph Fürbacher
9cf80b7f60
return opt from train.run() (#6581) 2022-02-09 15:11:52 +01:00
Glenn Jocher
9c513ca629
Add DATASETS_DIR global in general.py (#6578) 2022-02-08 22:20:44 +01:00
Matthias Vadcard
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
Glenn Jocher
f627bc53f8
Fix plot_labels() colored histogram bug (#6574) 
* Fix `plot_labels()` colored histogram bug

* Cleanup
 2022-02-08 14:03:50 +01:00
Glenn Jocher
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
Glenn Jocher
006eb40ab7
Improved AutoBatch DDP error message (#6568) 
* Improved AutoBatch DDP error message

* Cleanup
 2022-02-08 11:45:30 +01:00
Glenn Jocher
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
Glenn Jocher
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
Jirka Borovec
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
Jirka Borovec
e1a6a0b100
flake8: code meanings (#6481) 2022-02-04 19:54:34 +01:00
bilzard
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
greg2451
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
Glenn Jocher
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
Glenn Jocher
9bc72a3ac2
Edge TPU export 'list index out of range' fix (#6533) 2022-02-04 00:04:06 +01:00
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
cb40c9afda
TODO issues exempt from stale action (#6530) 2022-02-03 18:11:28 +01:00
Glenn Jocher
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
Ayush Chaurasia
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
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
77977e0791
CoreML inference fix list() -> sorted() (#6496) 2022-02-01 22:34:15 +01:00
Glenn Jocher
5e4ff195b2
Improved export.py usage examples (#6495) 
* Improved `export.py` usage examples

* Cleanup
 2022-02-01 22:06:29 +01:00
Glenn Jocher
b884ea36c4
Simplify TF normalized to pixels (#6494) 2022-02-01 21:17:56 +01:00
Glenn Jocher
6445a8137e
Resolve dataset paths (#6489) 2022-02-01 15:54:51 +01:00
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
fe7de6a82d
Revert "Remove dataset_stats() autodownload capability (#6303)" (#6442) 
This reverts commit 3119b2f27c198c6b9c02fb57d3b00b61a7bd2356.
 2022-01-26 19:10:59 +01:00
toschi23
3b7ac28ed1
Add *.asf video support (#6436) 2022-01-26 14:26:02 +01:00
Glenn Jocher
d5966c93f1
Namespace VERBOSE env variable to YOLOv5_VERBOSE (#6428) 
* Verbose updates

* Verbose updates
 2022-01-25 17:57:27 -05:00
Glenn Jocher
ff8646cdea
Update workflows (#6427) 
* Workflow updates

* quotes fix

* best to weights fix
 2022-01-25 17:33:22 -05:00
Motoki Kimura
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
Jonathan Samelson
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
Glenn Jocher
cfecd903a3
Update greetings.yaml email address (#6412) 
* Update `greetings.yaml` email address

* Update greetings.yml
 2022-01-24 15:28:52 -08:00
Glenn Jocher
482af479c0
Add detect.py GIF video inference (#6410) 
* Add detect.py GIF video inference

* Cleanup
 2022-01-24 13:11:11 -08:00
Matthias
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