Commit Graph

1147 Commits (d62a4e7d9d71ac84d4eef430f53cfb38b4acf856)

Author SHA1 Message Date
Glenn Jocher be61a64c47
Update links to https://docs.ultralytics.com (#11412)
* Update links

* [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>
2023-04-21 23:47:29 +02:00
Glenn Jocher fb1e746b8a
Docs update for new YOLOv5 URLs (#11409)
* Update YOLOv5 Docs Tutorials

* Update YOLOv5 Docs Tutorials

* Update actions

* Update actions

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

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

* Update actions

* Update actions

* Update actions

* Update actions

* Update actions

* Update actions

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-04-21 13:50:28 +02:00
Glenn Jocher 94714fe6ad
Update requirements.txt to comment tensorboard (#11362)
* Update requirements.txt comment tensorboard

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Make tensorboard optional

* Update __init__.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-04-15 00:51:17 +02:00
Glenn Jocher 60e29e2d86
Update check_requirements() (#11360)
Update general.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-04-14 14:47:07 +02:00
Glenn Jocher 2118e3b88e
Update check_requirements.py (#11358)
Update general.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-04-14 14:37:02 +02:00
Glenn Jocher 34cf749958
Update LICENSE to AGPL-3.0 (#11359)
* Update LICENSE to AGPL-3.0

This pull request updates the license of the YOLOv5 project from GNU General Public License v3.0 (GPL-3.0) to GNU Affero General Public License v3.0 (AGPL-3.0).

We at Ultralytics have decided to make this change in order to better protect our intellectual property and ensure that any modifications made to the YOLOv5 source code will be shared back with the community when used over a network.

AGPL-3.0 is very similar to GPL-3.0, but with an additional clause to address the use of software over a network. This change ensures that if someone modifies YOLOv5 and provides it as a service over a network (e.g., through a web application or API), they must also make the source code of their modified version available to users of the service.

This update includes the following changes:
- Replace the `LICENSE` file with the AGPL-3.0 license text
- Update the license reference in the `README.md` file
- Update the license headers in source code files

We believe that this change will promote a more collaborative environment and help drive further innovation within the YOLOv5 community.

Please review the changes and let us know if you have any questions or concerns.


Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update headers to AGPL-3.0

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-04-14 14:36:16 +02:00
Glenn Jocher f7c656278c
Update general.py (#11322)
* Update general.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

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

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

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-04-09 17:51:25 +02:00
Grzegorz K a82132c10b
Do not monkey-patch cv2 methods for unaware caller (#11210) 2023-03-30 11:39:11 +02:00
Glenn Jocher b54fd0ac28
FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime (#11246)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-03-26 20:27:43 +02:00
Grzegorz K f095258e0c
Sync signature of cv2.imread and cv2.imwrite (#11209)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-03-23 20:34:03 +01:00
Eljas Hyyrynen 6dd17516c8
dataloaders: fix class filtering for segmentation (#11171)
* dataloaders: fix class filtering for segmentation

self.segments[i] and segment[j] are lists so they cannot be indexed with booleans
self.segments is a tuple so it has to be converted into a list first

* [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>
2023-03-23 20:29:11 +01:00
imyhxy d223460f3a
Fixed creation of empty directories in path was entered mistaken (#11174)
FFixed creation of empty directories in path was entered mistaken

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-03-23 20:27:46 +01:00
Sheng Hu 78a90c9661
Remove duplicate assignment code (#11178)
Signed-off-by: Sheng Hu <husheng_7@163.com>
2023-03-23 11:29:44 +01:00
Jonas Heinle ef7039ea18
Error in tensor shape of docstring (#11206)
Corrected the tensor shape in the doc string.
The incoming masks are stacked in dim=0 therefore the doc is wrong

Signed-off-by: Jonas Heinle <jonasheinle@googlemail.com>
2023-03-23 11:28:31 +01:00
Sheng Hu 5543b89466
Fix a visualization bug (#11134)
Fix a visualization bug reported here: https://github.com/ultralytics/yolov5/issues/11133

Signed-off-by: Sheng Hu <husheng_7@163.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-03-09 23:55:02 +01:00
Glenn Jocher 3c0a6e664b
Security fixes for IPython (#11069)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-25 16:15:07 -08:00
Glenn Jocher feca55719b
Update "YOLOv5 is out of date" msg (#11061) 2023-02-24 20:09:02 -08:00
Glenn Jocher 7a972e86c4
Update .pre-commit-config.yaml (#11009)
* Update .pre-commit-config.yaml

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

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

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

* Update __init__.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update .pre-commit-config.yaml

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Pre-commit updates

* Pre-commit updates

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-02-18 01:06:24 +01:00
Glenn Jocher 34e1bc8ee3
Update downloads.py (#11005)
* Update downloads.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

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

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

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-02-17 10:53:45 +01:00
Snyk bot 6d283ec167
[Snyk] Security upgrade werkzeug from 1.0.1 to 2.2.3 (#10995)
* fix: utils/google_app_engine/additional_requirements.txt to reduce vulnerabilities


The following vulnerabilities are fixed by pinning transitive dependencies:
- https://snyk.io/vuln/SNYK-PYTHON-WERKZEUG-3319935
- https://snyk.io/vuln/SNYK-PYTHON-WERKZEUG-3319936

* [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>
2023-02-16 18:08:13 +01:00
Aarni Koskela a2de5c5bf6
Subprocess improvements (#10973)
* Use list-form arguments for subprocess.run calls where possible

Augments #10944

* Deduplicate curl code

* Avoid eval() to parse integer

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-13 18:00:31 +04:00
Aarni Koskela 1a2eb532ce
Fix return value check for subprocess.run (#10972)
Subprocess.run does not return an integer.

Regressed in #10944
2023-02-13 17:38:27 +04:00
Glenn Jocher 5c3eba664e
Update Dockerfile `apt upgrade openssl` (#10951)
Update Dockerfile upgrade `openssl`

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-10 21:19:08 +04:00
David Strahm d389840f66
Allow int8 quantization for export_tfjs (#10948)
* Allow int8 quantization for export_tfjs

--int8 param currently has no effect on export_tfjs. With this change, 
` python export.py --weights ../path/to/best.pt --include tfjs --int8` will add the --quantize_uint8 param to the tensorflowjs_converter script, greatly reducing model size for web usage.

Signed-off-by: David Strahm <david.strahm@lambda-it.ch>

* Update Dockerfile

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

---------

Signed-off-by: David Strahm <david.strahm@lambda-it.ch>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-10 19:11:08 +04:00
Glenn Jocher 80e5447390
Dockerfile `openssl` security update (#10949)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-10 18:59:22 +04:00
Snyk bot a895e98172
[Snyk] Security upgrade ubuntu from latest to rolling (#10946)
* fix: utils/docker/Dockerfile-cpu to reduce vulnerabilities

The following vulnerabilities are fixed with an upgrade:
- https://snyk.io/vuln/SNYK-UBUNTU2204-OPENSSL-3314744
- https://snyk.io/vuln/SNYK-UBUNTU2204-OPENSSL-3314768
- https://snyk.io/vuln/SNYK-UBUNTU2204-OPENSSL-3314792
- https://snyk.io/vuln/SNYK-UBUNTU2204-OPENSSL-3314816
- https://snyk.io/vuln/SNYK-UBUNTU2204-OPENSSL-3314840

* Update Dockerfile-arm64

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-10 00:21:38 +04:00
Glenn Jocher 35d6d9f408
Update Dockerfile-arm64 (#10945)
* Update Dockerfile-arm64

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update Dockerfile-cpu

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-09 20:32:58 +04:00
Glenn Jocher 238da321cb
Security3 (#10944)
* Security improvements

* Security improvements

* [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>
2023-02-09 17:58:24 +04:00
Glenn Jocher 61407c93cc
Security improvements for subprocess.run() (#10943)
* Security improvements

* Security improvements

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

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

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-02-09 17:18:27 +04:00
Glenn Jocher e326252ee4
Security improvements (#10942)
* Security improvements

* Security improvements

* [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>
2023-02-09 16:57:18 +04:00
Snyk bot 976fa99e5c
[Snyk] Security upgrade gunicorn from 19.9.0 to 19.10.0 (#10933)
fix: utils/google_app_engine/additional_requirements.txt to reduce vulnerabilities


The following vulnerabilities are fixed by pinning transitive dependencies:
- https://snyk.io/vuln/SNYK-PYTHON-GUNICORN-541164
2023-02-09 11:52:22 +04:00
Glenn Jocher c3c8e71d7a
Update Dockerfile-arm64 (#10918)
Docker fixes
2023-02-07 02:15:54 +04:00
Glenn Jocher 9ba18266b2
Update Dockerfile (#10917)
* Update Dockerfile

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update Dockerfile-arm64

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-07 01:58:47 +04:00
Glenn Jocher 8b5a7d4179
Update Dockerfile (#10916)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-06 22:41:03 +04:00
dependabot[bot] b1a3126e5d
Bump docker/build-push-action from 3 to 4 (#10911)
* Bump docker/build-push-action from 3 to 4

Bumps [docker/build-push-action](https://github.com/docker/build-push-action) from 3 to 4.
- [Release notes](https://github.com/docker/build-push-action/releases)
- [Commits](https://github.com/docker/build-push-action/compare/v3...v4)

---
updated-dependencies:
- dependency-name: docker/build-push-action
  dependency-type: direct:production
  update-type: version-update:semver-major
...

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

* Docker (#10913)

* Dockerfile standardizations and improvements

* README fixes

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

---------

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
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>
2023-02-06 22:12:26 +04:00
Glenn Jocher b8a2c47fa9
Update Dockerfile `FROM pytorch/pytorch:latest` (#10902)
* Update Dockerfile `FROM pytorch/pytorch:latest`

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* isort

* precommit

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

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

* spelling

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

* Update .pre-commit-config.yaml

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Cleanup

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

* Cleanup

* Cleanup

* Cleanup

* Cleanup

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-02-06 15:11:32 +04:00
Laughing 6a62c94190
fix zero labels (#10820)
update
2023-02-04 16:21:30 +04:00
ZhuGeRoastedFish 3a059125dd
Fx confusion-matrix xlabel typo (#10692)
fix confusion-matrix xlabel typo

Signed-off-by: ZhuGeRoastedFish <77224640+ZhuGeRoastedFish@users.noreply.github.com>

Signed-off-by: ZhuGeRoastedFish <77224640+ZhuGeRoastedFish@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-01-14 14:43:27 +01:00
Glenn Jocher 2b356c0ab2
Update Dockerfile (#10768)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-01-14 12:40:15 +01:00
Glenn Jocher bd10f0f6c7
Revert PIL deprecation fix
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-01-12 18:01:36 +01:00
Glenn Jocher 0e24b7e2f5
PIL `.get_size()` deprecation fix (#10754)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-01-12 17:43:12 +01:00
Glenn Jocher 65071da718
Update Dockerfile `pip install -U pycocotools` (#10606)
* Update Dockerfile `pip install -U pycocotools`

Previous command not working.

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update Dockerfile

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-27 16:47:26 +01:00
Glenn Jocher 8ca1826134
Update `pycocotools>=2.0.6` (#10605)
* Update `pycocotools>=2.0.6`

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update val.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update val.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update Dockerfile

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-27 15:41:03 +01:00
Glenn Jocher e72dc1faba
Dockerfile uninstall torch nightly in favor of stable (#10604)
@AyushExel @Laughing-q fix for Docker error 

```
AttributeError: Can't get attribute '_rebuild_parameter_v2' on <module 'torch._utils' from '/Users/glennjocher/PycharmProjects/ultralytics/venv/lib/python3.10/site-packages/torch/_utils.py'> 
```

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-27 14:32:21 +01:00
Glenn Jocher 3c1afd9ab6
ENV OMP_NUM_THREADS=1 (#10593)
@Laughing-q @AyushExel setting to 1 due to recent issues

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-26 14:54:43 +01:00
Glenn Jocher 5f8054c47c
FROM nvcr.io/nvidia/pytorch:22.12-py3 (#10588)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-24 18:15:33 +01:00
Mr.Li 2370a5513e
Bugfix: update dataloaders.py to fix "resize to 0" (#10558)
* fix bug "resize to 0"

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

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

* Use math.ceil() for resize to enforce min floor of 1 pixel

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
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-12-22 14:55:09 +01:00
Yonghye Kwon 5545ff3545
Sort by confidence and remove excess boxes without explicit if (#10517)
* sort by confidence and remove excess boxes without explicit if

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>

* cleanup indexing boxes for remove excess boxes

it is related to https://github.com/ultralytics/yolov5/pull/10502.

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-19 17:19:14 +01:00
Yonghye Kwon 10e93d295f
Set a seed of generator with an option for more randomness when training several models with different seeds (#10486)
* set seed with parameter

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>

* make seed to be a large number

* set seed with a parameter

* set a seed of dataloader with opt for more randomness

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-19 10:27:34 +01:00
Yonghye Kwon 2c35c1b318
Limit detections without explicit if condition (#10502)
* limit detections without explicit if condition

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>

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

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

* cleanup indexing code for limit detections

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>
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-12-18 13:03:01 +01:00
Yonghye Kwon 8d65f9d8ce
Support extensive shape for functions related to bounding box localization (#10516)
* support extensive shape for functions related to bounding box localization

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>

* merge exp branch updates

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-17 12:10:26 +01:00
Yonghye Kwon b564c1f365
Check `conf_thres` and `iou_thres` prior to use (#10515)
* Checks conf_thres and iou_thres at beign

Why checks conf_thres after operation with it?

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>

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

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

* Update general.py

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>
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-12-17 12:05:00 +01:00
nerdyespresso 1ae91940ab
Update Comet hyperlinks (#10500)
* Update README.md

Signed-off-by: nerdyespresso <106761627+nerdyespresso@users.noreply.github.com>

* Update README.md

Signed-off-by: nerdyespresso <106761627+nerdyespresso@users.noreply.github.com>

* Update README.md

Signed-off-by: nerdyespresso <106761627+nerdyespresso@users.noreply.github.com>

* Update README.md

Signed-off-by: nerdyespresso <106761627+nerdyespresso@users.noreply.github.com>

* Update tutorial.ipynb

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update tutorial.ipynb

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update tutorial.ipynb

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: nerdyespresso <106761627+nerdyespresso@users.noreply.github.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-15 13:56:42 +01:00
Yonghye Kwon 443ef7f33e
Modify a comment for OpenCV File I/O Functions (#10467)
Modify comment for OpenCV File I/O Functions

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>
2022-12-11 19:29:19 -08:00
Victor Sonck f8539a6800
Allow users to specify how to override a ClearML Task (#10363)
* Added basic flag to enable reusing last task clearml

* Added option to provide task ID to override

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

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

* Use exist_ok argument instead

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-12-05 12:12:19 -08:00
Colin Wong 5dc1ce4e86
Support `.txt` files as a line-by-line media list rather than streams (#10059)
* Update streams.txt default

Signed-off-by: Colin Wong <cwongmath@gmail.com>

* Change streams list extension to .streams

* Read txt as media per line

* Missed one

* Missed another one

* Update dataloaders.py

* Update detect.py

* Update dataloaders.py

* Update detect.py

* Update predict.py

* Update predict.py

* Update README.md

Signed-off-by: Colin Wong <cwongmath@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-03 14:58:58 -08:00
Glenn Jocher 9722e6ffe5
`process_mask_native()` cleanup (#10366)
* process_mask_native() cleanup

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

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

* fix arg name

* cleanup anno_json

* Remove scale_image

* Remove scale_image

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

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

* update to native

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-03 14:41:08 -08:00
Glenn Jocher a1b6e79ccf
Revert TQDM bar format changes (#10343)
Per https://github.com/ultralytics/yolov5/issues/10342

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-02 13:28:33 -08:00
Michael Sutton d7955fe438
Fix clearml args logging when training is launch with run() (#10359)
* Connect opt to clearml args

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

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

* Update clearml_utils.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

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

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
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-12-01 15:00:43 -08:00
Glenn Jocher 7845cea913
Fix ClearML unconfigured error (#10369)
@thepycoder adds Try Except for installed but unconfigured clearml environments.

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-01 13:56:33 -08:00
Victor Sonck 1ce464f689
Add docker info for ClearML remote execution (#10142)
* Add docker info for ClearML remote execution

* add additional clearml options to handle different python versions

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

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

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-12-01 13:32:55 -08:00
Michael Ben ami 028b7cdb5a
fix_reading_nan_in_evolve (#10358)
when there is `nan` in evolve.csv pandas read it as str 
remove the space before fix that 

Signed-off-by: Michael Ben ami <31584614+mbenami@users.noreply.github.com>

Signed-off-by: Michael Ben ami <31584614+mbenami@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-01 12:44:14 -08:00
Laughing b412696ff3
Fix & speed up segment plot (#10350)
* fix plot&&speed up

* fix segment save-txt

* fix channel

* [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-12-01 08:39:24 -08:00
Hu Ye e808f2267d
Eliminate unused `ConfusionMatrix.matrix()` method (#10309)
* fix bug in confusion_matrix

Signed-off-by: Hu Ye <xiaohuzc@gmail.com>

* Update metrics.py

* Update metrics.py

* Update metrics.py

Signed-off-by: Hu Ye <xiaohuzc@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-29 19:32:34 -08:00
Glenn Jocher 31c1f11186
`bbox_iou()` optimizations (#10296)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-25 15:28:50 -08:00
Glenn Jocher bfa1f23045
FROM nvcr.io/nvidia/pytorch:22.11-py3 (#10279)
* Update Docker usage examples

* Update Dockerfile

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update DEBIAN_FRONTEND

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-24 16:34:14 +01:00
Glenn Jocher b32f67f6be
`--single-cls` segments fix (#10260)
--single-cls segments fix

May resolve #10230

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-22 20:30:14 +01:00
Glenn Jocher 915bbf294b
YOLOv5 v7.0 release updates (#10245)
* YOLOv5 v7.0 splash image update

* Update tutorial.ipynb

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update tutorial.ipynb

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update tutorial.ipynb

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* readme segmentation section

* readme segmentation section

* readme segmentation section

* readme segmentation section

* readme segmentation section

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update tutorial.ipynb

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

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

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

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update download URLs to 7.0 assets

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-22 16:23:47 +01:00
Glenn Jocher 72cad39854
Squeezenet reshape outputs fix (#10222)
@AyushExel

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-19 16:44:56 +01:00
Glenn Jocher 40bb8030f8
Scope gitpyhon import in `check_git_info()` (#10221)
* Scope gitpyhon import in `check_git_info()`

* [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-11-19 14:22:07 +01:00
Glenn Jocher 6992dde4bd
Update Comet preview image (#10220)
* Update Comet preview image

Pass through tinyjpg: 2.2MB -> 497kB :)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update tutorial.ipynb

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update tutorial.ipynb

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update tutorial.ipynb

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-19 13:44:46 +01:00
Glenn Jocher 9286336cb4
Add `git` info to training checkpoints (#9655)
* Add git status on train checkpoints

* Update

* Update

* Update

* Update

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

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

* Update general.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update general.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-19 03:16:17 +01:00
Glenn Jocher 0322bb3196
New global `TQDM_BAR_FORMAT` (#10211)
* New global TQDM_BAR_FORMAT

* [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-11-18 21:39:13 +01:00
Glenn Jocher 74b3886edd
Simplify dataloader tqdm descriptions (#10210)
* Simplify dataloader tqdm descriptions

@AyushExel this should help our tqdm dataloader messages fit better within a single line in our Colab notebooks and also help avoid confusion about missing/empty labels, now combined into 'backgrounds'.

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

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

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

* Update dataloaders.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-18 21:18:57 +01:00
Glenn Jocher 467a57f01b
Segment prediction labels normalization fix (#10205)
* normalize_segments

* round remove

* swap axes fix
2022-11-18 14:19:46 +01:00
Ryan Echols ff6e6e328e
Fix: try 2 - prevent logging config clobbering (#10192)
* fix: try 2 - prevent logging config clobbering

Previous behavior: loading this repository with `torch.hub.load` clobbers the existing logging configuration by modifying the root logger's configuration.
New behavior: loading this repository with `torch.hub.load` only clobbers the logging configuration for logger `yolov5` and its descendants. This is done in a way compatible with Google Colab

Signed-off-by: Ryan Echols <ryan@shadylakemedia.com>

* chore: fill in comment

no-op so a pre-commit hook can auto-format files

Signed-off-by: Ryan Echols <ryan@shadylakemedia.com>

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

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

Signed-off-by: Ryan Echols <ryan@shadylakemedia.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-17 20:47:46 +01:00
Glenn Jocher 1510111b46
data.yaml `names.keys()` integer assert (#10190)
* data.yaml `names.keys()` integer assert

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update general.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-17 19:22:18 +01:00
Glenn Jocher e40662ffdd
Revert "fix: prevent logging config clobbering" (#10177)
Revert "fix: prevent logging config clobbering (#10133)"

This reverts commit 166b9f2fa7.
2022-11-16 19:06:19 +01:00
tripleMu 90575107e7
Filter PyTorch 1.13 UserWarnings (#10166)
FilterWarning for torch.distributed._all_gather_base

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-16 16:10:15 +01:00
Ryan Echols 166b9f2fa7
fix: prevent logging config clobbering (#10133)
Previous behavior: loading this repository with `torch.hub.load` clobbers the existing logging configuration by modifying the root logger's configuration.
New behavior: loading this repository with `torch.hub.load` only clobbers the logging configuration for logger `yolov5` and its descendants.

Signed-off-by: Ryan Echols <ryan@shadylakemedia.com>

Signed-off-by: Ryan Echols <ryan@shadylakemedia.com>
2022-11-16 16:09:30 +01:00
Amol Dumrewal 5e03f5fc8c
Fix dataloader filepath modification to perform replace only once and not for all occurences of string (#10163)
* Fix dataloader filepath modification to perform only once and not for all occurences of string

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

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

* cleanup

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
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-11-15 19:00:33 +01:00
Glenn Jocher 9dd40f0723
Add `join_threads()` (#10086)
* Update __init__.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update __init__.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update __init__.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-15 16:27:07 +01:00
Glenn Jocher ea73386e5a
Add Copy-Paste to `mosaic9()` (#10165)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-15 14:19:47 +01:00
Glenn Jocher abbfd69523
Copy-Paste zero value fix (#10152)
* Copy-Paste zero value fix

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update augmentations.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-14 18:54:49 +01:00
Glenn Jocher 078059c5b3
Update IoU functions (#10123)
Remove box area function and support expandable bbox_iou() calls.

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-10 18:32:34 +01:00
Victor Sonck 75728bb3ea
Mapped project and name to ClearML (#10100)
* Mapped project and name to ClearML

* Add project and task name docs

* [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-11-09 17:45:09 +01:00
Glenn Jocher c1fcfe8cd9
YouTube Usage example comments (#10106)
* YouTube Usage example comments

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update dataloaders.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-09 17:20:26 +01:00
Glenn Jocher 86decb3c49
Add `ultralytics` pip package (#10103) 2022-11-09 15:05:19 +01:00
Glenn Jocher 0039870b89
Update Dockerfile keep default torch installation (#10071)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-08 00:58:00 +01:00
Glenn Jocher 78ed31c95a
IoU `eps` adjustment (#10051)
IoU eps adjustment

Unify h1 and h2 with eps values

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-06 15:06:18 +01:00
Glenn Jocher fde7758468
YOLOv5 AutoCache Update (#10027)
* AutoCache

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

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

* AutoCache

* AutoCache

* AutoCache

* AutoCache

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-04 15:30:12 +01:00
Glenn Jocher 82a558557a
Fix `min_items` (#10001)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 16:41:37 +01:00
Glenn Jocher 067ad9a2d1
Improved `check_online()` robustness (#10000)
* Improved check_online() robustness 

YOLOv5-wide improvement, not just in check_requirements()

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update general.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 15:55:18 +01:00
Glenn Jocher c55e2cd73b
Add `min_items` filter option (#9997)
* Add `min_items` filter option

@AyushExel @Laughing-q dataset filter

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update dataloaders.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* fix

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

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-01 14:53:14 +01:00
Glenn Jocher cf99788823
Check online twice on AutoUpdate (#9999)
Increased robustness to network failures

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 14:27:36 +01:00
Glenn Jocher 49156eb1d1
Fix `is_colab()` (#9994)
@AyushExel @kalenmike 

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 12:02:05 +01:00
Glenn Jocher 6e544d5f7c
FROM nvcr.io/nvidia/pytorch:22.10-py3 (#9966)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-29 13:31:01 +02:00
Kalen Michael 575055ce70
Switch from suffix checks to archive checks (#9963)
* fix: switched from suffix checks to archive checks

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

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

* Cleanup

seems like both functions accept Path type input so removing str()

* import always

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-10-28 21:16:03 +02:00
Glenn Jocher f9bb984e81
Add ClearML minimum version requirement (#9933)
* Add ClearML minimum version requirement

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-26 14:06:05 +02:00
Glenn Jocher a5d875adca
Add `gnupg` to Dockerfile-cpu (#9932)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-26 13:42:52 +02:00