mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
2,141 Commits 10 Branches 10 Tags
Commit Graph

2141 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher
b94b59e199
DetectMultiBackend() --half handling (#6945) 
* DetectMultiBackend() `--half` handling

* CI fixes

* rename .half to .fp16 to avoid conflict

* warmup fix

* val update

* engine update

* engine update
 2022-03-11 16:31:52 +01:00
Glenn Jocher
84efa62b2d
Fix PyTorch Hub export inference shapes (#6949) 
May resolve https://github.com/ultralytics/yolov5/issues/6947
 2022-03-11 16:18:40 +01:00
Glenn Jocher
caf7ad0500
Allow 3-point segments (#6938) 
May resolve https://github.com/ultralytics/yolov5/issues/6931
 2022-03-10 18:41:47 +01:00
Glenn Jocher
055e72af5b
Optimize PyTorch 1.11.0 compatibility update (#6933) 2022-03-10 12:58:41 +01:00
Glenn Jocher
d3d9cbce22
PyTorch 1.11.0 compatibility updates (#6932) 
Resolves `AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'` first raised in https://github.com/ultralytics/yolov5/issues/5499
 2022-03-10 12:41:06 +01:00
Glenn Jocher
6dd82c0252
Move git_describe() to general.py (#6918) 
* Move `git_describe()` to general.py

* Move `git_describe()` to general.py
 2022-03-09 18:22:53 +01:00
Glenn Jocher
e6e36aac10
Update bytes to GB with bitshift (#6886) 2022-03-07 19:26:37 +01:00
Glenn Jocher
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
dependabot[bot]
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
dependabot[bot]
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
DavidB
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
vnekat
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
Glenn Jocher
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
Glenn Jocher
8a66ebad44
Add *.ts to VID_FORMATS (#6859) 2022-03-04 14:10:13 +01:00
Glenn Jocher
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
Louis Combaldieu
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
Glenn Jocher
63ddb6f0d0
Update autoanchor.py (#6794) 
* Update autoanchor.py

* Update autoanchor.py
 2022-02-26 19:15:12 +01:00
Glenn Jocher
c2403eb04c
Update PULL_REQUEST_TEMPLATE.md (#6783) 2022-02-25 14:01:58 +01:00
Glenn Jocher
9ec51a6396
YOLOv5s6 params FLOPs fix (#6782) 2022-02-25 13:59:21 +01:00
Glenn Jocher
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
Louis Combaldieu
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
Glenn Jocher
cea994b3f6
Update hyp.VOC.yaml (#6772) 2022-02-25 00:14:17 +01:00
Glenn Jocher
741cd0eb23
Update tutorial.ipynb (#6771) 2022-02-25 00:12:27 +01:00
Glenn Jocher
c161557563
Default OMP_NUM_THREADS=8 (#6770) 2022-02-25 00:09:14 +01:00
Glenn Jocher
47f265dde2
Update min warmup iterations from 1k to 100 (#6768) 2022-02-24 23:37:25 +01:00
Glenn Jocher
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
Glenn Jocher
66aaf5184d
Pre-commit table fix (#6744) 2022-02-22 19:35:31 +01:00
Glenn Jocher
3752807c0b
YOLOv5 v6.1 release (#6739) v6.1 2022-02-22 12:35:24 +01:00
Glenn Jocher
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
Glenn Jocher
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
Jirka Borovec
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
Glenn Jocher
c43f135571
Update nw to max(nd, 1) (#6714) 2022-02-20 21:47:46 +01:00
Glenn Jocher
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
Glenn Jocher
de9c25b35e
Use export_formats() in export.py (#6705) 
* Use `export_formats()` in export.py

* list fix
 2022-02-19 16:08:33 +01:00
Raffaele Galliera
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
Samuel Yvon
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
Glenn Jocher
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
Glenn Jocher
3670d91b05
Fix --evolve --bucket gs://... (#6698) 2022-02-18 19:15:37 +01:00
Jiacong Fang
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
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
bb054d5f55
FROM nvcr.io/nvidia/pytorch:22.01-py3 (#6670) 2022-02-17 14:20:42 +01:00
Glenn Jocher
9a8ebe671c
Update Dockerfile torch==1.10.2+cu113 (#6669) 2022-02-17 13:24:44 +01:00
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
6769021901
Validate with 2x --workers (#6658) 2022-02-16 14:44:44 +01:00
Glenn Jocher
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