Commit Graph

1023 Commits (467a57f01b393989867426261d3e9a95566e3e24)

Author SHA1 Message Date
Glenn Jocher 467a57f01b
Segment prediction labels normalization fix (#10205)
* normalize_segments

* round remove

* swap axes fix
2022-11-18 14:19:46 +01:00
Ryan Echols ff6e6e328e
Fix: try 2 - prevent logging config clobbering (#10192)
* fix: try 2 - prevent logging config clobbering

Previous behavior: loading this repository with `torch.hub.load` clobbers the existing logging configuration by modifying the root logger's configuration.
New behavior: loading this repository with `torch.hub.load` only clobbers the logging configuration for logger `yolov5` and its descendants. This is done in a way compatible with Google Colab

Signed-off-by: Ryan Echols <ryan@shadylakemedia.com>

* chore: fill in comment

no-op so a pre-commit hook can auto-format files

Signed-off-by: Ryan Echols <ryan@shadylakemedia.com>

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

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

Signed-off-by: Ryan Echols <ryan@shadylakemedia.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-17 20:47:46 +01:00
Glenn Jocher 1510111b46
data.yaml `names.keys()` integer assert (#10190)
* data.yaml `names.keys()` integer assert

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-11-17 19:22:18 +01:00
Glenn Jocher e40662ffdd
Revert "fix: prevent logging config clobbering" (#10177)
Revert "fix: prevent logging config clobbering (#10133)"

This reverts commit 166b9f2fa7.
2022-11-16 19:06:19 +01:00
tripleMu 90575107e7
Filter PyTorch 1.13 UserWarnings (#10166)
FilterWarning for torch.distributed._all_gather_base

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-16 16:10:15 +01:00
Ryan Echols 166b9f2fa7
fix: prevent logging config clobbering (#10133)
Previous behavior: loading this repository with `torch.hub.load` clobbers the existing logging configuration by modifying the root logger's configuration.
New behavior: loading this repository with `torch.hub.load` only clobbers the logging configuration for logger `yolov5` and its descendants.

Signed-off-by: Ryan Echols <ryan@shadylakemedia.com>

Signed-off-by: Ryan Echols <ryan@shadylakemedia.com>
2022-11-16 16:09:30 +01:00
Amol Dumrewal 5e03f5fc8c
Fix dataloader filepath modification to perform replace only once and not for all occurences of string (#10163)
* Fix dataloader filepath modification to perform only once and not for all occurences of string

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

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

* cleanup

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>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-15 19:00:33 +01:00
Glenn Jocher 9dd40f0723
Add `join_threads()` (#10086)
* Update __init__.py

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

* Update __init__.py

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-11-15 16:27:07 +01:00
Glenn Jocher ea73386e5a
Add Copy-Paste to `mosaic9()` (#10165)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-15 14:19:47 +01:00
Glenn Jocher abbfd69523
Copy-Paste zero value fix (#10152)
* Copy-Paste zero value fix

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

* Update augmentations.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-14 18:54:49 +01:00
Glenn Jocher 078059c5b3
Update IoU functions (#10123)
Remove box area function and support expandable bbox_iou() calls.

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-10 18:32:34 +01:00
Victor Sonck 75728bb3ea
Mapped project and name to ClearML (#10100)
* Mapped project and name to ClearML

* Add project and task name docs

* [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-11-09 17:45:09 +01:00
Glenn Jocher c1fcfe8cd9
YouTube Usage example comments (#10106)
* YouTube Usage example comments

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-11-09 17:20:26 +01:00
Glenn Jocher 86decb3c49
Add `ultralytics` pip package (#10103) 2022-11-09 15:05:19 +01:00
Glenn Jocher 0039870b89
Update Dockerfile keep default torch installation (#10071)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-08 00:58:00 +01:00
Glenn Jocher 78ed31c95a
IoU `eps` adjustment (#10051)
IoU eps adjustment

Unify h1 and h2 with eps values

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-06 15:06:18 +01:00
Glenn Jocher fde7758468
YOLOv5 AutoCache Update (#10027)
* AutoCache

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

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

* AutoCache

* AutoCache

* AutoCache

* AutoCache

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-04 15:30:12 +01:00
Glenn Jocher 82a558557a
Fix `min_items` (#10001)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 16:41:37 +01:00
Glenn Jocher 067ad9a2d1
Improved `check_online()` robustness (#10000)
* Improved check_online() robustness 

YOLOv5-wide improvement, not just in check_requirements()

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-11-01 15:55:18 +01:00
Glenn Jocher c55e2cd73b
Add `min_items` filter option (#9997)
* Add `min_items` filter option

@AyushExel @Laughing-q dataset filter

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

* Update dataloaders.py

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

* fix

* [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-11-01 14:53:14 +01:00
Glenn Jocher cf99788823
Check online twice on AutoUpdate (#9999)
Increased robustness to network failures

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 14:27:36 +01:00
Glenn Jocher 49156eb1d1
Fix `is_colab()` (#9994)
@AyushExel @kalenmike 

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 12:02:05 +01:00
Glenn Jocher 6e544d5f7c
FROM nvcr.io/nvidia/pytorch:22.10-py3 (#9966)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-29 13:31:01 +02:00
Kalen Michael 575055ce70
Switch from suffix checks to archive checks (#9963)
* fix: switched from suffix checks to archive checks

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

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

* Cleanup

seems like both functions accept Path type input so removing str()

* import always

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-10-28 21:16:03 +02:00
Glenn Jocher f9bb984e81
Add ClearML minimum version requirement (#9933)
* Add ClearML minimum version requirement

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>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-26 14:06:05 +02:00
Glenn Jocher a5d875adca
Add `gnupg` to Dockerfile-cpu (#9932)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-26 13:42:52 +02:00
SSTato fba61e5583
Windows Python 3.7 .isfile() fix (#9879)
* Update dataloaders.py

Signed-off-by: SSTato <1210546396@qq.com>

* Update general.py

Signed-off-by: SSTato <1210546396@qq.com>

* Update general.py

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

* Update dataloaders.py

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

* Update dataloaders.py

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

* Update general.py

Signed-off-by: SSTato <1210546396@qq.com>

* Update ci-testing.yml

Signed-off-by: SSTato <1210546396@qq.com>

* Update ci-testing.yml

Signed-off-by: SSTato <1210546396@qq.com>

* Update ci-testing.yml

Signed-off-by: SSTato <1210546396@qq.com>

* Update general.py

Signed-off-by: SSTato <1210546396@qq.com>

* Update general.py

Signed-off-by: SSTato <1210546396@qq.com>

* Update dataloaders.py

Signed-off-by: SSTato <1210546396@qq.com>

Signed-off-by: SSTato <1210546396@qq.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-24 16:20:47 +02:00
Glenn Jocher eef90572bf
ClearML Dockerfile fix (#9876)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-20 20:16:58 +02:00
Glenn Jocher 6371de8879
Webcam show fix (#9847)
* Webcam show fix

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

* Update common.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-10-18 22:26:53 +02:00
Glenn Jocher d0df6c8403
Update ZipFile to context manager (#9843)
* Update zipFile to context manager

* [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

* Cleanup

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-18 16:35:04 +02:00
Glenn Jocher 010cd0db7d
Update HUBDatasetStats() usage examples (#9842)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-18 15:25:21 +02:00
Glenn Jocher acff977af3
Logger Cleanup (#9828) 2022-10-17 15:30:42 +02:00
Glenn Jocher e3ff780676
Allow PyTorch Hub results to display in notebooks (#9825)
* Allow PyTorch Hub results to display in notebooks

* fix CI

* fix CI

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

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

* fix CI

* fix CI

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

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

* fix CI

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

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

* fix CI

* fix CI

* fix CI

* fix CI

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

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

* fix CI

* fix CI

* fix CI

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-17 14:34:33 +02:00
Glenn Jocher e42c89d4ef
`smart_optimizer()` revert to weight with decay (#9817)
If a parameter does not fall into any other category

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-16 20:51:32 +02:00
Glenn Jocher 16f87bb38e
HUB VOC fix (#9792)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-13 02:32:06 +02:00
Vladislav Veklenko 85ae985b6a
update mask2segments and saving results (#9785)
* update mask2segments and saving results

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

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

* Update general.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>
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-10-13 01:44:01 +02:00
Glenn Jocher 7a69035eb8
Update Loggers (#9760)
* Update

* Update tutorial.ipynb

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

* Update tutorial.ipynb

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

* Update tutorial.ipynb

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

* Update requirements.txt

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

* Update

* Update README.md

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

* Update

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-12 12:53:12 +02:00
Glenn Jocher 209be932de
Fix segment evolution keys (#9742)
* Update

* Cleanup
2022-10-09 23:51:29 +02:00
Paul Guerrie 5ef69ef3e6
Error in utils/segment/general `masks2segments()` (#9724)
When running segmentation predict on gpu, the conversion from tensor to numpy fails. Calling `.cpu()` solves this problem. 

Signed-off-by: Paul Guerrie <97041392+paulguerrie@users.noreply.github.com>

Signed-off-by: Paul Guerrie <97041392+paulguerrie@users.noreply.github.com>
2022-10-06 22:55:15 +02:00
Glenn Jocher 7f097ddb6c
FROM nvcr.io/nvidia/pytorch:22.09-py3 (#9711)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-05 22:29:46 +02:00
KristenKehrer c98128fe71
Allow list for Comet artifact class 'names' field (#9654)
* Update __init__.py

In the Comet logger, when I run train.py, it wants to download the data artifact. It was requiring me to format the 'names' field in the data artifact metadata as a dictionary, so I've changed this so that it also accepts a list.

Signed-off-by: KristenKehrer <34010022+KristenKehrer@users.noreply.github.com>

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

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

* Update utils/loggers/comet/__init__.py

Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
Signed-off-by: KristenKehrer <34010022+KristenKehrer@users.noreply.github.com>

Signed-off-by: KristenKehrer <34010022+KristenKehrer@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-03 00:25:10 +02:00
Anant Sakhare 8a19437690
Added cutout import from utils/augmentations.py to use Cutout Aug in … (#9668)
* Added cutout import from utils/augmentations.py to use Cutout Aug in data loader by un-commenting line 679, 680, 681

* [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-10-01 16:42:31 +02:00
Soumik Rakshit 0860e58557
Disabled upload_dataset flag temporarily due to an artifact related bug (#9652)
* disabled upload_dataset flag temporarily due to an artifact related bug

* [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-09-29 23:01:45 +02:00
Glenn Jocher 2373d5470e
NMS MPS device wrapper (#9620)
* NMS MPS device wrapper

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

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-27 18:02:48 +02:00
Glenn Jocher 1460e57157
Fix `is_url('https://ultralytics.com')` (#9610)
Failing on missing path, i.e. no 'www.'

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-27 00:29:23 +02:00
Glenn Jocher c4c0ee8fc3
Apple MPS -> CPU NMS fallback strategy (#9600)
Until more ops are fully supported this update will allow for seamless MPS inference (but slower MPS to CPU transfer before NMS, so slower NMS times).

Partially resolves https://github.com/ultralytics/yolov5/issues/9596

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-26 14:13:03 +02:00
Glenn Jocher 9006b41498
Add global YOLOv5_DATASETS_DIR (#9586)
Update general.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-25 21:40:28 +02:00
Glenn Jocher 639d82fbab
Update Comet links (#9587)
* Update Comet links

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

* Update tutorial.ipynb

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

* Update README.md

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-25 20:12:57 +02:00
Glenn Jocher 2787ad701f
Add segment line predictions (#9571)
* Add segment line predictions

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

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

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

* Update

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-25 14:52:49 +02:00
Glenn Jocher 55fbac933b
Update default GitHub assets (#9573)
* Update default GitHub assets

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

* Update downloads.py

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

* Update downloads.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-25 02:59:25 +02:00