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,234 Commits 10 Branches 10 Tags
Commit Graph

2234 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
David Matos
bff6e51bc4
Disallow --dynamic when --half is passed (#7669) 
* disallow dynamic arg when half is given

* Update export.py

Co-authored-by: David Matos <david@track32.nl>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-05-02 09:00:24 -07:00
dependabot[bot]
1994ad93e4
Bump github/codeql-action from 1 to 2 (#7665) 
Bumps [github/codeql-action](https://github.com/github/codeql-action) from 1 to 2.
- [Release notes](https://github.com/github/codeql-action/releases)
- [Changelog](https://github.com/github/codeql-action/blob/main/CHANGELOG.md)
- [Commits](https://github.com/github/codeql-action/compare/v1...v2)

---
updated-dependencies:
- dependency-name: github/codeql-action
  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-05-01 22:25:00 -07:00
SylvanDing
5743deb9df
Update plot.py (#7654) 
fix bug in issue #7650
 2022-04-30 20:01:21 -07:00
Glenn Jocher
488fb0a36a
Bump scipy>=1.5 to meet numpy constraints (#7646) 
Resolves https://github.com/ultralytics/yolov5/issues/7638
 2022-04-29 12:00:30 -07:00
Glenn Jocher
b24a576823
Reduce opencv-python>=4.1.1 for Jetson Nano (#7645) 
May help https://github.com/ultralytics/yolov5/issues/7631#issuecomment-1113007451
 2022-04-29 11:50:30 -07:00
Glenn Jocher
1a3ecb8b38
increment_path() robustness improvements (#7628) 
Improved robustness to filename edge cases like `data/images/zidane.23.jpg` that currently fail. May resolve https://github.com/ultralytics/yolov5/issues/7432
 2022-04-28 14:00:43 -07:00
Duncan Moss
177da7f348
Add --half support for OpenVINO exports (#7615) 
* feature(export): add half support for openvino models

* Update export.py

* Update export.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-27 17:29:59 -07:00
Glenn Jocher
b53917de8d
Remove tqdm.auto (#7599) 2022-04-26 15:00:01 -07:00
Jack Liu
2611477d2e
Fix from yolov5 import utils statement (#7578) 2022-04-25 21:05:48 -07:00
Glenn Jocher
e54e758cfb
Add PyTorch-only benchmark arg (#7564) 
Usage:
```
python utils/benchmarks.py --weights yolov5s.pt --img 640 --pt-only
```
 2022-04-24 20:03:23 -07:00
Glenn Jocher
950a85d9f6
TensorRT PyTorch Hub inference fix (#7560) 
Solution proposed in https://github.com/ultralytics/yolov5/issues/7128 to TRT PyTorch Hub CUDA illegal memory errors.
 2022-04-24 12:45:56 -07:00
Glenn Jocher
c16671fc74
Add print(results) override for PyTorch Hub results (#7559) 2022-04-24 12:07:05 -07:00
Glenn Jocher
be67572279
Enable results.print() when _verbose=False (#7558) 
Follows implementation of _verbose flag for PyTorch Hub models. Currently these are so silent that result.print() does nothing 😂
 2022-04-24 11:30:40 -07:00
Glenn Jocher
7043872f25
PyTorch Hub _verbose=False fix2 (#7550) 
* PyTorch Hub `_verbose=False` fix2

* Update downloads.py

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

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

* Update hubconf.py

* [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

* Update

* Update

* Update

* Update

* Update

* Update

* Update

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-04-23 11:26:25 -07:00
Glenn Jocher
1f1ec1c3e7
Restrict TRT autoinstall to Linux-only (#7549) 
May partially resolve concerns in https://github.com/ultralytics/yolov5/pull/7537#discussion_r856843711
 2022-04-23 09:36:36 -07:00
Glenn Jocher
404b4fefbe
Update restapi.py (#7309) 
* Update restapi.py

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

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

* Update restapi.py

* Update restapi.py

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

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

* Update restapi.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-04-22 16:19:21 -07:00
Glenn Jocher
3a89377c04
Improve availability of YOLOv5 in Russia (#7545) 
* Add redundant weights download mirrors

This PR seeks to improve access worldwide to YOLOv5 weights. Universal access to AI for all is our core value, and we are against any censorship or restriction efforts.

I've uploaded the official YOLOv5 v6.1 weights to a primary backup bucket on GCP and to secondary backup on Google Drive at https://drive.google.com/drive/folders/1EFQTEUeXWSFww0luse2jB9M1QNZQGwNl. Autodownload with try the first two locations (GitHuB release assets and GCP bucket), and point users to the Google Drive folder if the first two fail.

* [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-04-22 16:07:33 -07:00
Ayush Chaurasia
4cac1cfc8f
Add --noplots flag to suppress figures and images logging (#7534) 
* support nomedia

* support nomedia for validation

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

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

* Update train.py

* Revert no plot evolve

evolve plots do not contain any images

* Revert plot_results

contains no media

* Update wandb_utils.py

* sync-bn cleanup

* Cleanup

* Rename nomedia -> noplots

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-04-22 15:04:08 -07:00
Glenn Jocher
f62609eb52
Update check_requirements() with cmds=() argument (#7543) 2022-04-22 14:31:05 -07:00
Powercube7
4b284a12c7
Update downloads.py current release (#7541) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-22 13:44:54 -07:00
Glenn Jocher
e04d8b1cb0
Update precommit monthly python 3.7+ (#7542) 2022-04-22 13:43:50 -07:00
Zengyf-CVer
c264795f50
Add mdformat to precommit checks and update other version (#7529) 
* Update .pre-commit-config.yaml

* [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 CONTRIBUTING.md

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

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

* Update README.md

* Update README.md

* Update README.md

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-04-22 13:36:27 -07:00
Glenn Jocher
cc1d7df03c
Autoinstall TensorRT if missing (#7537) 
* Autoinstall TensorRT if missing

May resolve https://github.com/ultralytics/yolov5/issues/7464

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

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

* Update export.py

* Update export.py

* Update export.py

* Update export.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-04-22 12:31:33 -07:00
Glenn Jocher
813eba85b2
Empty val batch CUDA device fix (#7539) 
Verified fix for https://github.com/ultralytics/yolov5/pull/7525#issuecomment-1106081123
 2022-04-22 12:01:14 -07:00
Glenn Jocher
b804b36bc4
Add Docker --file argument to build (#7527) 2022-04-21 20:31:26 -07:00
Glenn Jocher
d2e698c75c
Reduce val device transfers (#7525) 2022-04-21 20:06:57 -07:00
Glenn Jocher
23718df1c6
Fix val plots=plots (#7524) 2022-04-21 18:21:01 -07:00
Zengyf-CVer
6ea81bb3a9
Add yesqa to precommit checks (#7511) 
* Update .pre-commit-config.yaml

* Update .pre-commit-config.yaml
 2022-04-20 18:44:52 -07:00
Glenn Jocher
918d7b2b3f
Refactor Dockerfiles to utils/docker (#7510) 
* Refactor Docker files

* Refactor Docker files

* Update Dockerfile
 2022-04-20 14:23:55 -07:00
Joseph Kocherhans
b77c8d9d72
Added YOLOv5_AUTOINSTALL environment variable (#7505) 
* Added a way to skip dependency auto-installation.

Setting the environment variable `YOLOv5_AUTOINSTALL=False` will
skip installing any missing dependencies as if the user had passed
`install=False` to `check_requirements`.

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-20 12:08:22 -07:00
Glenn Jocher
3f3852e2ff
Fix val.py Ensemble() (#7490) 2022-04-19 21:15:04 -07:00
Glenn Jocher
ab5b917494
check_fonts() download to CONFIG_DIR fix (#7489) 
Follows https://github.com/ultralytics/yolov5/pull/7488. Correct bug where fonts were downloading to current working directory rather than global CONFIG_DIR
 2022-04-19 17:50:02 -07:00
Glenn Jocher
c9042dc2ad
Improved non-latin Annotator() plotting (#7488) 
* Improved non-latin labels Annotator plotting

May resolve https://github.com/ultralytics/yolov5/issues/7460

* Update train.py

* Update train.py

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

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

* add progress arg

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-04-19 17:32:15 -07:00
HERIUN
d876caab4d
Update val.py (#7478) 
* Update val.py

is_coco doesn't work!! '/' -> os.sep!!

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

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

* Cleanup

* fix

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-04-19 15:40:06 -07:00
dependabot[bot]
3a25e81b30
Bump cirrus-actions/rebase from 1.5 to 1.6 (#7462) 
Bumps [cirrus-actions/rebase](https://github.com/cirrus-actions/rebase) from 1.5 to 1.6.
- [Release notes](https://github.com/cirrus-actions/rebase/releases)
- [Commits](https://github.com/cirrus-actions/rebase/compare/1.5...1.6)

---
updated-dependencies:
- dependency-name: cirrus-actions/rebase
  dependency-type: direct:production
  update-type: version-update:semver-minor
...

Signed-off-by: dependabot[bot] <support@github.com>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2022-04-19 15:07:05 -07:00
Cedric Perauer
7926afccde
Add --half support for FP16 CoreML exports with (#7446) 
* add fp16 for coreml using --half

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

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

* fix

* Cleanup

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

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

* Update export.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>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-16 18:00:50 +02:00
Glenn Jocher
c9a3b14a74
Disable pbar for DDP ranks > 0 (#7440) 2022-04-16 15:12:38 +02:00
Glenn Jocher
3eefab1bb1
Remove tensorrt pip install check (#7439) 2022-04-15 21:48:52 +02:00
Glenn Jocher
014acde79d
Update git_describe() (#7402) 
* Update `git_describe()`

Add .git path check to avoid `fatal: not a git repository (or any of the parent directories): .git` printout

* Update general.py
 2022-04-12 17:26:53 +02:00
Glenn Jocher
2da2466168
Fix EdgeTPU output directory (#7399) 
* Fix EdgeTPU output directory

Outputs to same directory as --weights

* Update export.py
 2022-04-12 15:08:53 +02:00
Glenn Jocher
5333b55e74
Remove OpenVINO ONNX opset<=12 check (#7398) 
No longer needed.
 2022-04-12 14:57:50 +02:00
Ayush Chaurasia
74aaab3312
Add version warning for wandb (#7385) 
* add version warning

* [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

* Update __init__.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>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-12 11:46:56 +02:00
Glenn Jocher
4bb7eb8b84
Dynamic normalization layer selection (#7392) 
* Dynamic normalization layer selection

Based on actual available layers. Torch 1.7 compatible, resolves https://github.com/ultralytics/yolov5/issues/7381

* Update train.py
 2022-04-12 11:02:11 +02:00
Vardan Agarwal
fa569cdae5
Add support for different normalization layers (#7377) 
* Add support for different normalization layers.

* [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>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-11 14:04:22 +02:00
Glenn Jocher
bd2dda8e64
Update optimizer param group strategy (#7376) 
* Update optimizer param group strategy

Avoid empty lists on missing BathNorm2d models as in https://github.com/ultralytics/yolov5/issues/7375

* fix init
 2022-04-11 12:34:35 +02:00
dependabot[bot]
71685cbf91
Bump actions/stale from 4 to 5 (#7371) 
Bumps [actions/stale](https://github.com/actions/stale) from 4 to 5.
- [Release notes](https://github.com/actions/stale/releases)
- [Changelog](https://github.com/actions/stale/blob/main/CHANGELOG.md)
- [Commits](https://github.com/actions/stale/compare/v4...v5)

---
updated-dependencies:
- dependency-name: actions/stale
  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-04-11 10:26:13 +02:00
Glenn Jocher
8c420c4c1f
Update ci-testing.yml (#7365) 
Remove keras==2.6.0 patch
 2022-04-10 15:17:25 +02:00
Glenn Jocher
b8d4f2bf74
Replace Slack with Community Forum in issues (#7364) 2022-04-10 14:50:01 +02:00
Glenn Jocher
db36f13c7a
Delete FUNDING.yml (#7363) 
Deleting as redundant with FUNDING.yml present in organization repo at https://github.com/ultralytics/.github
 2022-04-10 14:40:33 +02:00
Glenn Jocher
1993efd59e
Swap unsafe_chunk() for chunk() (#7362) 
Eliminates all unsafe function in YOLOv5 out of an abundance of caution.
 2022-04-10 13:53:28 +02:00