Commit Graph

1090 Commits (20a7368373b592d9832a10f79a7a527d1976e321)

Author SHA1 Message Date
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