Glenn Jocher
d5966c93f1
Namespace `VERBOSE` env variable to `YOLOv5_VERBOSE` ( #6428 )
* Verbose updates
* Verbose updates
2 years ago
Glenn Jocher
ff8646cdea
Update workflows ( #6427 )
* Workflow updates
* quotes fix
* best to weights fix
2 years ago
Motoki Kimura
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
Jonathan Samelson
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
Glenn Jocher
cfecd903a3
Update `greetings.yaml` email address ( #6412 )
* Update `greetings.yaml` email address
* Update greetings.yml
2 years ago
Glenn Jocher
482af479c0
Add `detect.py` GIF video inference ( #6410 )
* Add detect.py GIF video inference
* Cleanup
2 years ago
Matthias
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
Glenn Jocher
c43439aa31
Add `albumentations` to Dockerfile ( #6392 )
2 years ago
Glenn Jocher
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
Glenn Jocher
1b41a1d059
FROM nvcr.io/nvidia/pytorch:21.12-py3 ( #6377 )
2 years ago
Ayush Chaurasia
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
sitecao
e1893c894a
`DEVICE_COUNT` instead of `WORLD_SIZE` to calculate `nw` ( #6324 )
2 years ago
imyhxy
4e841b9b16
Reuse `de_parallel()` rather than `is_parallel()` ( #6354 )
2 years ago
johnk2hawaii
9708cf56ea
New environment variable `VERBOSE` ( #6353 )
New environment variable `VERBOSE`
2 years ago
Glenn Jocher
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
Glenn Jocher
e5219099cd
Created using Colaboratory
2 years ago
Glenn Jocher
0cf932bf63
`export.py` return exported files/dirs ( #6343 )
* `export.py` return exported files/dirs
* Path to str
2 years ago
Glenn Jocher
e2e95b2d8e
TensorRT `assert im.device.type != 'cpu'` on export ( #6340 )
* TensorRT `assert im.device.type != 'cpu'` on export
* Update export.py
2 years ago
Glenn Jocher
fd55271c04
Console corrupted -> corrupt ( #6338 )
* Console corrupted -> corrupt
Minor style changes.
* Update export.py
2 years ago
Glenn Jocher
3119b2f27c
Remove `dataset_stats()` autodownload capability ( #6303 )
* Remove `dataset_stats()` autodownload capability
@kalenmike security update per Slack convo
* Update datasets.py
2 years ago
Otfot
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
Glenn Jocher
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
Henry
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
Glenn Jocher
e7bf38277f
Fix `device` count check ( #6290 )
* Fix device count check()
* Update torch_utils.py
* Update torch_utils.py
* Update hubconf.py
2 years ago
Glenn Jocher
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
Glenn Jocher
80473a6551
Update `export.py` with Detect, Validate usages ( #6280 )
2 years ago
Glenn Jocher
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
Glenn Jocher
b3eaf5008b
TensorRT pip install
2 years ago
Jinwoong Yoo
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
Glenn Jocher
6865d19a92
TensorRT 7 export fix ( #6235 )
2 years ago
Glenn Jocher
33a67b4918
Update P2-P7 `models/hub` variants ( #6230 )
* Update p2-p7 `models/hub` variants
* Update common.py
* AutoAnchor camelcase corrections
2 years ago
Glenn Jocher
ad565e31d2
Update README speed reproduction command ( #6228 )
2 years ago
Glenn Jocher
f80c463010
Attempt `edgetpu-compiler` autoinstall ( #6223 )
* Attempt `edgetpu-compiler` autoinstall
Attempt to install edgetpu-compiler dependency if missing on Linux.
* Update export.py
* Update export.py
2 years ago
Glenn Jocher
b4ac3df6ff
Add `edgetpu_compiler` checks ( #6218 )
* Add `edgetpu_compiler` checks
* Update export.py
* Update export.py
* Update export.py
* Update export.py
* Update export.py
* Update export.py
2 years ago
Glenn Jocher
00d7b97869
TFLite `--int8` 'flatbuffers==1.12' fix 2 ( #6217 )
* TFLite `--int8` 'flatbuffers==1.12' fix 2
Reorganizes #6216 fix to update before `tensorflow` import so no restart required.
* Update export.py
2 years ago
Glenn Jocher
8125ec5d42
TFLite `--int8` 'flatbuffers==1.12' fix ( #6216 )
* TFLite `--int8` 'flatbuffers==1.12' fix
Temporary workaround for TFLite INT8 export.
* Update export.py
* Update export.py
2 years ago
Glenn Jocher
5402753a53
Edge TPU compiler comment ( #6196 )
* Edge TPU compiler comment
* 7 to 8 fix
2 years ago
Glenn Jocher
9e9219fe17
Fix `nan`-robust stream FPS ( #6198 )
Fix for Webcam stop working suddenly (Issue #6197 )
2 years ago
Glenn Jocher
b5b56a3c88
Add CoreML inference ( #6195 )
* Add Apple CoreML inference
* Cleanup
2 years ago
Glenn Jocher
7b31a531b4
Add `tensorrt>=7.0.0` checks ( #6193 )
* Add `tensorrt>=7.0.0` checks
* Update export.py
* Update common.py
* Update export.py
2 years ago
imyhxy
a2f4a1799b
TensorRT 7 `anchor_grid` compatibility fix ( #6185 )
* fix: TensorRT 7 incompatiable
* Add comment
* Add if: else and comment
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
Yin Rong
fb839298a1
Fix TorchScript on mobile export ( #6183 )
* fix export of TorchScript on mobile
* Cleanup
Co-authored-by: yinrong <yinrong@xiaomi.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
Glenn Jocher
5bd6a97b18
Global export format sort ( #6182 )
* Global export sort
* Cleanup
2 years ago
Glenn Jocher
7cad6597bb
Ignore `*_openvino_model/` dir ( #6180 )
2 years ago
Glenn Jocher
63a4d862aa
Add OpenVINO inference ( #6179 )
2 years ago
Glenn Jocher
b4a29b5a8d
Update NMS `max_wh=7680` for 8k images ( #6178 )
2 years ago
pre-commit-ci[bot]
5344e54da6
[pre-commit.ci] pre-commit suggestions ( #6177 )
updates:
- [github.com/pre-commit/pre-commit-hooks: v4.0.1 → v4.1.0](https://github.com/pre-commit/pre-commit-hooks/compare/v4.0.1...v4.1.0 )
- [github.com/asottile/pyupgrade: v2.23.1 → v2.31.0](https://github.com/asottile/pyupgrade/compare/v2.23.1...v2.31.0 )
- [github.com/PyCQA/isort: 5.9.3 → 5.10.1](https://github.com/PyCQA/isort/compare/5.9.3...5.10.1 )
- [github.com/PyCQA/flake8: 3.9.2 → 4.0.1](https://github.com/PyCQA/flake8/compare/3.9.2...4.0.1 )
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
968e30065a
Update greetings.yml ( #6165 )
2 years ago
Glenn Jocher
ec4b6dd2a3
Update export format docstrings ( #6151 )
* Update export documentation
* Cleanup
* Update export.py
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update README.md
* Update README.md
* Update README.md
* Update train.py
* Update train.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
bilzard
e1dc894364
Enable AdamW optimizer ( #6152 )
2 years ago