8fcdf3b60b
Fixing minor multi-streaming issues with TensoRT engine ( #6504 )
* Update batch-size in model.warmup() + indentation for logging inference results
* These changes are in response to PR comments
* [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>
2 years ago
079b36d72b
Edge TPU `tf.lite.experimental.load_delegate` fix ( #6536 )
* Edge TPU `tf.lite.experimental.load_delegate` fix
Fix attempt for #6535
* [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>
2 years ago
9bc72a3ac2
Edge TPU export 'list index out of range' fix ( #6533 )
2 years ago
8d05716b66
Edge TPU compiler `sudo` fix ( #6531 )
* Edge TPU compiler sudo fix
Allows for auto-install of Edge TPU compiler on non-sudo systems like the YOLOv5 Docker image.
@kalenmike
* Update export.py
* Update export.py
* Update export.py
2 years ago
a82292ec53
Social icons after text ( #6473 )
* Social icons after text
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update README.md
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
c3e599cfda
Update val_batch*.jpg for Chinese fonts ( #6526 )
* Update plots for Chinese fonts
* make is_chinese() non-str safe
* Add global FONT
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update general.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
cb40c9afda
TODO issues exempt from stale action ( #6530 )
2 years ago
19e0208fc9
Update hyp.scratch-high.yaml ( #6525 )
Update `lrf: 0.1`, tested on YOLOv5x6 to 55.0 mAP@0.5:0.95, slightly higher than current.
2 years ago
b73c62ebc5
W&B: Remember batchsize on resuming ( #6512 )
* log best.pt metrics at train end
* update
* Update __init__.py
* flush callbacks when using evolve
* remember batch size on resuming
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
4c40933266
Suppress export.run() TracerWarnings ( #6499 )
Suppresses warnings when calling export.run() directly, not just CLI python export.py.
Also adds Requirements examples for CPU and GPU backends
2 years ago
842d049e1b
Suppress `torch.jit.TracerWarning` on export ( #6498 )
* Suppress torch.jit.TracerWarning
TracerWarnings can be safely ignored.
* Cleanup
2 years ago
77977e0791
CoreML inference fix `list()` -> `sorted()` ( #6496 )
2 years ago
5e4ff195b2
Improved `export.py` usage examples ( #6495 )
* Improved `export.py` usage examples
* Cleanup
2 years ago
b884ea36c4
Simplify TF normalized to pixels ( #6494 )
2 years ago
6445a8137e
Resolve dataset paths ( #6489 )
2 years ago
7539cd75c3
Add Product Hunt social media icon ( #6464 )
* Social media icons update
* fix URL
* Update README.md
2 years ago
d8b5beb0b0
Fix2 `select_device()` for Multi-GPU ( #6461 )
* Fix2 select_device() for Multi-GPU
* Cleanup
* Cleanup
* Simplify error message
* Improve assert
* Update torch_utils.py
2 years ago
856d4e5733
Fix `select_device()` for Multi-GPU ( #6434 )
* Fix `select_device()` for Multi-GPU
Possible fix for https://github.com/ultralytics/yolov5/issues/6431
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
2 years ago
fe7de6a82d
Revert "Remove `dataset_stats()` autodownload capability ( #6303 )" ( #6442 )
This reverts commit 3119b2f27c
2 years ago
3b7ac28ed1
Add `*.asf` video support ( #6436 )
2 years ago
d5966c93f1
Namespace `VERBOSE` env variable to `YOLOv5_VERBOSE` ( #6428 )
* Verbose updates
* Verbose updates
2 years ago
ff8646cdea
Update workflows ( #6427 )
* Workflow updates
* quotes fix
* best to weights fix
2 years ago
16563ac5b5
Prefer `tflite_runtime` for TFLite inference if installed ( #6406 )
* import tflite_runtime if tensorflow not installed
* rename tflite to tfli
* Attempt tflite_runtime for all TFLite workflows
Also rename tfli to tfl
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
ed9bac8392
Rename logger from 'utils.logger' to 'yolov5' ( #6421 )
* Gave a more explicit name to the logger
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
cfecd903a3
Update `greetings.yaml` email address ( #6412 )
* Update `greetings.yaml` email address
* Update greetings.yml
2 years ago
482af479c0
Add `detect.py` GIF video inference ( #6410 )
* Add detect.py GIF video inference
* Cleanup
2 years ago
8efe97719c
Add `stop_training=False` flag to callbacks ( #6365 )
* New flag 'stop_training' in util.callbacks.Callbacks class to prematurely stop training from callback handler
* Removed most of the new checks, leaving only the one after calling 'on_train_batch_end'
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
c43439aa31
Add `albumentations` to Dockerfile ( #6392 )
2 years ago
bd815d48df
FROM nvcr.io/nvidia/pytorch:21.10-py3 ( #6379 )
21.12 generates dockerhub errors so rolling back to 21.10 with latest pytorch install. Not sure if this torch install will work on non-GPU dockerhub autobuild so this is an experiment.
2 years ago
1b41a1d059
FROM nvcr.io/nvidia/pytorch:21.12-py3 ( #6377 )
2 years ago
9bcc32a5bf
Flush callbacks when on `--evolve` ( #6374 )
* log best.pt metrics at train end
* update
* Update __init__.py
* flush callbacks when using evolve
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
e1893c894a
`DEVICE_COUNT` instead of `WORLD_SIZE` to calculate `nw` ( #6324 )
2 years ago
4e841b9b16
Reuse `de_parallel()` rather than `is_parallel()` ( #6354 )
2 years ago
9708cf56ea
New environment variable `VERBOSE` ( #6353 )
New environment variable `VERBOSE`
2 years ago
750c42e43e
`export.py` automatic `forward_export` ( #6352 )
* `export.py` automatic `forward_export`
* [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>
2 years ago
e5219099cd
Created using Colaboratory
2 years ago
0cf932bf63
`export.py` return exported files/dirs ( #6343 )
* `export.py` return exported files/dirs
* Path to str
2 years ago
e2e95b2d8e
TensorRT `assert im.device.type != 'cpu'` on export ( #6340 )
* TensorRT `assert im.device.type != 'cpu'` on export
* Update export.py
2 years ago
fd55271c04
Console corrupted -> corrupt ( #6338 )
* Console corrupted -> corrupt
Minor style changes.
* Update export.py
2 years ago
3119b2f27c
Remove `dataset_stats()` autodownload capability ( #6303 )
* Remove `dataset_stats()` autodownload capability
@kalenmike security update per Slack convo
* Update datasets.py
2 years ago
db1f83be63
Fix `train.py` parameter groups desc error ( #6318 )
* Fix `train.py` parameter groups desc error
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
436ffc417a
`select_device()` cleanup ( #6302 )
* `select_device()` cleanup
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
2 years ago
a1a9c6884c
Fixing bug multi-gpu training ( #6299 )
* Fixing bug multi-gpu training
This solves this issue: https://github.com/ultralytics/yolov5/issues/6297#issue-1103853348
* Update torch_utils.py for pep8
2 years ago
e7bf38277f
Fix `device` count check ( #6290 )
* Fix device count check()
* Update torch_utils.py
* Update torch_utils.py
* Update hubconf.py
2 years ago
af001349e4
Add `is_kaggle()` function ( #6285 )
* Add `is_kaggle()` function
Return True if environment is Kaggle Notebook.
* Remove root loggers only if is_kaggle() == True
* Update general.py
2 years ago
80473a6551
Update `export.py` with Detect, Validate usages ( #6280 )
2 years ago
f3085accd3
Enable ONNX `--half` FP16 inference ( #6268 )
* Enable ONNX ``--half` FP16 inference
* [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>
2 years ago
b3eaf5008b
TensorRT pip install
2 years ago
9b13a594e9
Fix `cmd` string on `tfjs` export ( #6243 )
* Fix cmd string on tfjs export
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
6865d19a92
TensorRT 7 export fix ( #6235 )
2 years ago
greg2451
Glenn Jocher
Ayush Chaurasia
toschi23
Motoki Kimura
Jonathan Samelson
Matthias
sitecao
imyhxy
johnk2hawaii
Otfot
Henry
Jinwoong Yoo