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
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
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
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
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
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
9559601b9a
Update requirements.txt with tf-cpu and tf-aarch64 ( #8807 )
2022-07-31 14:54:55 +02:00
3e858633b2
Revert protobuf<=3.20.1 ( #8742 )
...
Resolve #8012 (again)
2022-07-27 17:27:44 +02:00
d5116bbe9c
coremltools>=5.2 for CoreML export ( #8725 )
2022-07-26 17:50:49 +02:00
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
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
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
cf28dda366
Update `scipy>=1.7.3` ( #8595 )
2022-07-16 15:54:34 +02:00
72a81e7a1c
Update requirements.txt comment spacing ( #8562 )
2022-07-15 16:01:29 +02:00
574ceedfc5
Add `thop>=0.1.0` ( #8558 )
2022-07-12 14:49:54 +02:00
fbd3020525
Add TensorRT dependencies ( #8553 )
...
Update requirements.txt
2022-07-12 13:19:25 +02:00
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
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
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
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
1dcb774998
Update requirements.txt with Usage example ( #7966 )
...
Update requirements.txt
2022-05-25 01:40:17 +02:00
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
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
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
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
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
2993c3fa7a
Add `roboflow` ( #4956 )
2021-09-27 19:00:05 -07:00
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
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
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
808bcad3bb
Add TensorFlow and TFLite export ( #1127 )
...
* Add models/tf.py for TensorFlow and TFLite export
* Set auto=False for int8 calibration
* Update requirements.txt for TensorFlow and TFLite export
* Read anchors directly from PyTorch weights
* Add --tf-nms to append NMS in TensorFlow SavedModel and GraphDef export
* Remove check_anchor_order, check_file, set_logging from import
* Reformat code and optimize imports
* Autodownload model and check cfg
* update --source path, img-size to 320, single output
* Adjust representative_dataset
* Put representative dataset in tfl_int8 block
* detect.py TF inference
* weights to string
* weights to string
* cleanup tf.py
* Add --dynamic-batch-size
* Add xywh normalization to reduce calibration error
* Update requirements.txt
TensorFlow 2.3.1 -> 2.4.0 to avoid int8 quantization error
* Fix imports
Move C3 from models.experimental to models.common
* Add models/tf.py for TensorFlow and TFLite export
* Set auto=False for int8 calibration
* Update requirements.txt for TensorFlow and TFLite export
* Read anchors directly from PyTorch weights
* Add --tf-nms to append NMS in TensorFlow SavedModel and GraphDef export
* Remove check_anchor_order, check_file, set_logging from import
* Reformat code and optimize imports
* Autodownload model and check cfg
* update --source path, img-size to 320, single output
* Adjust representative_dataset
* detect.py TF inference
* Put representative dataset in tfl_int8 block
* weights to string
* weights to string
* cleanup tf.py
* Add --dynamic-batch-size
* Add xywh normalization to reduce calibration error
* Update requirements.txt
TensorFlow 2.3.1 -> 2.4.0 to avoid int8 quantization error
* Fix imports
Move C3 from models.experimental to models.common
* implement C3() and SiLU()
* Fix reshape dim to support dynamic batching
* Add epsilon argument in tf_BN, which is different between TF and PT
* Set stride to None if not using PyTorch, and do not warmup without PyTorch
* Add list support in check_img_size()
* Add list input support in detect.py
* sys.path.append('./') to run from yolov5/
* Add int8 quantization support for TensorFlow 2.5
* Add get_coco128.sh
* Remove --no-tfl-detect in models/tf.py (Use tf-android-tfl-detect branch for EdgeTPU)
* Update requirements.txt
* Replace torch.load() with attempt_load()
* Update requirements.txt
* Add --tf-raw-resize to set half_pixel_centers=False
* Add --agnostic-nms for TF class-agnostic NMS
* Cleanup after merge
* Cleanup2 after merge
* Cleanup3 after merge
* Add tf.py docstring with credit and usage
* pb saved_model and tflite use only one model in detect.py
* Add use cases in docstring of tf.py
* Remove redundant `stride` definition
* Remove keras direct import
* Fix `check_requirements(('tensorflow>=2.4.1',))`
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-08-17 13:18:16 +02:00
7fdcc77bf4
Albumentations >= 1.0.3 ( #4068 )
2021-07-19 13:23:19 +02:00
a544d59f52
Update requirements.txt `albumentations>=1.0.2` ( #3972 )
2021-07-11 19:07:42 +02:00
33202b7f0b
YOLOv5 + Albumentations integration ( #3882 )
...
* Albumentations integration
* ToGray p=0.01
* print confirmation
* create instance in dataloader init method
* improved version handling
* transform not defined fix
* assert string update
* create check_version()
* add spaces
* update class comment
2021-07-05 18:01:54 +02:00
ef0b5c9d29
On-demand `pycocotools` pip install ( #3547 )
2021-06-09 11:22:21 +02:00
f3c3d2ce5d
Merge `develop` branch into `master` ( #3518 )
...
* update ci-testing.yml (#3322 )
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
* update ci-testing.yml (#3322 )
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
* Enable direct `--weights URL` definition (#3373 )
* Enable direct `--weights URL` definition
@KalenMike this PR will enable direct --weights URL definition. Example use case:
```
python train.py --weights https://storage.googleapis.com/bucket/dir/model.pt
```
* cleanup
* bug fixes
* weights = attempt_download(weights)
* Update experimental.py
* Update hubconf.py
* return bug fix
* comment mirror
* min_bytes
* Update tutorial.ipynb (#3368 )
add Open in Kaggle badge
* `cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379 )
* Update datasets.py
* comment
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* COCO evolution fix (#3388 )
* COCO evolution fix
* cleanup
* update print
* print fix
* Create `is_pip()` function (#3391 )
Returns `True` if file is part of pip package. Useful for contextual behavior modification.
```python
def is_pip():
# Is file in a pip package?
return 'site-packages' in Path(__file__).absolute().parts
```
* Revert "`cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379 )" (#3395 )
This reverts commit 21a9607e00#3422 )
* Update README.md
* Changing FLOPS to FLOPs.
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
* Parse URL authentication (#3424 )
* Parse URL authentication
* urllib.parse.unquote()
* improved error handling
* improved error handling
* remove %3F
* update check_file()
* Add FLOPs title to table (#3453 )
* Suppress jit trace warning + graph once (#3454 )
* Suppress jit trace warning + graph once
Suppress harmless jit trace warning on TensorBoard add_graph call. Also fix multiple add_graph() calls bug, now only on batch 0.
* Update train.py
* Update MixUp augmentation `alpha=beta=32.0` (#3455 )
Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye
* Add `timeout()` class (#3460 )
* Add `timeout()` class
* rearrange order
* Faster HSV augmentation (#3462 )
remove datatype conversion process that can be skipped
* Add `check_git_status()` 5 second timeout (#3464 )
* Add check_git_status() 5 second timeout
This should prevent the SSH Git bug that we were discussing @KalenMike
* cleanup
* replace timeout with check_output built-in timeout
* Improved `check_requirements()` offline-handling (#3466 )
Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).
* Add `output_names` argument for ONNX export with dynamic axes (#3456 )
* Add output names & dynamic axes for onnx export
Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3
* use first output only + cleanup
Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Revert FP16 `test.py` and `detect.py` inference to FP32 default (#3423 )
* fixed inference bug ,while use half precision
* replace --use-half with --half
* replace space and PEP8 in detect.py
* PEP8 detect.py
* update --half help comment
* Update test.py
* revert space
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Add additional links/resources to stale.yml message (#3467 )
* Update stale.yml
* cleanup
* Update stale.yml
* reformat
* Update stale.yml HUB URL (#3468 )
* Stale `github.actor` bug fix (#3483 )
* Explicit `model.eval()` call `if opt.train=False` (#3475 )
* call model.eval() when opt.train is False
call model.eval() when opt.train is False
* single-line if statement
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* check_requirements() exclude `opencv-python` (#3495 )
Fix for 3rd party or contrib versions of installed OpenCV as in https://github.com/ultralytics/yolov5/issues/3494 .
* Earlier `assert` for cpu and half option (#3508 )
* early assert for cpu and half option
early assert for cpu and half option
* Modified comment
Modified comment
* Update tutorial.ipynb (#3510 )
* Reduce test.py results spacing (#3511 )
* Update README.md (#3512 )
* Update README.md
Minor modifications
* 850 width
* Update greetings.yml
revert greeting change as PRs will now merge to master.
Co-authored-by: Piotr Skalski <SkalskiP@users.noreply.github.com>
Co-authored-by: SkalskiP <piotr.skalski92@gmail.com>
Co-authored-by: Peretz Cohen <pizzaz93@users.noreply.github.com>
Co-authored-by: tudoulei <34886368+tudoulei@users.noreply.github.com>
Co-authored-by: chocosaj <chocosaj@users.noreply.github.com>
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Sam_S <SamSamhuns@users.noreply.github.com>
Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: edificewang <609552430@qq.com>
2021-06-08 10:22:10 +02:00
ee24ae11ea
Comment Cython ( #3297 )
...
Ultralytics REST API fix.
2021-05-23 17:53:54 +02:00
7b863a34dc
Add Cython ( #3217 )
...
Cython required by pycocotools in certain cases, i.e. https://github.com/cocodataset/cocoapi/issues/172
2021-05-19 19:58:53 +02:00
b098600763
Update requirements.txt `onnx>=1.9.0` ( #3143 )
2021-05-12 19:56:12 +02:00
fca16dc4b3
Remove Cython from requirements.txt ( #2604 )
...
Cython should be a dependency of the remaining packages in requirements.txt, so should be installed anyway even if not a direct requirement.
2021-03-25 21:48:25 +01:00
e5b0200cd2
Update tensorboard>=2.4.1 ( #2576 )
...
* Update tensorboard>=2.4.1
Update tensorboard version to attempt to address https://github.com/ultralytics/yolov5/issues/2573 (tensorboard logging fail in Docker image).
* cleanup
2021-03-24 01:29:00 +01:00
ace3e02e40
Update requirements.txt ( #2173 )
2021-02-09 22:03:29 -08:00
3551b072b3
Docker pyYAML>=5.3.1 fix ( #2031 )
2021-01-24 11:54:15 -08:00
c76c607265
PyYAML==5.4.1 ( #2030 )
2021-01-24 10:05:40 -08:00
9a3da79b4a
Update requirements.txt ( #2021 )
...
* Update requirements.txt
* Update ci-testing.yml
* Update hubconf.py
2021-01-23 12:51:04 -08:00
ab0db8d104
Fix: Seaborn error due to histplot ( #1740 )
2020-12-19 09:30:43 -08:00
fa8f1fb0e9
Simplify autoshape() post-process ( #1653 )
...
* Simplify autoshape() post-process
* cleanup
* cleanup
2020-12-09 07:44:06 -08:00
0a3ff71ae0
Confusion matrix ( #1474 )
...
* initial commit
* add plotting
* matrix to cpu
* bug fix
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* seaborn pandas to requirements.txt
* seaborn pandas to requirements.txt
* update wandb plotting
* remove pandas
* if plots
* if plots
* if plots
* if plots
* if plots
* initial commit
* add plotting
* matrix to cpu
* bug fix
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* seaborn pandas to requirements.txt
* seaborn pandas to requirements.txt
* update wandb plotting
* remove pandas
* if plots
* if plots
* if plots
* if plots
* if plots
* Cat apriori to autolabels
* cleanup
2020-11-23 15:27:14 +01:00
93fc607f07
PyTorch Hub and autoShape update ( #1415 )
2020-11-16 23:14:07 +01:00
177f39e98e
onnx>=1.8.0
2020-11-08 19:41:54 +01:00
Glenn Jocher
Victor Sonck
daquexian
曾逸夫（Zeng Yifu）
Junya Morioka
dependabot[bot]
hamid mohammadi
Jiacong Fang
Durgesh