Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity
2,524 Commits 18 Branches 0 Tags
b2a0f1cdc579bd81b3c4543752abaa4a90a53c8b
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher b2a0f1cdc5 Update onnx>=1.12.0 (#10526) 2022-12-18 20:06:01 +01:00
Glenn Jocher b2f94e8c35 Update to ONNX opset 17 (#10522) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-12-17 12:26:57 +01:00
Glenn Jocherandpre-commit-ci[bot] 1607aec431 Automatic README translation to Simplified Chinese (#10445) 
* Create translate-readme.yml

@AyushExel @pderrenger @Laughing-q adding README translation action since we are unable to manually maintain our Chinese-translated README

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>

* Double hyperlinks

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

* Delete README_cn.md

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

* Create README.zh-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 README.md

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

* Update .pre-commit-config.yaml

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-12-08 17:27:36 -08:00
Glenn Jocher de812396fe Add README App section (#10446) 
* Add README App section

@AyushExel @pderrenger this should increase our app visibility per https://github.com/ultralytics/yolov5/issues/10431

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-12-08 17:25:45 -08:00
Glenn Jocherandpre-commit-ci[bot] 9722e6ffe5 process_mask_native() cleanup (#10366) 
* process_mask_native() cleanup

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

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

* fix arg name

* cleanup anno_json

* Remove scale_image

* Remove scale_image

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

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

* update to native

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-12-03 14:41:08 -08:00
Glenn Jocher a1b6e79ccf Revert TQDM bar format changes (#10343) 
Per https://github.com/ultralytics/yolov5/issues/10342

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-12-02 13:28:33 -08:00
Glenn Jocher e96113e485 Update CITATION.cff to version: v7.0 (#10389) 
Update version: v7.0

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-12-02 13:27:45 -08:00
Glenn Jocher d1ffc3a3a7 Create CITATION.cff (#10387) 
* Create CITATION.cff

@AyushExel @pderrenger new citation file!! :)

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

* Update CITATION.cff

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-12-02 13:18:39 -08:00
Glenn Jocher 7845cea913 Fix ClearML unconfigured error (#10369) 
@thepycoder adds Try Except for installed but unconfigured clearml environments.

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-12-01 13:56:33 -08:00
Glenn Jocher 185d475d93 Add DNN warning comment (#10368) 
Update export.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-12-01 13:01:46 -08:00
Glenn Jocher 7f5724ba4b Correct Segmentation Comparison Plot (#10344) 
@AyushExel @Laughing-q updated plot here in README

Addresses https://github.com/ultralytics/yolov5/pull/10245#issuecomment-1328482213

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-29 19:38:29 -08:00
Glenn Jocher 10c025d794 Add README License section (#10327) 
* Add README License section

@pderrenger @AyushExel 

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

* live fix

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

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

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-11-28 14:50:29 -08:00
Glenn Jocher f9ca3657f8 Ultralytics Live 1 - ClearML https://youtu.be/KS4weDInJYs (#10324) 
* Ultralytics Live Session banner - ClearML

@taliabender @thepycoder @pderrenger 

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

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-11-28 09:20:59 -08:00
Glenn Jocher 350e8eb69e Fix SegmentationModel Usage (#10303) 
Update export.py

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-26 08:33:43 -08:00
Glenn Jocher 85f8379a68 README Segmentation Usage fixes (#10298) 
Fixes per https://github.com/ultralytics/yolov5/issues/10288

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-25 21:06:22 -08:00
Glenn Jocher 31c1f11186 bbox_iou() optimizations (#10296) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-25 15:28:50 -08:00
Glenn Jocher bfa1f23045 FROM nvcr.io/nvidia/pytorch:22.11-py3 (#10279) 
* Update Docker usage examples

* Update Dockerfile

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

* Update DEBIAN_FRONTEND

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-24 16:34:14 +01:00
Glenn Jocher 7398d2d77c Created using Colaboratory 2022-11-22 21:27:33 +01:00
Glenn Jocher 61ebf5e5ed Created using Colaboratory 2022-11-22 20:47:54 +01:00
Glenn Jocher c9d47ae056 Created using Colaboratory 2022-11-22 20:37:45 +01:00
Glenn Jocher b32f67f6be --single-cls segments fix (#10260) 
--single-cls segments fix

May resolve #10230

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-22 20:30:14 +01:00
Glenn Jocherandpre-commit-ci[bot] 915bbf294b YOLOv5 v7.0 release updates (#10245) 
* YOLOv5 v7.0 splash image 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>

* readme segmentation section

* readme segmentation section

* readme segmentation section

* readme segmentation section

* readme segmentation section

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

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

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

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

* Update README.md

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

* Update download URLs to 7.0 assets

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-22 16:23:47 +01:00
Glenn Jocher be348cc339 Validate --task speed CPU fix (#10244) 2022-11-21 16:54:36 +01:00
Glenn Jocher 72cad39854 Squeezenet reshape outputs fix (#10222) 
@AyushExel

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-19 16:44:56 +01:00
Glenn Jocherandpre-commit-ci[bot] 40bb8030f8 Scope gitpyhon import in check_git_info() (#10221) 
* Scope gitpyhon import in `check_git_info()`

* [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-11-19 14:22:07 +01:00
Glenn Jocher 6992dde4bd Update Comet preview image (#10220) 
* Update Comet preview image

Pass through tinyjpg: 2.2MB -> 497kB :)

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>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-19 13:44:46 +01:00
Glenn Jocher 0307954e4e Add git info to cls, seg checkpoints (#10217) 2022-11-19 03:32:21 +01:00
Glenn Jocherandpre-commit-ci[bot] 9286336cb4 Add git info to training checkpoints (#9655) 
* Add git status on train checkpoints

* Update

* Update

* Update

* Update

* [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 general.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-11-19 03:16:17 +01:00
Glenn Jocher 9bc60349b6 Add --source screen Usage example (#10215) 2022-11-18 23:48:47 +01:00
Glenn Jocher 9bf18554c3 Revert --save-txt to default False (#10213) 
* Revert `--save-txt` to default False

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

* Update tutorial.ipynb

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-18 23:46:15 +01:00
Glenn Jocher 23f43ce0cc Created using Colaboratory 2022-11-18 23:12:09 +01:00
Glenn Jocher 7fc7ed77e0 Created using Colaboratory 2022-11-18 22:41:46 +01:00
Glenn Jocher 2ecaa96c84 Created using Colaboratory 2022-11-18 22:17:52 +01:00
Glenn Jocher 005161514f Remove Colab notebook High-Memory notices (#10212) 
* Remove Colab notebook High-Memory notices

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

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

* Update tutorial.ipynb

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-11-18 22:10:35 +01:00
Glenn Jocherandpre-commit-ci[bot] 0322bb3196 New global TQDM_BAR_FORMAT (#10211) 
* New global TQDM_BAR_FORMAT

* [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-11-18 21:39:13 +01:00
Glenn Jocherandpre-commit-ci[bot] 74b3886edd Simplify dataloader tqdm descriptions (#10210) 
* Simplify dataloader tqdm descriptions

@AyushExel this should help our tqdm dataloader messages fit better within a single line in our Colab notebooks and also help avoid confusion about missing/empty labels, now combined into 'backgrounds'.

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 dataloaders.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-11-18 21:18:57 +01:00
Glenn Jocher 946765bbc3 Created using Colaboratory 2022-11-18 16:32:29 +01:00
Glenn Jocher 241d798bb4 Created using Colaboratory 2022-11-18 15:05:25 +01:00
Glenn Jocher 467a57f01b Segment prediction labels normalization fix (#10205) 
* normalize_segments

* round remove

* swap axes fix
 2022-11-18 14:19: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
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 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