acc58c1dcf
Fix TRT `max_workspace_size` deprecation notice ( #6856 )
...
* Fix TRT `max_workspace_size` deprecation notice
* Update export.py
* Update export.py
2022-03-07 13:52:53 +01:00
a5a1760ea6
Bump actions/checkout from 2 to 3 ( #6881 )
...
Bumps [actions/checkout](https://github.com/actions/checkout ) from 2 to 3.
- [Release notes](https://github.com/actions/checkout/releases )
- [Changelog](https://github.com/actions/checkout/blob/main/CHANGELOG.md )
- [Commits](https://github.com/actions/checkout/compare/v2...v3 )
---
updated-dependencies:
- dependency-name: actions/checkout
dependency-type: direct:production
update-type: version-update:semver-major
...
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
2022-03-07 13:49:27 +01:00
c8a589920e
Bump actions/setup-python from 2 to 3 ( #6880 )
...
Bumps [actions/setup-python](https://github.com/actions/setup-python ) from 2 to 3.
- [Release notes](https://github.com/actions/setup-python/releases )
- [Commits](https://github.com/actions/setup-python/compare/v2...v3 )
---
updated-dependencies:
- dependency-name: actions/setup-python
dependency-type: direct:production
update-type: version-update:semver-major
...
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
2022-03-07 13:48:59 +01:00
596de6d5a0
Default FP16 TensorRT export ( #6798 )
...
* Assert engine precision #6777
* Default to FP32 inputs for TensorRT engines
* Default to FP16 TensorRT exports #6777
* Remove wrong line #6777
* Automatically adjust detect.py input precision #6777
* Automatically adjust val.py input precision #6777
* Add missing colon
* Cleanup
* Cleanup
* Remove default trt_fp16_input definition
* Experiment
* Reorder detect.py if statement to after half checks
* Update common.py
* Update export.py
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-03-06 21:21:16 +01:00
7e98b4801a
Update yolov5s.yaml ( #6865 )
...
* Update yolov5s.yaml
* Update yolov5s.yaml
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-03-06 20:20:01 +01:00
4728840745
Update `--cache disk` deprecate `*_npy/` dirs ( #6876 )
...
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Cleanup
* Cleanup
2022-03-06 16:16:17 +01:00
8a66ebad44
Add `*.ts` to `VID_FORMATS` ( #6859 )
2022-03-04 14:10:13 +01:00
601dbb83f0
AutoAnchor improved initialization robustness ( #6854 )
...
* Update AutoAnchor
* Update AutoAnchor
* [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-03-04 10:32:18 +01:00
bcc92e2169
Update sweep.yaml ( #6825 )
...
* Update sweep.yaml
Changed focal loss gamma search range between 1 and 4
* Update sweep.yaml
lowered the min value to match default
2022-03-04 09:39:23 +01:00
63ddb6f0d0
Update autoanchor.py ( #6794 )
...
* Update autoanchor.py
* Update autoanchor.py
2022-02-26 19:15:12 +01:00
c2403eb04c
Update PULL_REQUEST_TEMPLATE.md ( #6783 )
2022-02-25 14:01:58 +01:00
9ec51a6396
YOLOv5s6 params FLOPs fix ( #6782 )
2022-02-25 13:59:21 +01:00
0f819919ad
Update EMA decay `tau` ( #6769 )
...
* Update EMA
* Update EMA
* ratio invert
* fix ratio invert
* fix2 ratio invert
* warmup iterations to 100
* ema_k
* implement tau
* implement tau
2022-02-25 12:33:09 +01:00
b2adc7c39a
Fix export for 1-channel images ( #6780 )
...
Export failed for 1-channel input shape, 1-liner fix
2022-02-25 10:56:37 +01:00
cea994b3f6
Update hyp.VOC.yaml ( #6772 )
2022-02-25 00:14:17 +01:00
741cd0eb23
Update tutorial.ipynb ( #6771 )
2022-02-25 00:12:27 +01:00
c161557563
Default `OMP_NUM_THREADS=8` ( #6770 )
2022-02-25 00:09:14 +01:00
47f265dde2
Update min warmup iterations from 1k to 100 ( #6768 )
2022-02-24 23:37:25 +01:00
4bab56526b
Update tutorial.ipynb (2 CPUs, 12.7 GB RAM, 42.2/166.8 GB disk) ( #6767 )
2022-02-24 18:29:59 +01:00
66aaf5184d
Pre-commit table fix ( #6744 )
2022-02-22 19:35:31 +01:00
3752807c0b
YOLOv5 v6.1 release ( #6739 )
2022-02-22 12:35:24 +01:00
2692e67c5f
Updated VOC hyperparameters ( #6732 )
...
* Update hyps
* Update hyp.VOC.yaml
* Update pathlib
* Update hyps
* Update hyps
* Update hyps
* Update hyps
2022-02-22 12:11:18 +01:00
a936f5f219
Switch default LR scheduler from cos to linear ( #6729 )
...
* Switch default LR scheduler from cos to linear
Based on empirical results of training both ways on all YOLOv5 models.
* linear bug fix
2022-02-21 14:23:27 +01:00
dbbb57cf0b
GH: add PR template ( #6482 )
...
* GH: add PR template
* Update CONTRIBUTING.md
* Update PULL_REQUEST_TEMPLATE.md
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update PULL_REQUEST_TEMPLATE.md
* Update PULL_REQUEST_TEMPLATE.md
* Update PULL_REQUEST_TEMPLATE.md
* Update PULL_REQUEST_TEMPLATE.md
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-02-21 12:36:43 +01:00
c43f135571
Update `nw` to `max(nd, 1)` ( #6714 )
2022-02-20 21:47:46 +01:00
4de8b24881
Suppress `torch` AMP-CPU warnings ( #6706 )
...
This is a torch bug, but they seem unable or unwilling to fix it so I'm creating a suppression in YOLOv5.
Resolves https://github.com/ultralytics/yolov5/issues/6692
2022-02-19 16:48:33 +01:00
de9c25b35e
Use `export_formats()` in export.py ( #6705 )
...
* Use `export_formats()` in export.py
* list fix
2022-02-19 16:08:33 +01:00
a297efc383
Edge TPU inference fix ( #6686 )
...
* refactor: use edgetpu flag
* fix: remove bitwise and assignation to tflite
* Cleanup and fix tflite
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-02-19 15:10:07 +01:00
0365379016
Fix floating point in number of workers `nw` ( #6701 )
...
Integer division by a float yields a (rounded) float. This causes
the dataloader to crash when creating a range.
2022-02-18 21:06:21 +01:00
56697c43ff
Fix CoreML P6 inference ( #6700 )
...
* Fix CoreML P6 inference
* [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-02-18 20:38:33 +01:00
3670d91b05
Fix `--evolve --bucket gs://...` ( #6698 )
2022-02-18 19:15:37 +01:00
66e5d794c7
Fix TF exports >= 2GB ( #6292 )
...
* Fix exporting saved_model: pb exceeds 2GB
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Replace TF v1.x API with TF v2.x API for saved_model export
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Clean up
* Remove lambda in tf.function()
* Revert "Remove lambda in tf.function()" to be compatible with TF v2.4
This reverts commit 46c7931f11dfdea6ae340c77287c35c30b9e0779.
* Fix for pre-commit.ci
* Cleanup1
* Cleanup2
* Backwards compatibility update
* Update common.py
* Update common.py
* Cleanup3
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-02-18 16:36:45 +01:00
07221f1591
FROM nvcr.io/nvidia/pytorch:21.10-py3 ( #6677 )
...
Revert to 21.10 on autobuild fail
2022-02-17 16:05:30 +01:00
18b5081166
FROM nvcr.io/nvidia/pytorch:21.10-py3 ( #6673 )
...
Reordered installation may help reduce resource usage in autobuild
2022-02-17 15:42:47 +01:00
25456284b4
Update Dockerfile reorder installs ( #6672 )
...
Also `nvidia-tensorboard-plugin-dlprof`, `nvidia-tensorboard` are no longer installed in NVCR base.
2022-02-17 15:10:19 +01:00
09d263b376
FROM nvcr.io/nvidia/pytorch:21.10-py3 ( #6671 )
...
22.10 returns 'no space left on device' error message.
Seems like a bug at docker. Raised issue in https://github.com/docker/hub-feedback/issues/2209
2022-02-17 14:51:53 +01:00
bb054d5f55
FROM nvcr.io/nvidia/pytorch:22.01-py3 ( #6670 )
2022-02-17 14:20:42 +01:00
9a8ebe671c
Update Dockerfile `torch==1.10.2+cu113` ( #6669 )
2022-02-17 13:24:44 +01:00
2e5c67e537
Robust `scipy.cluster.vq.kmeans` too few points ( #6668 )
...
* Handle `scipy.cluster.vq.kmeans` too few points
Resolves #6664
* Update autoanchor.py
* Cleanup
2022-02-17 12:55:03 +01:00
7b80545e8e
Add `--cache val` ( #6663 )
...
New `--cache val` argument will cache validation set only into RAM. Should help multi-GPU training speeds without consuming as much RAM as full `--cache ram`.
2022-02-17 00:47:42 +01:00
1ff43702a8
Validate with 2x `--workers` single-GPU/CPU fix ( #6659 )
...
Fix for #6658 for single-GPU and CPU training use cases
2022-02-16 14:50:37 +01:00
6769021901
Validate with 2x `--workers` ( #6658 )
2022-02-16 14:44:44 +01:00
ee6c70ae43
Fix yolov3.yaml remove list ( #6655 )
...
Per https://github.com/ultralytics/yolov3/issues/1887#issuecomment-1041135181
2022-02-16 10:13:05 +01:00
ca0a00784d
Fixed wandb logger KeyError ( #6637 )
2022-02-14 17:08:36 +01:00
538930023c
Fix ConfusionMatrix scale `vmin=0.0` ( #6638 )
...
Fix attempt for https://github.com/ultralytics/yolov5/issues/6626
2022-02-14 14:54:55 +01:00
a45e472358
YOLOv5 Export Benchmarks ( #6613 )
...
* Add benchmarks.py
* Update
* Add requirements
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* dataset autodownload from root
* Update
* Redirect to /dev/null
* sudo --help
* Cleanup
* Add exports pd df
* Updates
* Updates
* Updates
* Cleanup
* dir handling fix
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
* Cleanup2
* Cleanup3
* Cleanup model_type
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-02-12 16:05:43 +01:00
96d8f86085
W&B: don't log media in evolve ( #6617 )
2022-02-12 13:04:50 +01:00
51f7750aad
W&B: Improve resume stability ( #6611 )
...
* log best.pt metrics at train end
* update
* Update __init__.py
* flush callbacks when using evolve
* remember batch size on resuming
* Update train.py
* improve stability of resume
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-02-12 13:03:09 +01:00
3eef80457f
Add YOLOv5n to Reproduce section ( #6619 )
2022-02-11 21:48:01 +01:00
b40bdd6b2d
YouTube dependency fix `youtube_dl==2020.12.2` ( #6612 )
...
Per https://github.com/ultralytics/yolov5/issues/5860#issuecomment-1035320018 by @hdnh2006
2022-02-11 13:46:20 +01:00
Glenn Jocher
dependabot[bot]
DavidB
vnekat
Louis Combaldieu
Louis Combaldieu
Jirka Borovec
Raffaele Galliera
Samuel Yvon
Jiacong Fang
imyhxy
Ayush Chaurasia