Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity
2,420 Commits 1 Branch 0 Tags
3b1a9d22a45f1e16e21c8e8ebec9ccd17068cd08
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
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
Glenn Jocher abea53ea5b Continue on Docker fail (all backends) fix (#9433) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-16 00:10:10 +02:00
Glenn Jocher 65afaa78be Continue on Docker failure (all backends) (#9432) 
Continue on Docker failure (all)

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-15 23:53:36 +02:00
Glenn Jocher 36cb05b7b2 Continue on Docker arm64 failure (#9430) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-15 23:28:33 +02:00
Glenn Jocherandpre-commit-ci[bot] 1323b48053 Remove .train() mode exports (#9429) 
* Remove `.train()` mode exports

No common use cases.

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-15 19:05:10 +02:00
Glenn Jocher 23701eac7a AMP Check fix (#9367) 
Resolves https://github.com/ultralytics/yolov5/issues/9365

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-11 13:56:51 +03:00
Glenn Jocher 23d0456b08 torch.jit.trace() fix (#9363) 
* 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-10 23:55:18 +03:00
Glenn Jocher cafdd18939 Exclude ipython from hubconf.py check_requirements() (#9362) 
Exclude ipython from hubconf.py `check_requirements()`

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-10 22:58:24 +03:00
Glenn Jocher 2b5c9a83ec labels.jpg names fix (#9361) 
Partially resolves https://github.com/ultralytics/yolov5/issues/9360

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-10 22:24:46 +03:00
Glenn Jocher 4e8504abd9 PaddlePaddle Usage examples (#9358) 2022-09-10 13:25:01 +03:00
Glenn Jocher 57ef676af2 Update check_requirements() multiple string (#9356) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-10 12:14:31 +03:00
Glenn Jocherandpre-commit-ci[bot] e9ddc5b527 Update check_requirements(args, cmds='') (#9355) 
* Update `check_requirements(args, cmds='')`

* [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-10 12:00:16 +03:00
Glenn Jocher 24bf9cceb4 Update check_requirements() single install (#9353) 
* Update `check_requirements()` single install

Faster install and better conflict resolution with single installation

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

* Update

* Update

* Update

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-10 11:11:56 +03:00
Glenn Jocher 5a134e0653 Replace deprecated np.int with int (#9307) 
Per 
```
/content/yolov5/utils/dataloaders.py:458: DeprecationWarning: `np.int` is a deprecated alias for the builtin `int`. To silence this warning, use `int` by itself. Doing this will not modify any behavior and is safe. When replacing `np.int`, you may wish to use e.g. `np.int64` or `np.int32` to specify the precision. If you wish to review your current use, check the release note link for additional information.
Deprecated in NumPy 1.20; for more details and guidance: https://numpy.org/devdocs/release/1.20.0-notes.html#deprecations
```

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-07 10:11:30 +03:00
Glenn Jocher e45d335bbc Update benchmarks CI with --hard-fail min metric floor (#9276) 
* Update benchmarks CI with min metric floor

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

* Update ci-testing.yml

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

* Update benchmarks.py

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

* Update benchmarks.py

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

* Update benchmarks.py

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

* Update benchmarks.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-04 16:35:16 +02:00
Glenn Jocherandpre-commit-ci[bot] 7aa263c5f2 Update DetectMultiBackend for tuple outputs 2 (#9275) 
* Update DetectMultiBackend for tuple outputs 2

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

* Update

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

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

* Update

* Update

* 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-04 15:39:57 +02:00
Glenn Jocher 96c3c7f71d Update DetectMultiBackend for tuple outputs (#9274) 
Update
 2022-09-04 14:01:43 +02:00
Glenn Jocher 63ecce60ea DetectMultiBackend improvements (#9269) 
* Update common.py

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>

* Update common.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-09-04 01:33:38 +02:00
Glenn Jocher 15e82d2967 Update TryExcept(msg='...')` (#9261) 2022-09-02 16:24:30 +02:00
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
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 Jocherandpre-commit-ci[bot] 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
Glenn Jocherandpre-commit-ci[bot] 4a37381ee8 Add ClassificationModel TF export assert (#9226) 
* Add ClassificationModel TF export assert

Export to TF not yet supported, warning alerts users.

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-30 16:18:01 +02:00
Glenn Jocherandpre-commit-ci[bot] 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