Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity
2,456 Commits 1 Branch 0 Tags
abbfd695232b1bfcbd8e122e2aeb37fcc3d146d5
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
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 f33718f36f Update HUB banner image (#10134) 
* Update HUB banner image

Passed through tinypng for filesize reduction

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

* Update README_cn.md

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

* Update README_cn.md

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-11 21:20:48 +01:00
Glenn Jocherandpre-commit-ci[bot] 55e9516846 Add Ultralytics HUB to README (#10070) 
* Add Ultralytics HUB section to README

@pderrenger @kalenmike @AlanDimmer @AyushExel new Ultralytics HUB section in YOLOv5 README.

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>

* Update README.md

@AlanDimmer @kalenmike new integrations image

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 README.md

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 README.md

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

* Update tutorial.ipynb

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

* Update README_cn.md

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-11-11 00:21:29 +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
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 e00d02d78b Use MNIST160 (#10069) 
New 160-image MNIST subset composed of first 8 examples of each class. Suitable for fast CI.

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-07 23:07:10 +01:00
Glenn Jocher 6ae3dff7d4 Update get_coco.sh (#10057) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-06 18:43:07 +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 Jocherandpre-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 02b8a4c21b Update default --epochs 100 (#10024) 
* Update default `--epochs 100`

@AyushExel @kalenmike updating default Detection and Segmentation trainings to 100 epochs

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

* Update train.py

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

* Update README.md

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

* Update README_cn.md

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-03 17:58:15 +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 Jocherandpre-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 Jocherandpre-commit-ci[bot] e704970f7f Update README.md (#9970) 
* Update README.md

@taliabender updated spacing per our convo

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 README.md

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-10-31 12:43:17 +01:00
Glenn Jocherandpre-commit-ci[bot] a625f29967 Full-size proto code (optional) (#9980) 
* Update tf.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 tf.py

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

* Update tf.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-10-30 22:09:36 +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
Glenn Jocher 6db0fac66c Update README.md (#9961) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-10-28 18:25:33 +02:00
Glenn Jocherandpre-commit-ci[bot] be9ef3871e Update README.md (#9958) 
* Update README.md

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

* Update README_cn.md

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 tutorial.ipynb

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-10-28 17:48:41 +02:00
Glenn Jocher 38e5aae9a2 Update README.md (#9957) 
* Update README.md

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>

* Update README_cn.md

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-10-28 17:37:25 +02:00
Glenn Jocher 32a9218573 Update Comet Integrations table text (#9937) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-10-26 23:51:40 +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
Glenn Jocher 8236d8818b Created using Colaboratory 2022-10-25 18:13:48 +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 3b1a9d22a4 Fix OpenVINO Usage example (#9874) 
* Fix OpenVINO Usage example

* Fix OpenVINO Usage example
 2022-10-20 19:54:07 +02:00
Glenn Jocherandpre-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 c4710012d8 Update README.md (#9846) 
@pderrenger 

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-10-18 21:22:37 +02:00
Glenn Jocherandpre-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 f1482b0667 Remove ipython from check_requirements exclude list (#9841) 
May resolve https://github.com/ultralytics/yolov5/commit/e3ff7806769444de864060494d1be8e18ce046a1#commitcomment-87136818

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-10-18 13:54:33 +02:00
Glenn Jocher acff977af3 Logger Cleanup (#9828) 2022-10-17 15:30:42 +02:00
Glenn Jocherandpre-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 df80e7c723 Argoverse HUB fix (#9809) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-10-14 18:18:58 +02:00
Glenn Jocher 2a19d070d8 Fix xView dataloaders import (#9807) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-10-14 12:28:52 +02:00
Glenn Jocher 15b75659dd Update hubconf.py local repo Usage example (#9803) 
* Update hubconf.py

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

* Update hubconf.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-10-13 15:27:16 +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
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 2f1eb21ad6 Remove #YOLOVISION22 notice (#9751) 
Update README.md

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

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

* Cleanup
 2022-10-09 23:51:29 +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
Glenn Jocher e4398cf179 TensorRT --dynamic fix (#9691) 
* Update export.py

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

* Update export.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-10-04 16:32:19 +02:00
Glenn Jocher 1158a50abd Simplify val.py benchmark mode with speed mode (#9674) 
Update
 2022-10-02 13:37:54 +02:00
Glenn Jocher 82bec4c878 Add NVIDIA Jetson Nano Deployment tutorial (#9656) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-29 23:35:39 +02:00
Glenn Jocher 799e3d0cc9 Add SegmentationModel unsupported warning (#9632) 
* Add SegmentationModel unsupported warning

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

* Update hubconf.py

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

* Update hubconf.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-28 16:43:11 +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 7314363f26 Add results.save(save_dir='path', exist_ok=False) (#9617) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-27 16:58:14 +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 6b2c9d1d0f Update export.py Usage examples (#9609) 
* Update export.py Usage examples

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

* Update export.py

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

* Update export.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-26 20:46:50 +02:00
Glenn Jocher a5748e4b93 Updated Segmentation and Classification usage (#9607) 
* Updated Segmentation and Classification usage

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

* Update export.py

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

* Update export.py

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

* Update export.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-26 20:10:24 +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 bd9c0c42ae #YOLOVISION22 update (#9598) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-26 13:27:34 +02:00
Glenn Jocher 959a4665f8 #YOLOVISION22 announcement (#9590) 
* #YOLOVISION22 announcement

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 23:26:15 +02:00
Glenn Jocher 9f1cf8dd1c Add Paperspace Gradient badges (#9588) 
* Add Paperspace Gradient badges

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

* Update README.md

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 tutorial.ipynb

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

* Update README.md

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

* Update tutorial.ipynb

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

* Update greetings.yml

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

* Update greetings.yml

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

* Update README_cn.md

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-25 23:03:14 +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 966b0e09f0 TensorRT detect.py inference fix (#9581) 
* Update

* Update ci-testing.yml

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

* Update ci-testing.yml

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

* Segment fix

* Segment fix

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-25 16:21:26 +02:00
Glenn Jocherandpre-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
Glenn Jocher c8e52304cf New scale_segments() function (#9570) 
* Rename scale_coords to scale_boxes

* add scale_segments
 2022-09-24 16:02:41 +02:00
Glenn Jocher 1320ce183e Update is_url() (#9566) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-23 23:20:19 +02:00
Glenn Jocher b25d5a75f2 Refactor dataset batch-size (#9551) 2022-09-22 23:23:40 +02:00
Glenn Jocherandpre-commit-ci[bot] 489920ab30 TF.js fix (#9536) 
* TF.js fix

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

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 ci-testing.yml

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

* Update ci-testing.yml

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

* Update ci-testing.yml

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-22 00:34:35 +02:00
Glenn Jocherandpre-commit-ci[bot] 999482b451 import re (#9535) 
* import re

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-09-21 23:08:52 +02:00
Glenn Jocherandpre-commit-ci[bot] 6f0284763b Update albumentations (#9503) 
* Add `RandomResizedCrop(ratio)`

* Update ratio

* Update ratio

* Update ratio

* Update ratio

* Update ratio

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

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

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

* Update augmentations.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-21 19:53:26 +02:00
Glenn Jocher db6847431b Update coremltools>=6.0 (#9532) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-21 17:55:25 +02:00
Glenn Jocher 499a6bf573 Update scikit-learn constraint per coremltools 6.0 (#9531) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-21 15:14:54 +02:00
Glenn Jocher 6ebef28894 Remove scikit-learn constraint on coremltools 6.0 (#9530) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-21 15:12:12 +02:00
Glenn Jocher 77dcf55168 FROM nvcr.io/nvidia/pytorch:22.08-py3 (#9520) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-20 22:57:42 +02:00
Glenn Jocherandpre-commit-ci[bot] c0d9713845 TFLite fix 2 (#9518) 
* TFLite fix 2

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-09-20 19:01:03 +02:00
Glenn Jocherandpre-commit-ci[bot] bd35191033 Fix TF Lite exports (#9517) 
* Update tf.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-09-20 18:47:14 +02:00
Glenn Jocher e233c038ed Remove check_requirements('flatbuffers==1.12') (#9514) 
* Remove `check_requirements('flatbuffers==1.12')`

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

* Update ci-testing.yml

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

* Update ci-testing.yml

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-20 16:38:04 +02:00
Glenn Jocherandpre-commit-ci[bot] 095f601d9d Segmentation polygons2masks_overlap() in np.int32 (#9493) 
* Segmentation `polygons2masks_overlap()` in `np.int32`

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

WARNING: Masks should be uint8 for fastest speed, change needs profiling results to determine impact.

@AyushExel @Laughing-q 

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

* Update dataloaders.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-09-20 12:22:02 +02:00
Glenn Jocher 0171198f38 Fix visualization title bug (#9500) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-19 20:38:11 +02:00
Glenn Jocher 868c0e9bbb Update Detect() grid init for loop (#9494) 
May resolve threaded inference issue in https://github.com/ultralytics/yolov5/pull/9425#issuecomment-1250802928 by avoiding memory sharing on init.


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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-19 13:31:24 +02:00
Glenn Jocher f038ad7172 AutoBatch report include reserved+allocated (#9491) 
May resolve https://github.com/ultralytics/yolov5/issues/9287#issuecomment-1250767031

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-19 12:18:55 +02:00
Glenn Jocherandpre-commit-ci[bot] fda8aa551d TensorFlow SegmentationModel support (#9472) 
* TensorFlow SegmentationModel support

* TensorFlow SegmentationModel support

* TensorFlow SegmentationModel support

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

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

* TFLite fixes

* GraphDef fixes

* Update ci-testing.yml

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-18 19:52:46 +02:00
Glenn Jocher 120e27e38e classify/predict --save-txt fix (#9478) 
Classify --save-txt

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-18 19:34:10 +02:00
Glenn Jocherandpre-commit-ci[bot] 92b52424d4 TensorFlow macOS AutoUpdate (#9471) 
* TensorFlow macOS AutoUpdate

* [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-18 17:34:34 +02:00
Glenn Jocherandpre-commit-ci[bot] ca9c993d6c Standardize warnings with WARNING ⚠️ ... (#9467) 
* Standardize warnings

* [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-18 16:15:25 +02:00
Glenn Jocher 295c5e9d3c Update ClassificationModel default training imgsz=224 (#9469) 
Update ClassificationModel default training imgsz=224

To match classify/val.py and classify/predict.py

Helps https://github.com/ultralytics/yolov5/issues/9462

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-18 16:13:22 +02:00
Glenn Jocher 4d50cd3469 Conv() dilation argument fix (#9466) 
Resolves https://github.com/ultralytics/yolov5/issues/9384

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-18 15:02:04 +02:00
Glenn Jocherandpre-commit-ci[bot] dc42e6ef22 TensorRT SegmentationModel fix (#9465) 
* TensorRT SegmentationModel fix

* TensorRT SegmentationModel fix

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

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

* TensorRT SegmentationModel fix

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

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

* TensorRT SegmentationModel fix

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

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

* TensorRT SegmentationModel fix

* TensorRT SegmentationModel fix

* fix

* sort output names

* Update ci-testing.yml

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

* Update ci-testing.yml

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-18 14:45:08 +02:00
Glenn Jocherandpre-commit-ci[bot] 95cef1ae6b Optimize imports (#9464) 
* Optimize imports

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

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

* Reformat

* [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-18 12:42:23 +02:00
Glenn Jocher e8a9c5ae41 Add macos-latest runner for CoreML benchmarks (#9453) 
* Add `macos-latest` runner for CoreML benchmarks

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

* Update common.py

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

* Update ci-testing.yml

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-18 00:57:48 +02:00
Glenn Jocher afb9860522 Add Paddle exports to benchmarks (#9459) 
* Add Paddle exports to benchmarks

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

* Update plots.py

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

* Update common.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-18 00:17:31 +02:00
Glenn Jocherandpre-commit-ci[bot] 0608374065 Detect() and Segment() fixes for CoreML and Paddle (#9458) 
* Detect() and Segment() fixes for CoreML and Paddle

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-09-17 23:17:59 +02:00
Glenn Jocher 6a9fffd19a Update general.py (#9454) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-17 15:42:24 +02:00
Glenn Jocher 4a4308001c Do not move downloaded zips (#9455) 
* Do not move downloaded zips

Prevent multiple downloads on HUB of same dataset

@kalenmike 

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-17 15:19:43 +02:00
Glenn Jocher 5e1a9553fb AutoBatch cudnn.benchmark=True fix (#9448) 
* AutoBatch `cudnn.benchmark=True` fix

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

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

* Update autobatch.py

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

* Update autobatch.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-16 21:46:07 +02:00
Glenn Jocher db06f495db AutoUpdate TensorFlow in export.py (#9447) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-16 20:44:56 +02:00
Glenn Jocher fe10b4abc0 Update Detections() instance printing (#9445) 
* Update Detections() instance printing

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

* Update common.py

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

* Update common.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-16 18:36:55 +02:00
Glenn Jocher 2ac4b634c7 AutoShape explicit arguments fix (#9443) 
* AutoShape explicit arguments fix

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

* Update common.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-16 17:25:44 +02:00
Glenn Jocherandpre-commit-ci[bot] 03f2ca8eff Fix TensorRT exports to ONNX opset 12 (#9441) 
* Fix TensorRT exports to ONNX opset 12

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-09-16 12:31:43 +02:00
Glenn Jocher a1e5f9a97d New model.yaml activation: field (#9371) 
* New model.yaml `activation:` field

Add optional model yaml activation field to define model-wide activations, i.e.:

```yaml
activation: nn.LeakyReLU(0.1)  # activation with arguments
activation: nn.SiLU()  # activation with no arguments
```

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

* Update yolo.py

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

* Add example models

* l to m models

* update

* Add yolov5s-LeakyReLU.yaml

* Update yolov5s-LeakyReLU.yaml

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-16 00:55:21 +02:00
Glenn Jocher 58ad5ca5ce Fix val.py zero-TP bug (#9431) 
Resolves https://github.com/ultralytics/yolov5/issues/9400

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

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