Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity
2,001 Commits 18 Branches 0 Tags
f00071416f2b20fc0d363c1cddeacbfbb2117a4a
Commit Graph
2001 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher f00071416f Refactor collections and fstrings (#7821) 
* Update torch_utils.py

* Additional code refactoring

* tuples to sets

* Cleanup
 2022-05-15 16:38:26 +02:00
Glenn Jocher 3356f2609c Add Python version output (#7814) 2022-05-14 18:30:47 +02:00
Glenn Jocherandpre-commit-ci[bot] 4a295b1a89 Add @threaded decorator (#7813) 
* Add `@threaded` decorator

* [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-05-14 16:12:08 +02:00
Glenn Jocher 9d8ed37df7 Rename utils/datasets.py > utils/dataloaders.py (#7799) 2022-05-13 14:34:16 +02:00
Yonghye Kwonpre-commit-ci[bot]Glenn Jocher
 		5a1ef32553 Add random interpolation method augmentation (#6826) 
* add random_interpolation option to make model robust to interpolation methods

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

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

* Fix precommit error

* Update augmentations.py

* Update augmentations.py

* Update augmentations.py

* Update datasets.py

* Update datasets.py

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-05-13 14:26:11 +02:00
Glenn Jocher 5e077bfd39 Add Dockerfile-M1 (#7720) 
* Add Dockerfile-M1

* Update

* Update

* Update Dockerfile-M1
 2022-05-13 14:15:21 +02:00
Glenn Jocher 8a4175cde0 Add smoothing to curve plots and max F1 index (#7798) 
@sergiossm
 2022-05-13 13:53:21 +02:00
Glenn Jocherandpre-commit-ci[bot] d95a728f55 Implement DDP static_graph=True (#6940) 
* Implement DDP `static_graph=True`

Experimental implementation of new PyTorch 1.11.0 DDP feature.

* Add 1.11.0 check

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

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

* [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-05-13 12:32:47 +02:00
Glenn Jocherandpre-commit-ci[bot] f3fecf94a9 Pin downloads to release version (#7790) 
* Pin downloads to release version

Fixes a release version to avoid forward-compatibility issues in future releases.

* [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-05-13 00:24:53 +02:00
Cristi Fatipre-commit-ci[bot]Glenn Jocher
 		4870064629 Ability to dowlnoad older assets (#7767) 
* Ability to dowlnoad older assets

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

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

* Cleanup

* Cleanup2

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-05-11 12:52:10 +02:00
Glenn Jocher d059d1da03 Report fused model summary by default (#7722) 2022-05-06 23:51:59 -07:00
Glenn Jocher aa7a0e91b1 Update on_train_end callback (#7716) 2022-05-05 21:25:13 -07:00
Glenn Jocherandpre-commit-ci[bot] 1d0a0944c1 Update tutorial.ipynb (#7715) 
* Update tutorial.ipynb

* Created using Colaboratory

* [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-05-05 20:55:52 -07:00
Glenn Jocher c4cb7c684a Update README.md (#7686) 2022-05-03 15:18:01 -07:00
Glenn Jocherandpre-commit-ci[bot] d4ea61e043 FROM nvcr.io/nvidia/pytorch:22.04-py3 (#7680) 
* FROM nvcr.io/nvidia/pytorch:22.04-py3

* Update Docker

* Update Docker

* Update Docker

* Update Docker

* Update TRT auto-install

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

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

* Cleanup

* Cleanup

* Cleanup cpu

* Cleanup cpu

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-05-03 13:47:30 -07:00
Glenn Jocher e305aba686 Pin Docker-cpu FROM ubuntu:20.04 (#7677) 2022-05-02 23:27:24 -07:00
Glenn Jocher c4862fcd31 Update Dockerfile-cpu to force python3.9 (#7675) 
* Update

* Update

* Do not install torch

* Pillow>9.1

* Update Dockerfile-cpu

* Update Dockerfile-cpu

* Update Dockerfile-cpu

* Update Dockerfile-cpu

* Update Dockerfile-cpu

* Update Dockerfile-cpu

* Update Dockerfile-cpu

* Update Dockerfile-cpu

* Update Dockerfile-cpu

* Update Dockerfile

* fix emoji

* reduce scipy

* add libpython3.9

* Update Dockerfile
 2022-05-02 20:53:02 -07:00
David MatosDavid MatosGlenn Jocher
 		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 MossandGlenn Jocher 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 Jocherandpre-commit-ci[bot] 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 Jocherandpre-commit-ci[bot] 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 Jocherandpre-commit-ci[bot] 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 Chaurasiapre-commit-ci[bot]Glenn Jocher
 		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
Powercube7andGlenn Jocher 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-CVerpre-commit-ci[bot]Glenn Jocher
 		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 Jocherandpre-commit-ci[bot] 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 KocherhansandGlenn Jocher 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 Jocherandpre-commit-ci[bot] 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