85ae985b6a
update mask2segments and saving results ( #9785 )
...
* update mask2segments and saving results
* [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 predict.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>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-10-13 01:44:01 +02:00
Glenn Jocher
7a69035eb8
Update Loggers ( #9760 )
...
* 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 >
* Update requirements.txt
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-10-12 12:53:12 +02:00
Glenn Jocher
2f1eb21ad6
Remove #YOLOVISION22 notice ( #9751 )
...
Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-10-10 14:54:21 +02:00
Glenn Jocher
209be932de
Fix segment evolution keys ( #9742 )
...
* Update
* Cleanup
2022-10-09 23:51:29 +02:00
Paul Guerrie
5ef69ef3e6
Error in utils/segment/general masks2segments() ( #9724 )
...
When running segmentation predict on gpu, the conversion from tensor to numpy fails. Calling `.cpu()` solves this problem.
Signed-off-by: Paul Guerrie <97041392+paulguerrie@users.noreply.github.com >
Signed-off-by: Paul Guerrie <97041392+paulguerrie@users.noreply.github.com >
2022-10-06 22:55:15 +02:00
Glenn Jocher
7f097ddb6c
FROM nvcr.io/nvidia/pytorch:22.09-py3 ( #9711 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-10-05 22:29:46 +02:00
Glenn Jocher
e4398cf179
TensorRT --dynamic fix ( #9691 )
...
* Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-10-04 16:32:19 +02:00
pre-commit-ci[bot]
68d654d8c4
[pre-commit.ci] pre-commit suggestions ( #9685 )
...
updates:
- [github.com/asottile/pyupgrade: v2.37.3 → v2.38.2](https://github.com/asottile/pyupgrade/compare/v2.37.3...v2.38.2 )
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-04 16:31:51 +02:00
c98128fe71
Allow list for Comet artifact class 'names' field ( #9654 )
...
* Update __init__.py
In the Comet logger, when I run train.py, it wants to download the data artifact. It was requiring me to format the 'names' field in the data artifact metadata as a dictionary, so I've changed this so that it also accepts a list.
Signed-off-by: KristenKehrer <34010022+KristenKehrer@users.noreply.github.com >
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update utils/loggers/comet/__init__.py
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com >
Signed-off-by: KristenKehrer <34010022+KristenKehrer@users.noreply.github.com >
Signed-off-by: KristenKehrer <34010022+KristenKehrer@users.noreply.github.com >
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com >
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-10-03 00:25:10 +02:00
Glenn Jocher
1158a50abd
Simplify val.py benchmark mode with speed mode ( #9674 )
...
Update
2022-10-02 13:37:54 +02:00
Anant Sakhare and pre-commit-ci[bot]
8a19437690
Added cutout import from utils/augmentations.py to use Cutout Aug in … ( #9668 )
...
* Added cutout import from utils/augmentations.py to use Cutout Aug in data loader by un-commenting line 679, 680, 681
* [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-10-01 16:42:31 +02:00
Glenn Jocher
82bec4c878
Add NVIDIA Jetson Nano Deployment tutorial ( #9656 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-29 23:35:39 +02:00
Soumik Rakshit and pre-commit-ci[bot]
0860e58557
Disabled upload_dataset flag temporarily due to an artifact related bug ( #9652 )
...
* disabled upload_dataset flag temporarily due to an artifact related bug
* [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-09-29 23:01:45 +02:00
Glenn Jocher
799e3d0cc9
Add SegmentationModel unsupported warning ( #9632 )
...
* Add SegmentationModel unsupported warning
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update hubconf.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update hubconf.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-28 16:43:11 +02:00
Glenn Jocher
2373d5470e
NMS MPS device wrapper ( #9620 )
...
* NMS MPS device wrapper
May resolve https://github.com/ultralytics/yolov5/issues/9613
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-09-27 18:02:48 +02:00
Glenn Jocher
7314363f26
Add results.save(save_dir='path', exist_ok=False) ( #9617 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-27 16:58:14 +02:00
Glenn Jocher
1460e57157
Fix is_url(' https://ultralytics.com ') ( #9610 )
...
Failing on missing path, i.e. no 'www.'
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-27 00:29:23 +02:00
Glenn Jocher
6b2c9d1d0f
Update export.py Usage examples ( #9609 )
...
* Update export.py Usage examples
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-26 20:46:50 +02:00
Glenn Jocher
a5748e4b93
Updated Segmentation and Classification usage ( #9607 )
...
* Updated Segmentation and Classification usage
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-26 20:10:24 +02:00
Glenn Jocher
c4c0ee8fc3
Apple MPS -> CPU NMS fallback strategy ( #9600 )
...
Until more ops are fully supported this update will allow for seamless MPS inference (but slower MPS to CPU transfer before NMS, so slower NMS times).
Partially resolves https://github.com/ultralytics/yolov5/issues/9596
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-26 14:13:03 +02:00
Glenn Jocher
bd9c0c42ae
#YOLOVISION22 update ( #9598 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-26 13:27:34 +02:00
dependabot[bot]
bfe052b8e1
Bump actions/stale from 5 to 6 ( #9595 )
...
Bumps [actions/stale](https://github.com/actions/stale ) from 5 to 6.
- [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/v5...v6 )
---
updated-dependencies:
- dependency-name: actions/stale
dependency-type: direct:production
update-type: version-update:semver-major
...
Signed-off-by: dependabot[bot] <support@github.com >
Signed-off-by: dependabot[bot] <support@github.com >
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
2022-09-26 12:39:08 +02:00
Glenn Jocher
959a4665f8
#YOLOVISION22 announcement ( #9590 )
...
* #YOLOVISION22 announcement
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-09-25 23:26:15 +02:00
Glenn Jocher
9f1cf8dd1c
Add Paperspace Gradient badges ( #9588 )
...
* Add Paperspace Gradient badges
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 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 tutorial.ipynb
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update greetings.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update greetings.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update README_cn.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-25 23:03:14 +02:00
Glenn Jocher
9006b41498
Add global YOLOv5_DATASETS_DIR ( #9586 )
...
Update general.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-25 21:40:28 +02:00
Glenn Jocher
639d82fbab
Update Comet links ( #9587 )
...
* Update Comet links
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-09-25 20:12:57 +02:00
Glenn Jocher
966b0e09f0
TensorRT detect.py inference fix ( #9581 )
...
* Update
* Update ci-testing.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update ci-testing.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Segment fix
* Segment fix
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-25 16:21:26 +02:00
Glenn Jocher and pre-commit-ci[bot]
2787ad701f
Add segment line predictions ( #9571 )
...
* Add segment line predictions
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
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update
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-09-25 14:52:49 +02:00
David A. Macey and Glenn Jocher
ee91dc9bb3
Update requirements.txt comment https://pytorch.org/get-started/locally/ ( #9576 )
...
* Update Requirements with PyTorch CUDA
Added --extra-index-url https://download.pytorch.org/whl/cu116 URL to requirements file for ease of creating venv with CUDA enabled PyTorch. Otherwise CPU PyTorch is installed an unable to use local GPUs.
Signed-off-by: David A. Macey <davidamacey@gmail.com >
* Update requirements.txt
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update requirements.txt
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: David A. Macey <davidamacey@gmail.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-25 14:47:16 +02:00
Glenn Jocher
55fbac933b
Update default GitHub assets ( #9573 )
...
* Update default GitHub assets
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update downloads.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update downloads.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-25 02:59:25 +02:00
f11a8a62d2
generator seed fix for DDP mAP drop ( #9545 )
...
* Try to fix DDP mAP drop by setting generator's seed to RANK
* Fix default activation bug
* Update dataloaders.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update dataloaders.py
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-09-24 19:35:07 +02:00
Glenn Jocher
c8e52304cf
New scale_segments() function ( #9570 )
...
* Rename scale_coords to scale_boxes
* add scale_segments
2022-09-24 16:02:41 +02:00
d669a74623
Detect.py supports running against a Triton container ( #9228 )
...
* update coco128-seg comments
* Enables detect.py to use Triton for inference
Triton Inference Server is an open source inference serving software
that streamlines AI inferencing.
https://github.com/triton-inference-server/server
The user can now provide a "--triton-url" argument to detect.py to use
a local or remote Triton server for inference.
For e.g., http://localhost:8000 will use http over port 8000
and grpc://localhost:8001 will use grpc over port 8001.
Note, it is not necessary to specify a weights file to use Triton.
A Triton container can be created by first exporting the Yolov5 model
to a Triton supported runtime. Onnx, Torchscript, TensorRT are
supported by both Triton and the export.py script.
The exported model can then be containerized via the OctoML CLI.
See https://github.com/octoml/octo-cli#getting-started for a guide.
* added triton client to requirements
* fixed support for TFSavedModels in Triton
* reverted change
* Test CoreML update
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update ci-testing.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Use pathlib
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Refacto DetectMultiBackend to directly accept triton url as --weights http://...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Deploy category
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update detect.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update common.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update common.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update predict.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update predict.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update predict.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update triton.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update triton.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
* Add printout and requirements check
* Cleanup
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* triton fixes
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* fixed triton model query over grpc
* Update check_requirements('tritonclient[all]')
* group imports
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Fix likely remote URL bug
* update comment
* Update is_url()
* Fix 2x download attempt on http://path/to/model.pt
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Co-authored-by: glennjocher <glenn.jocher@ultralytics.com >
Co-authored-by: Gaz Iqbal <giqbal@octoml.ai >
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-24 00:56:42 +02:00
Glenn Jocher
1320ce183e
Update is_url() ( #9566 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-23 23:20:19 +02:00
30fa9b610a
Add --source screen for screenshot inference ( #9542 )
...
* add screenshot as source
* fix: screen number support
* Fix: mutiple screen specific area
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* parse screen args in LoadScreenshots
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* sequence+ '_' as file name for save-txt save-crop
* screenshot as stream
* Update requirements.txt
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update detect.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update detect.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update detect.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update detect.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update detect.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update predict.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update detect.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update predict.py
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 >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Co-authored-by: xin <xin@zhiyoung.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-09-22 23:58:14 +02:00
Glenn Jocher
b25d5a75f2
Refactor dataset batch-size ( #9551 )
2022-09-22 23:23:40 +02:00
Glenn Jocher and pre-commit-ci[bot]
489920ab30
TF.js fix ( #9536 )
...
* TF.js fix
May resolve https://github.com/ultralytics/yolov5/issues/9534
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 ci-testing.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update ci-testing.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update ci-testing.yml
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-09-22 00:34:35 +02:00
Glenn Jocher and pre-commit-ci[bot]
999482b451
import re ( #9535 )
...
* import re
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>
2022-09-21 23:08:52 +02:00
Glenn Jocher and pre-commit-ci[bot]
6f0284763b
Update albumentations ( #9503 )
...
* Add `RandomResizedCrop(ratio)`
* Update ratio
* Update ratio
* Update ratio
* Update ratio
* Update ratio
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Create augmentations.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
* Update augmentations.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-09-21 19:53:26 +02:00
Glenn Jocher
db6847431b
Update coremltools>=6.0 ( #9532 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-21 17:55:25 +02:00
Glenn Jocher
499a6bf573
Update scikit-learn constraint per coremltools 6.0 ( #9531 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-21 15:14:54 +02:00
Glenn Jocher
6ebef28894
Remove scikit-learn constraint on coremltools 6.0 ( #9530 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-21 15:12:12 +02:00
Glenn Jocher
77dcf55168
FROM nvcr.io/nvidia/pytorch:22.08-py3 ( #9520 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-20 22:57:42 +02:00
Glenn Jocher and pre-commit-ci[bot]
c0d9713845
TFLite fix 2 ( #9518 )
...
* TFLite fix 2
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>
2022-09-20 19:01:03 +02:00
Glenn Jocher and pre-commit-ci[bot]
bd35191033
Fix TF Lite exports ( #9517 )
...
* Update tf.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>
2022-09-20 18:47:14 +02:00
Glenn Jocher
e233c038ed
Remove check_requirements('flatbuffers==1.12') ( #9514 )
...
* Remove `check_requirements('flatbuffers==1.12')`
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update ci-testing.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update ci-testing.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-20 16:38:04 +02:00
f8b74631e5
Fix random_perspective param bug in segment ( #9512 )
...
* fix random_perspective param bug when mosaic=False
Signed-off-by: FeiGeChuanShu <774074168@qq.com >
* Update dataloaders.py
* Update dataloaders.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Signed-off-by: FeiGeChuanShu <774074168@qq.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-09-20 13:04:45 +02:00
Glenn Jocher and pre-commit-ci[bot]
095f601d9d
Segmentation polygons2masks_overlap() in np.int32 ( #9493 )
...
* Segmentation `polygons2masks_overlap()` in `np.int32`
May resolve https://github.com/ultralytics/yolov5/issues/9461
WARNING: Masks should be uint8 for fastest speed, change needs profiling results to determine impact.
@AyushExel @Laughing-q
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update dataloaders.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>
2022-09-20 12:22:02 +02:00
曾逸夫（Zeng Yifu） and pre-commit-ci[bot]
63368e71d2
Add paddle tips ( #9502 )
...
* Update export.py
Signed-off-by: 曾逸夫（Zeng Yifu） <41098760+Zengyf-CVer@users.noreply.github.com >
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Signed-off-by: 曾逸夫（Zeng Yifu） <41098760+Zengyf-CVer@users.noreply.github.com >
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-20 01:11:29 +02:00
Glenn Jocher
0171198f38
Fix visualization title bug ( #9500 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-19 20:38:11 +02:00
