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 and pre-commit-ci[bot]
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 and Glenn Jocher
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
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
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 and pre-commit-ci[bot]
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 and pre-commit-ci[bot]
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
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 and Glenn Jocher
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 and pre-commit-ci[bot]
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 and pre-commit-ci[bot]
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 and pre-commit-ci[bot]
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
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
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 and pre-commit-ci[bot]
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 and pre-commit-ci[bot]
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 and pre-commit-ci[bot]
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