Commit Graph
100 Commits
Author SHA1 Message Date
Glenn Jocher 42812fcb18 Add bash script arguments (#8962) 2022-08-14 22:33:55 +02:00
Glenn Jocher 7639e4c7ef Fix windows LOGGER with emojis output (#8958)
* Fix duplicate LOGGER output

* Update general.py
2022-08-14 00:12:09 +02:00
Glenn Jocher dc38cd03f4 New smart_inference_mode() conditional decorator (#8957)
New smart_inference_mode()
2022-08-13 20:38:51 +02:00
Glenn Jocher 6aed0a7c00 GFLOPs computation fix for classification models (#8954)
* GFLOPs computation fix for classification models

Improved robustness in reading input channel count

* Update torch_utils.py

* Update torch_utils.py
2022-08-13 16:38:11 +02:00
Glenn Jocher f1214f237d Remove DDP destroy_process_group() on train end (#8935)
May resolve https://github.com/ultralytics/yolov5/issues/7307
2022-08-13 03:57:31 +02:00
Glenn Jocher 2e57b8418d Deci.ai updates (#8928)
* Deci.ai updates

Content implementation for https://deci.ai/

* Update README.md

* Update README.md

* Update README.md
2022-08-11 00:01:38 +02:00
Glenn Jocher 259e055b42 Fix --source 'path/*.jpg' glob example (#8927)
* Fix `--source 'path/*.jpg'` glob example

* Update README_cn.md

* Update tutorial.ipynb
2022-08-10 21:53:12 +02:00
Glenn Jocher 4c6bdbfcde Direct download table links (#8917)
* Direct download table links

Per recent partner feedback

* Update README_cn.md

* Update README_cn.md

* Update README.md

* Update README_cn.md

* Update README.md
2022-08-10 17:01:01 +02:00
Glenn Jocher 3d47fc6c55 Update README.md 2022-08-10 13:10:30 +02:00
Glenn Jocher e242ede14c Add torch==1.7.0 to CI tests (#8916)
* Add torch==1.7.0 to CI tests

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Attempt crash on 1.7.0

* revert crash

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml

* Update ci-testing.yml
2022-08-09 23:46:30 +02:00
Glenn Jocher 7ab1f50633 Scope torchvision.utils.draw_bounding_boxes import (#8915)
* Scope `torchvision.utils.draw_bounding_boxes` import

Quick fix for https://github.com/ultralytics/yolov5/issues/8898#issuecomment-1208851874 until a more permanent fix is implemented

* Update clearml_utils.py
2022-08-09 20:29:25 +02:00
Glenn Jocher f5335f22bb Capitalize YouTube (#8903) 2022-08-08 20:36:18 +02:00
Glenn Jocher 4bb305275a Edge TPU add --search_delegate fix (#8902)
Resolves https://github.com/ultralytics/yolov5/issues/8842
2022-08-08 17:51:28 +02:00
Glenn Jocher 48a85314bc Update README.md Environments icons (#8895) 2022-08-07 14:51:30 +02:00
Glenn Jocher fcfaf0a29b Update README_cn.md 2022-08-07 14:04:06 +02:00
Glenn Jocher 8f28ea4945 emoji-safe default logging (#8888)
* emoji-safe default logging

* implement everywhere
2022-08-06 12:30:08 +02:00
Glenn Jocher 73ed7a5e7a Created using Colaboratory 2022-08-06 11:53:29 +02:00
Glenn Jocher d631a6904e Created using Colaboratory 2022-08-06 11:44:52 +02:00
Glenn Jocher 20f1b7ea08 smart_optimizer() improved reporting (#8887)
Update smart_optimizer() weight_decay reporting
2022-08-06 11:30:54 +02:00
Glenn Jocher 030586a31c Created using Colaboratory 2022-08-06 01:14:31 +02:00
Glenn Jocher 5a0faef353 Created using Colaboratory 2022-08-06 01:10:17 +02:00
Glenn Jocher 2794483e09 Update CI to default Python 3.10 (#8883)
* Update CI to default Python 3.10

* Update ci-testing.yml

* Update ci-testing.yml
2022-08-05 17:10:44 +02:00
Glenn Jocher daed7a844e --resume training from URL weights fix (#8884)
--resume training from URL weights fix

@kalenmike should fix data error on HUB resume
2022-08-05 16:42:10 +02:00
Glenn Jocher e073658e11 --resume training from URL weights (#8882)
@kalenmike
2022-08-05 16:27:28 +02:00
Glenn Jocher bc9fcb1767 HUBDatasetStats() preview images to 50 quality (#8880)
@kalenmike should represent a 30% filesize reduction vs 75 quality
2022-08-05 14:45:41 +02:00
Glenn Jocher 731a2f8c1f Single-line TRT dynamic assertion (#8871) 2022-08-04 23:34:15 +02:00
Glenn Jocher 38a6eb6e99 Fix TensorRT --dynamic excess outputs bug (#8869)
* Fix TensorRT --dynamic excess outputs bug

Potential fix for https://github.com/ultralytics/yolov5/issues/8790

* Cleanup

* Update common.py

* Update common.py

* New fix
2022-08-04 23:26:30 +02:00
Glenn Jocher 84e7748564 Update dataloaders.py remove float64 shapes (#8865)
May help https://github.com/ultralytics/yolov5/issues/8862
2022-08-04 17:06:08 +02:00
Glenn Jocherandpre-commit-ci[bot] 628c05ca6f export.py replace check_file -> check_yaml (#8852)
* export.py replace `check_file` -> `check_yaml`

* [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-08-03 23:38:36 +02:00
Glenn Jocherandpre-commit-ci[bot] 6884da3a32 Add check_file(data) i.e. --data coco128.yaml (#8851)
* Add check_file(data) i.e. `--data coco128.yaml`

* [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-08-03 23:32:31 +02:00
Glenn Jocherandpre-commit-ci[bot] a75a1105a1 Self-contained checkpoint --resume (#8839)
* Single checkpoint resume

* Update train.py

* Add hyp

* Add hyp

* Add hyp

* FIX

* avoid resume on url data

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

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

* avoid resume on url data

* avoid resume on url data

* Update

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-03 21:28:22 +02:00
Glenn Jocher 4d8d84b0ea Created using Colaboratory 2022-08-03 12:49:02 +02:00
Glenn Jocher e5991c9867 Created using Colaboratory 2022-08-03 00:57:40 +02:00
Glenn Jocher 08c8c3e00a New smart_resume() (#8838)
* New `smart_resume()`

* Update torch_utils.py

* Update torch_utils.py

* Update torch_utils.py

* fix
2022-08-02 15:13:58 +02:00
Glenn Jocher ba140e5685 RUN git clone instead of COPY to /usr/src/app (#8827)
Update
2022-08-01 21:45:31 +02:00
Glenn Jocher f3c78a387e Remove hook torch.nan_to_num(x) (#8826)
* Remove hook `torch.nan_to_num(x)`

Observed erratic training behavior (green line) with the nan_to_num hook in classifier branch. I'm going to remove it from master.

* Update train.py
2022-08-01 21:39:04 +02:00
Glenn Jocherandpre-commit-ci[bot] 59578f2782 Update README.md with contributors.png (#8820)
* Update README.md with contributors.png

Replace dynamic svg from opencollective with static png for improved stability and lighter size (400kB vs 2MB).

@AyushExel

* Update README.md

* Update README.md

* Update README_cn.md

* [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-08-01 20:28:24 +02:00
Glenn Jocher 7b9cc3205a Add .git to .dockerignore (#8815) 2022-08-01 03:33:28 +02:00
Glenn Jocher 39ce8ca19a Remove assert not is_docker() from GitHub checks (#8813)
* Update

* Update
2022-08-01 03:01:44 +02:00
Glenn Jocherandpre-commit-ci[bot] 6717519635 Update .pre-commit-config.yaml (#8812)
* Update .pre-commit-config.yaml

Comment EOF fixer

* [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-08-01 02:09:36 +02:00
Glenn Jocher 0e165c50f7 Created using Colaboratory 2022-07-31 20:34:03 +02:00
Glenn Jocher 685332ede4 Update .pre-commit-config.yaml (#8811) 2022-07-31 20:06:35 +02:00
Glenn Jocher 40c41e4269 Fix Colab-update pre-commit EOF bug (#8810) 2022-07-31 19:57:40 +02:00
Glenn Jocher 7b72d9a607 Update ci-testing.yml streamlined tests (#8809)
* Update ci-testing.yml streamlined tests

* Update ci-testing.yml

* Update ci-testing.yml
2022-07-31 16:12:32 +02:00
Glenn Jocher 555976b346 FROM nvcr.io/nvidia/pytorch:22.07-py3 (#8808) 2022-07-31 15:23:57 +02:00
Glenn Jocher 9559601b9a Update requirements.txt with tf-cpu and tf-aarch64 (#8807) 2022-07-31 14:54:55 +02:00
Glenn Jocher 34cb277dc5 Fix val.py 'no labels found bug' (#8806)
Resolves https://github.com/ultralytics/yolov5/issues/8791

Bug first introduced in #8782
2022-07-31 14:17:23 +02:00
Glenn Jocher 59595c1365 Update train.py for val.run(half=amp) (#8804)
Disable FP16 validation if AMP checks fail or amp=False.
2022-07-31 04:17:39 +02:00
Glenn Jocher 1e89807d9a Detect.inplace=False for multithread-safe inference (#8801)
Detect.inplace=False for safe multithread inference
2022-07-30 22:19:40 +02:00
Glenn Jocher 7921351b4e Deprecate torch 1.6.0 compat _non_persistent_buffers_set (#8797)
Deprecate torch 1.6.0 compat _non_persistent_buffers_set
2022-07-30 21:25:16 +02:00
Glenn Jocherandpre-commit-ci[bot] ec4de43a8a Update Colab Notebook CI (#8798)
* Update Colab Notebook CI

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

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

* Created using Colaboratory

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

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

* Update tutorial.ipynb

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-07-30 21:11:19 +02:00
Glenn Jocher 56f5cb5a28 Print hubconf.py args (#8800) 2022-07-30 21:02:26 +02:00
Glenn Jocher 9111246208 Add hubconf.py argparser (#8799)
* Add hubconf.py argparser

* Add hubconf.py argparser
2022-07-30 21:00:28 +02:00
Glenn Jocher e34ae8837b ci-testing.yml Windows-friendly ENV variables (#8794)
Per https://docs.github.com/en/actions/using-workflows/workflow-syntax-for-github-actions
2022-07-30 14:04:03 +02:00
Glenn Jocher 52d3a9aee1 Fix detect.py --update list bug (#8783)
Fix detect.py --update

Resolves https://github.com/ultralytics/yolov5/issues/8776
2022-07-29 17:07:24 +02:00
Glenn Jocher e309a85586 Add val.py no label warning (#8782)
Help resolve confusion around zero-metrics val.py results when no labels are found in https://github.com/ultralytics/yolov5/issues/8753
2022-07-29 14:45:29 +02:00
Glenn Jocher 3e858633b2 Revert protobuf<=3.20.1 (#8742)
Resolve #8012 (again)
2022-07-27 17:27:44 +02:00
Glenn Jocher c775a296a7 Revert "Fix BGR->RGB Bug in albumentations #8641" (#8727)
Revert "Fix BGR->RGB Bug in albumentations #8641 (#8695)"

This reverts commit 2e1291fdce.
2022-07-26 19:00:48 +02:00
Glenn Jocher d5116bbe9c coremltools>=5.2 for CoreML export (#8725) 2022-07-26 17:50:49 +02:00
Glenn Jocherandpre-commit-ci[bot] b367860196 New HUBDatasetStats() class (#8716)
* New `HUBDatasetStats()` class

Usage examples:
```
from utils.dataloaders import *

stats = HUBDatasetStats('coco128.yaml', autodownload=True)  # method 1
stats = HUBDatasetStats('path/to/coco128_with_yaml.zip')  # method 1

stats.get_json(save=False)
stats.process_images()
```

@kalenmike

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

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

* Update dataloaders.py

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

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

* Update dataloaders.py

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

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

* Update dataloaders.py

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

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

* Update dataloaders.py

* [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-07-25 18:20:01 +02:00
Glenn Jocher a6f197ae79 Update dataloaders.py (#8714)
* Update dataloaders.py

* Update dataloaders.py
2022-07-25 16:52:28 +02:00
Glenn Jocher 7215a0fb41 Avoid cv2 window init code on Windows (#8712)
Resolves https://github.com/ultralytics/yolov5/issues/8642
2022-07-25 13:57:05 +02:00
Glenn Jocher 0ab303b044 Remove else: from load_image() (#8692) 2022-07-25 00:02:09 +02:00
Glenn Jocher 916bdb1d61 Faster crop saving (#8696)
Faster crops

Following https://github.com/ultralytics/yolov5/issues/8641#issuecomment-1193190325
2022-07-23 23:30:30 +02:00
Glenn Jocher b510957650 Move .dockerignore to root (#8690) 2022-07-23 18:50:19 +02:00
Glenn Jocher 7f7bd6fbcd Set torch.cuda.manual_seed_all() for DDP (#8688)
* Set `torch.cuda.manual_seed_all()` for DDP

* Update general.py

* Update general.py
2022-07-23 18:24:14 +02:00
Glenn Jocher b92430a83b Update hubconf.py to reset logging level to INFO (#8680) 2022-07-22 19:01:16 +02:00
Glenn Jocher b17629e54f Update warning emojis (#8678) 2022-07-22 15:23:22 +02:00
Glenn Jocher 38721de7ef Update hubconf.py to reset LOGGER.level after load (#8674)
Resolves silent outputs after model load
2022-07-22 11:54:31 +02:00
Glenn Jocher 4c1784bd15 Use contextlib's suppress method to silence an error (#8668) 2022-07-21 23:12:49 +02:00
Glenn Jocher fbe67e4653 Fix OMP_NUM_THREADS=1 for macOS (#8624)
Resolves https://github.com/ultralytics/yolov5/issues/8623
2022-07-18 17:53:30 +02:00
Glenn Jocherandpre-commit-ci[bot] 9cf5fd5ac3 assert torch!=1.12.0 for DDP training (#8621)
* assert torch!=1.12.0 for DDP training

* [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-07-18 15:05:58 +02:00
Glenn Jocherandpre-commit-ci[bot] 51fb467b63 Refactor optimizer initialization (#8607)
* Refactor optimizer initialization

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

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

* Update train.py

* Update train.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-07-17 11:43:52 +02:00
Glenn Jocher 24305787ae FROM nvcr.io/nvidia/pytorch:22.06-py3 (#8600) 2022-07-17 00:55:55 +02:00
Glenn Jocher a34b376d0f Link fuse() to AutoShape() for Hub models (#8599) 2022-07-16 23:46:23 +02:00
Glenn Jocher 6e86af3de8 Add pip install wheel to avoid legacy setup.py install (#8597)
* Update ci-testing with `pip install wheel`

* Update ci-testing.yml

* Update dockerfiles
2022-07-16 16:41:43 +02:00
Glenn Jocher 5c45a4b13d Update tqdm>=4.64.0 and thop>=0.1.1 (#8596)
* Update `tqdm>=4.64.0` and `thop>=0.1.1`

* Update requirements.txt
2022-07-16 16:14:16 +02:00
Glenn Jocher cf28dda366 Update scipy>=1.7.3 (#8595) 2022-07-16 15:54:34 +02:00
Glenn Jocher 72a81e7a1c Update requirements.txt comment spacing (#8562) 2022-07-15 16:01:29 +02:00
Glenn Jocher 574ceedfc5 Add thop>=0.1.0 (#8558) 2022-07-12 14:49:54 +02:00
Glenn Jocher 04146371b9 Update metrics.py with IoU protected divides (#8550) 2022-07-12 01:58:25 +02:00
Glenn Jocher a84cd02387 CIoU protected divides (#8546)
Protected divides in IOU function to resolve https://github.com/ultralytics/yolov5/issues/8539
2022-07-11 16:07:11 +02:00
Glenn Jocher 7dafd1cb29 val.py assert ncm == nc fix (#8545) 2022-07-11 15:09:42 +02:00
Glenn Jocher c215e87393 XML export --half fix (#8522)
Improved error reporting for https://github.com/ultralytics/yolov5/issues/8519
2022-07-08 13:49:20 +02:00
Glenn Jocher 63ba0cb18a Add --half arguments to export.py Usage examples (#8516) 2022-07-08 00:46:56 +02:00
Glenn Jocher f17444abcd Simplify benchmarks.py assertions (#8515) 2022-07-07 23:46:55 +02:00
Glenn Jocherandpre-commit-ci[bot] 3e54651fca Add --hard-fail argument to benchmarks for CI errors (#8513)
* Add `--hard-fail` list argument to benchmarks for CI

Will cause CI to fail on a benchmark failure for given indices.

* Update ci-testing.yml

* Attempt Failure (CI should fail)

* Update benchmarks.py

* Update export.py

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

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

* Update benchmarks.py

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

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

* Update ci-testing.yml

* Update benchmarks.py

* Update benchmarks.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-07-07 23:41:34 +02:00
Glenn Jocher dd28df98c2 Avoid FP64 ops for MPS support in train.py (#8511)
Avoid FP64 ops for MPS support

Resolves https://github.com/ultralytics/yolov5/pull/7878#issuecomment-1177952614
2022-07-07 20:36:23 +02:00
Glenn Jocher 9d7bc06ae7 Revert "Expose OpenVINO batch_size similarly to TensorRT" (#8510)
Revert "Expose OpenVINO `batch_size` similarly to TensorRT (#8437)"

This reverts commit da2ee3934e.
2022-07-07 20:13:42 +02:00
Glenn Jocher 36f64a981d Update tutorial.ipynb (#8507) 2022-07-07 12:34:01 +02:00
Glenn Jocher c7689198bc Update stale.yml (#8465) 2022-07-04 15:01:11 +02:00
Glenn Jocher 29d79a6360 Do not prefer Apple MPS (#8446)
Require explicit request for MPS, i.e.
```bash
python detect.py --device mps
```

Reverts https://github.com/ultralytics/yolov5/pull/8210 for preferring MPS if available. 

Note that torch MPS is experiencing ongoing compatibility issues in https://github.com/pytorch/pytorch/issues/77886
2022-07-02 18:35:45 +02:00
Glenn Jocher d94b4705a6 TRT --half fix autocast images to FP16 (#8435)
* TRT `--half` fix autocast images to FP16

Resolves bug raised in https://github.com/ultralytics/yolov5/issues/7822

* Update common.py
2022-07-01 15:41:14 +02:00
Glenn Jocher 898332433a Update cache comments (#8414)
* Update cache comments

For better readability

* Update dataloaders.py
2022-06-30 16:19:22 +02:00
Glenn Jocher f76a78e707 Update tutorial.ipynb (#8380) 2022-06-28 17:34:24 +02:00
Glenn Jocher 0c1324067c Fix ONNX --dynamic export on GPU (#8378)
* Fix ONNX `--dynamic` export on GPU

Patch forces --dynamic export model and image to CPU. Resolves bug raised in https://github.com/ultralytics/yolov5/issues/8377

* Update export.py
2022-06-28 15:22:15 +02:00
Glenn Jocherandpre-commit-ci[bot] 34df5032a7 Add File Size (MB) column to benchmarks (#8359)
* Add filesize to benchmarks.py

* Add filesize to benchmarks.py

* [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-06-27 17:46:49 +02:00
Glenn Jocher 8ebf569d14 Fix bias warmup LR init (#8356)
Per https://github.com/ultralytics/yolov5/issues/8352
2022-06-27 16:11:24 +02:00
Glenn Jocherandpre-commit-ci[bot] 5c990e3674 Create README_cn.md (#8344)
* Create README_cn.md

Add mandarin version of README

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

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

* Update README.md

* fix link

* fix english link

* remove line

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

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

* Update .pre-commit-config.yaml

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

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

* Update README_cn.md

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

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

* Update .pre-commit-config.yaml

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

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

* Update README_cn.md

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

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

* Update .pre-commit-config.yaml

* Update README.md

* Update README_cn.md

* Kiera fix

* Update README_cn.md

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-06-26 17:46:39 +02:00
Glenn Jocher 724d5b21b7 device_name="MYRIAD" for Intel NCS2 comment (#8327)
Display device_name="MYRIAD" for Intel NCS2 option per user question in https://github.com/ultralytics/yolov5/issues/8154
2022-06-24 14:18:58 +02:00