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
2,780 Commits 10 Branches 10 Tags
Commit Graph

1136 Commits

Author SHA1 Message Date
Glenn Jocher
5d4787baab
cudnn.benchmark = True on Seed 0 (#9259) 
* `cudnn.benchmark = True` on Seed 0

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

* Update general.py

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

* Update general.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-02 16:05:23 +02:00
Glenn Jocher
ffdb58b0e0
Move cudnn.benchmarks(True) to LoadStreams (#9258) 
* Move cudnn.benchmarks(True) to LoadStreams

* Update dataloaders.py

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

* Move cudnn.benchmarks(True) to LoadStreams

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-02 14:12:10 +02:00
Glenn Jocher
9da6d0f9f5
Add LoadImages._cv2_rotate() (#9249) 
Optional manual rotation code per iPhone rotation issue in https://github.com/ultralytics/yolov5/issues/8493

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-01 22:30:26 +02:00
Glenn Jocher
ea98199041
Update general.py (#9252) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-01 19:36:27 +02:00
Ayush Chaurasia
2d082a07bd
Refactor Loggers : Move code outside train.py (#9241) 
* update

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

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

* update

* update

* [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-09-01 19:17:36 +02:00
Glenn Jocher
c91d1db716
Update dataloaders.py (#9250) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-01 14:30:21 +02:00
Glenn Jocher
223c59dbe0
Add url_getsize() function (#9247) 
* Add `url_getsize()` function

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 downloads.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-09-01 12:13:53 +02:00
Glenn Jocher
55b009616b
Python check warning emoji (#9238) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-31 18:53:46 +02:00
spacewalk01
79e181a83b
Add support for *.pfm images (#9230) 
add support for pfm image
 2022-08-31 17:31:13 +02:00
Yannick Merkli
5f1000a499
Remove usage of pathlib.Path.unlink(missing_ok=...) (#9227) 
remove usage of pathlib.Path.unlink(missing_ok=...)

Co-authored-by: Yannick Merkli <ymerkli@latticeflow.ai>
 2022-08-30 21:57:36 +02:00
Glenn Jocher
6e7a7ae7ed
New LetterBox(size) CenterCrop(size), ToTensor() transforms (#9213) 
* New LetterBox transform

YOLOv5 LetterBox class for image preprocessing, i.e. T.Compose([T.ToTensor(), LetterBox(size)])

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

* Update augmentations.py

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

* Update augmentations.py

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

* Update augmentations.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

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

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-30 15:17:58 +02:00
Glenn Jocher
5fb267f3e5
Duplicate segment verification fix (#9225) 
Solution by @Laughing-q  to resolve duplicate segment verification bug in https://github.com/ultralytics/yolov5/pull/9052#issuecomment-1231426638

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-30 13:40:51 +02:00
Glenn Jocher
91a81d48fa
Default AutoBatch 0.8 fraction (#9212) 2022-08-29 20:41:54 +02:00
Glenn Jocher
3c64d89104
AutoBatch protect from extreme batch sizes (#9209) 
If < 1 or > 1024 set output to default batch size 16.

May partially address https://github.com/ultralytics/yolov5/issues/9156

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-29 20:06:00 +02:00
Glenn Jocher
da22e01a68
check_font() fstring update (#9208) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-29 18:10:19 +02:00
Glenn Jocher
fef1913d28
check_font() on notebook init (#9200) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-29 15:12:15 +02:00
Glenn Jocher
f65081c436
Add check_git_status(..., branch='master') argument (#9199) 
Add check_git_status(branch='master') argument

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-29 15:00:55 +02:00
HighMans
cf5d9cbc33
ClearML's set_report_period's time is defined in minutes not seconds. (#9186) 
* ClearML's set_report_period's time is defined in minutes not seconds.

https://clear.ml/docs/latest/docs/references/sdk/hpo_optimization_hyperparameteroptimizer/#set_report_period

set_report_period function takes in time in terms of minutes, not seconds.

Signed-off-by: HighMans <42877729+HighMans@users.noreply.github.com>

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

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

Signed-off-by: HighMans <42877729+HighMans@users.noreply.github.com>
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-29 14:58:55 +02:00
Glenn Jocher
e57275a2d8
Skip AMP check on MPS (#9189) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-29 02:00:42 +02:00
Glenn Jocher
13530402f8
Fix AutoAnchor MPS bug (#9188) 
Resolves https://github.com/ultralytics/yolov5/issues/8862

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-29 01:51:41 +02:00
Glenn Jocher
53711bacea
Update Dockerfile-arm64 to libpython3-dev (#9187) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-28 22:14:21 +02:00
Glenn Jocher
985e000d81
Update Dockerfile-cpu to libpython3-dev (#9185) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-28 21:48:58 +02:00
Glenn Jocher
f58fe6b6c1
Update Dockerfile-cpu (#9184) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-28 21:36:05 +02:00
HighMans
ffbce3858a
Fix confidence threshold for ClearML debug images (#9174) 
* Fix confidence threshold

The confidence is converted to a percentage on line 144, but it is being compared to a default conf_threshold value of a decimal value instead of percent value.

Signed-off-by: HighMans <42877729+HighMans@users.noreply.github.com>

* Revert "Fix confidence threshold"

This reverts commit f84a09967f83d70626ca8dfe7625dce60fb0102e.

* Fix confidence comparison

Fix the confidence percentage is being compared to a decimal value.

Signed-off-by: HighMans <42877729+HighMans@users.noreply.github.com>
 2022-08-27 01:39:11 +02:00
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
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
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
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
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
8665d557c1
Threaded TensorBoard graph logging (#9070) 
* Log TensorBoard graph on pretrain_routine_end

* fix
 2022-08-21 16:51:50 +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
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
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
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
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
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
0922bc2082
DockerHub tag update Usage example (#9005) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-17 22:50:08 +02:00
Ayush Chaurasia
64e0757edf
[Classify]: Allow inference on dirs and videos (#9003) 
* allow image dirs

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

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

* Update predict.py

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

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

* Update dataloaders.py

* Update predict.py

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

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

* Update predict.py

* Update predict.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-17 22:41:43 +02:00
Glenn Jocher
e83b422a69
Update dataset names from array to dictionary (#9000) 
* Migrate dataset names to dictionary

* fix check

* backwards compat

* predict fix

* val fix

* Keep dataset stats behavior identical

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-17 17:52:53 +02:00