Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
2,646 Commits 18 Branches 0 Tags
c3e4e94e944de3b41b3398e2f78e596384739339
Commit Graph
96 Commits
Author SHA1 Message Date
abuod0 4408d02e7a Fixing onnx and Tensorflow versionsm for Jetson (#11377) 
Signed-off-by: abuod0 <97100904+abuod0@users.noreply.github.com>
 2023-04-18 12:21:35 +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
whx-sGlenn Jocherpre-commit-ci[bot]
 		a190b3c16b Update requirements.txt (#11294) 
* Update requirements.txt

Signed-off-by: whx-s <127172288+whx-s@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: whx-s <127172288+whx-s@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-04-09 16:34:18 +02:00
Glenn Jocherandpre-commit-ci[bot] cca5e21995 Update greetings.yml (#11287) 
* Update greeting

* Cleanup README

* Created using Colaboratory

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

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

* 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>
 2023-04-01 13:02:51 +02: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 7dee52f94d Update requirements.txt (#11068) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-02-25 15:58:59 -08:00
Glenn Jocher b005788f36 Update requirements.txt (#11067) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-02-25 15:15:50 -08:00
Glenn Jocher b8731d855f Update requirements.txt (#11065) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-02-25 14:16:03 -08:00
Snyk botandGlenn Jocher a270b4f125 [Snyk] Security upgrade setuptools from 39.0.1 to 65.5.1 (#10934) 
* fix: requirements.txt to reduce vulnerabilities


The following vulnerabilities are fixed by pinning transitive dependencies:
- https://snyk.io/vuln/SNYK-PYTHON-SETUPTOOLS-3180412

* Update requirements.txt

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-09 11:54:04 +04:00
Snyk botandGlenn Jocher a3c0fd0521 [Snyk] Fix for 2 vulnerabilities (#10931) 
* fix: requirements.txt to reduce vulnerabilities


The following vulnerabilities are fixed by pinning transitive dependencies:
- https://snyk.io/vuln/SNYK-PYTHON-SETUPTOOLS-3180412
- https://snyk.io/vuln/SNYK-PYTHON-WHEEL-3180413

* Update requirements.txt

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-09 11:45:58 +04:00
Yonghye KwonandGlenn Jocher 632bf485b4 Remove rocket emoji causes cp949 codec errors (#10646) 
Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-01-02 21:10:13 +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 b2a0f1cdc5 Update onnx>=1.12.0 (#10526) 2022-12-18 20:06:01 +01:00
Glenn Jocherandpre-commit-ci[bot] 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 86decb3c49 Add ultralytics pip package (#10103) 2022-11-09 15:05:19 +01: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 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
David A. MaceyandGlenn 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
Gaz IqbalglennjocherGaz Iqbalpre-commit-ci[bot]
 		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
zombobxinpre-commit-ci[bot]Glenn Jocher
 		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 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 Jocherandpre-commit-ci[bot] 92b52424d4 TensorFlow macOS AutoUpdate (#9471) 
* TensorFlow macOS AutoUpdate

* [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-18 17:34:34 +02:00
Glenn Jocher bd5fd78411 Comment protobuf in requirements.txt (#9207) 
The low package version is causing conflicts among other dependencies, commenting it causes no ill effects in CI so this should be fine.

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

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-29 17:13:55 +02:00
Victor Sonckpre-commit-ci[bot]Glenn Jocher
 		378bde4bba ClearML experiment tracking integration (#8620) 
* Add titles to matplotlib plots

* Add ClearML Experiment Tracking integration.

* Add ClearML Data Version Management automatic download when requested

* Add ClearML Hyperparameter Optimization

* ClearML save period integration

* Fix wandb breaking when used with ClearML dataset

* Fix wandb breaking when used with ClearML resume and dataset

* Add ClearML documentation

* fixed small bug in clearml integration that misreports epoch number

* Final ClearMl additions before refactor

* Add correct epoch reporting

* Add remote execution and autoscaling docs for ClearML integration

* Added images to clearml integration docs

* fixed logo alignment bug and added hpo screenshot clearml

* Fixed small epoch number bug in clearml integration

* Remove saved model flush clearml

* Cleanup clearml readme section

* Cleaned up clearml logger docstring

* Remove resume readme section clearml

* Clearml integration cleanup

* Updated ClearML documentation

* Added dark vs light icons ClearML Readme

* Clearml Readme styling

* Add better gifs

* Fixed gif file size

* Add better images in tutorial notebook

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

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

* Addressed comments in PR #8620

* Fixed circular import

* Fixed circular import

* Update tutorial.ipynb

* Update tutorial.ipynb

* Inline comment

* Restructured tutorial notebook

* Add correct ClearML link to README

* Update tutorial.ipynb

* Update general.py

* Update __init__.py

* Update __init__.py

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

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

* Update __init__.py

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

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

* Update __init__.py

* Update README.md

* Update __init__.py

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

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

* spelling

* Update tutorial.ipynb

* notebook cutt.ly links

* Update README.md

* Update README.md

* cutt.ly links in tutorial

* Removed labels as they show up on last subplot only

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-08-05 20:50:49 +02:00
Glenn Jocher 9559601b9a Update requirements.txt with tf-cpu and tf-aarch64 (#8807) 2022-07-31 14:54:55 +02:00
Glenn Jocher 3e858633b2 Revert protobuf<=3.20.1 (#8742) 
Resolve #8012 (again)
 2022-07-27 17:27:44 +02:00
Glenn Jocher d5116bbe9c coremltools>=5.2 for CoreML export (#8725) 2022-07-26 17:50:49 +02:00
daquexianandGlenn Jocher 92e47b85d9 Upgrade onnxsim to v0.4.1 (#8632) 
* upgrade onnxsim to v0.4.1

Signed-off-by: daquexian <daquexian566@gmail.com>

* Update export.py

* Update export.py

* Update export.py

* Update export.py

* Update export.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-07-19 19:01:24 +02:00
Glenn Jocherandpre-commit-ci[bot] 9cf5fd5ac3 assert torch!=1.12.0 for DDP training (#8621) 
* assert torch!=1.12.0 for DDP training

* [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-07-18 15:05:58 +02:00
Glenn Jocher 5c45a4b13d Update tqdm>=4.64.0 and thop>=0.1.1 (#8596) 
* Update `tqdm>=4.64.0` and `thop>=0.1.1`

* Update requirements.txt
 2022-07-16 16:14:16 +02:00
Glenn Jocher cf28dda366 Update scipy>=1.7.3 (#8595) 2022-07-16 15:54:34 +02:00
Glenn Jocher 72a81e7a1c Update requirements.txt comment spacing (#8562) 2022-07-15 16:01:29 +02:00
Glenn Jocher 574ceedfc5 Add thop>=0.1.0 (#8558) 2022-07-12 14:49:54 +02:00
曾逸夫（Zeng Yifu） fbd3020525 Add TensorRT dependencies (#8553) 
Update requirements.txt
 2022-07-12 13:19:25 +02:00
Junya Morioka 1ab23fc67f Exclude torch==1.12.0, torchvision==0.13.0 (Fix #8395) (#8497) 
Exclude torch==1.12.0, torchvision==0.13.0
 2022-07-06 19:32:58 +02:00
dependabot[bot]andGlenn Jocher 50ff6eee31 Update protobuf requirement from <=3.20.1 to <4.21.3 (#8346) 
Updates the requirements on [protobuf](https://github.com/protocolbuffers/protobuf) to permit the latest version.
- [Release notes](https://github.com/protocolbuffers/protobuf/releases)
- [Changelog](https://github.com/protocolbuffers/protobuf/blob/main/generate_changelog.py)
- [Commits](https://github.com/protocolbuffers/protobuf/commits)

---
updated-dependencies:
- dependency-name: protobuf
  dependency-type: direct:production
...

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>
 2022-06-28 12:04:58 +02:00
Glenn Jocher 7433d3829a Add requirements.txt protobuf<=3.20.1 (#8017) 
* Pin `protobuf==3.20.1`

* Update requirements.txt

* Update ci-testing.yml
 2022-05-28 12:12:27 +02:00
Glenn Jocherandpre-commit-ci[bot] 980aace209 Add psutil and ipython to requirements.txt (#7998) 
* Add `psutil` and `ipython` to requirements.txt

Lightweight packages used by YOLOv5 for system utilization (psutil) and interactive notebooks (IPython)

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

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

* sort alphabetically

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-05-26 19:12:19 +02:00
Glenn Jocher 1dcb774998 Update requirements.txt with Usage example (#7966) 
Update requirements.txt
 2022-05-25 01:40:17 +02: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
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 95c7bc25d3 OpenVINO Export (#6057) 
* OpenVINO export

* Remove timeout

* Add 3 files

* str

* Constrain opset to 12

* Default ONNX opset to 12

* Make dir

* Make dir

* Cleanup

* Cleanup

* check_requirements(('openvino-dev',))
 2021-12-22 20:29:48 +01:00
hamid mohammadiandGlenn Jocher 97b6b14abe Add requests to requirements.txt (#5112) 
* Requirements updated

1. Requests added to requirements.txt. That might not be included in all docker base images, adding it to the requirements is safer.
2. Added a minimum version to Pandas. It's a good practice to have versions for all dependencies.

* Sort alphabetically

* Update requirements.txt

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-09 15:51:01 -07:00
Glenn Jocher 2993c3fa7a Add roboflow (#4956) 2021-09-27 19:00:05 -07:00
Glenn Jocher 3a822a22ce PIL.ImageDraw.text(anchor=...) removal, reduce to >=7.1.2 (#4842) 
* Unpin Pillow

* Update requirements.txt

* Update plots.py
 2021-09-17 13:02:37 +02:00
Glenn Jocher c3a93d783d Add TensorFlow formats to export.py (#4479) 
* Initial commit

* Remove unused export_torchscript return

* ROOT variable

* Add prefix to fcn arg

* fix ROOT

* check_yaml into run()

* interim fixes

* imgsz=(320, 320)

* Hardcode tf_raw_resize False

* Finish opt elimination

* Update representative_dataset_gen()

* Update export.py with TF methods

* SiLU and GraphDef fixes

* file_size() directory handling feature

* export fixes

* add lambda: to representative_dataset

* Detect training False default

* Fuse false for TF models

* Embed agnostic NMS arguments

* Remove lambda

* TensorFlow.js export success

* Add pb to Usage

* Add *_tfjs_model/ to ignore files

* prepend YOLOv5 to function headers

* Remove end --- comments

* parameterize tfjs export pb file

* update run() data default /ROOT

* update --include help

* update imports

* return ct_model

* Consolidate TFLite export

* pb prerequisite to tfjs

* TF modules CamelCase

* Remove exports from tf.py and cleanup

* pass agnostic NMS arguments

* CI

* CI

* ignore *_web_model/

* Add tensorflow to CI dependencies

* CI tensorflow-cpu

* Update requirements.txt

* Remove tensorflow check_requirement

* CI coreml tfjs

* export only onnx torchscript

* reorder exports torchscript first
 2021-09-12 15:52:24 +02:00
Glenn Jocher de44376d1b Create Annotator() class (#4591) 
* Add Annotator() class

* Download Arial

* 2x for loop

* Cleanup

* tuple 2 list

* max_size=1920

* bold logging results to

* tolist()

* im = annotator.im

* PIL save in detect.py

* Smart asarray in detect.py

* revert to cv2.imwrite

* Cleanup

* Return result asarray

* Add `Profile()` profiler

* CamelCase Timeout

* Resize after mosaic

* pillow>=8.0.0

* daemon imwrite

* Add cv2 support

* Remove plot_wh_methods and plot_one_box

* pil=False for hubconf.py annotations

* im.shape bug fix

* colorstr common.py

* join daemons

* Update t.daemon

* Removed daemon saving
 2021-08-29 16:46:13 +02:00