d271491cd7
Merge commit '5c91'
2024-12-18 14:27:42 +11:00
d25a07534c
rk opt for yolov5 base@rk_85c6d7d2
...
Signed-off-by: Randall Zhuo <randall.zhuo@rock-chips.com>
2023-12-19 16:00:51 +08:00
5ca8e822c8
Update export.py ( #11077 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-27 10:55:05 -08:00
7a972e86c4
Update .pre-commit-config.yaml ( #11009 )
...
* Update .pre-commit-config.yaml
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 __init__.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update .pre-commit-config.yaml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Pre-commit updates
* Pre-commit updates
---------
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>
2023-02-18 01:06:24 +01:00
a2de5c5bf6
Subprocess improvements ( #10973 )
...
* Use list-form arguments for subprocess.run calls where possible
Augments #10944
* Deduplicate curl code
* Avoid eval() to parse integer
---------
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-13 18:00:31 +04:00
d389840f66
Allow int8 quantization for export_tfjs ( #10948 )
...
* Allow int8 quantization for export_tfjs
--int8 param currently has no effect on export_tfjs. With this change,
` python export.py --weights ../path/to/best.pt --include tfjs --int8` will add the --quantize_uint8 param to the tensorflowjs_converter script, greatly reducing model size for web usage.
Signed-off-by: David Strahm <david.strahm@lambda-it.ch>
* Update Dockerfile
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
---------
Signed-off-by: David Strahm <david.strahm@lambda-it.ch>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-10 19:11:08 +04:00
064365d868
Update parse_opt() in export.py to work as in train.py ( #10789 )
...
Update parse_opt() to work as in train.py
Change parse_opt() be able to use parse_known_args(), same as in train.py, so export.main() can be called from other script without error.
e.g.:
from yolov5 import export
opt = export.parse_opt(True)
opt.weights = <model_path>
opt.include = ("torchscript", "onnx")
opt.data = <data>
opt.imgsz = [<height>, <width>]
export.main(opt)
Signed-off-by: Johan Bergman <35481994+duran67@users.noreply.github.com>
Signed-off-by: Johan Bergman <35481994+duran67@users.noreply.github.com>
2023-01-20 23:49:43 +01:00
b2a0f1cdc5
Update `onnx>=1.12.0` ( #10526 )
2022-12-18 20:06:01 +01:00
b2f94e8c35
Update to ONNX opset 17 ( #10522 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-17 12:26:57 +01:00
185d475d93
Add DNN warning comment ( #10368 )
...
Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-01 13:01:46 -08:00
350e8eb69e
Fix SegmentationModel Usage ( #10303 )
...
Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-26 08:33:43 -08:00
54f49fa581
Add TFLite Metadata to TFLite and Edge TPU models ( #9903 )
...
* added embedded meta data to tflite models
* added try block for inference
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* refactored tfite meta data into separate function
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Creat tmp file in /tmp
* Update export.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update export.py
* Update export.py
* Update export.py
* Update export.py
* Update common.py
* Update export.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update common.py
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-25 17:53:22 +02:00
3b1a9d22a4
Fix OpenVINO Usage example ( #9874 )
...
* Fix OpenVINO Usage example
* Fix OpenVINO Usage example
2022-10-20 19:54:07 +02:00
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
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
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
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
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
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
fda8aa551d
TensorFlow SegmentationModel support ( #9472 )
...
* TensorFlow SegmentationModel support
* TensorFlow SegmentationModel support
* TensorFlow SegmentationModel support
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* TFLite fixes
* GraphDef fixes
* 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-18 19:52:46 +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
ca9c993d6c
Standardize warnings with `WARNING ⚠️ ...` ( #9467 )
...
* Standardize warnings
* [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 16:15:25 +02:00
dc42e6ef22
TensorRT SegmentationModel fix ( #9465 )
...
* TensorRT SegmentationModel fix
* TensorRT SegmentationModel fix
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* TensorRT SegmentationModel fix
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* TensorRT SegmentationModel fix
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* TensorRT SegmentationModel fix
* TensorRT SegmentationModel fix
* fix
* sort output names
* 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-18 14:45:08 +02:00
db06f495db
AutoUpdate TensorFlow in export.py ( #9447 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-16 20:44:56 +02:00
03f2ca8eff
Fix TensorRT exports to ONNX opset 12 ( #9441 )
...
* Fix TensorRT exports to ONNX opset 12
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-16 12:31:43 +02:00
1323b48053
Remove `.train()` mode exports ( #9429 )
...
* Remove `.train()` mode exports
No common use cases.
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-15 19:05:10 +02:00
4e8504abd9
PaddlePaddle Usage examples ( #9358 )
2022-09-10 13:25:01 +03:00
e3e5122f82
Add PaddlePaddle export and inference ( #9240 )
...
* Add PaddlePaddle Model Export
Test on Yolov5 DockerEnviroment with paddlepaddle-gpu v2.2
Signed-off-by: Katteria <39751846+kisaragychihaya@users.noreply.github.com>
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup Paddle Export
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 common.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update export.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 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>
* Use PyTorch2Paddle
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
* Paddle no longer requires ONNX
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>
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update benchmarks.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Add inference code of PaddlePaddle
Signed-off-by: Katteria <39751846+kisaragychihaya@users.noreply.github.com>
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update common.py
Signed-off-by: Katteria <39751846+kisaragychihaya@users.noreply.github.com>
* Update common.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Add paddlepaddle-gpu install if cuda
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 common.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>
Signed-off-by: Katteria <39751846+kisaragychihaya@users.noreply.github.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-09-10 12:20:46 +03:00
e9ddc5b527
Update `check_requirements(args, cmds='')` ( #9355 )
...
* Update `check_requirements(args, cmds='')`
* [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-10 12:00:16 +03:00
24bf9cceb4
Update `check_requirements()` single install ( #9353 )
...
* Update `check_requirements()` single install
Faster install and better conflict resolution with single installation
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update
* Update
* Update
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-10 11:11:56 +03:00
4a37381ee8
Add ClassificationModel TF export assert ( #9226 )
...
* Add ClassificationModel TF export assert
Export to TF not yet supported, warning alerts users.
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-08-30 16:18:01 +02:00
85d7ae2195
Rename onnx_dynamic -> dynamic ( #9168 )
2022-08-26 14:34:28 +02:00
d0fa0042bd
New `@try_export` decorator ( #9096 )
...
* New export decorator
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* New export decorator
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* rename fcn to func
* rename to @try_export
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-23 13:06:33 +02:00
06831aa9e9
Improved Usage example docstrings ( #9075 )
...
* Updated Usage examples
* Update detect.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>
2022-08-22 01:06:29 +02:00
93f63ee33f
Refactor for simplification 2 ( #9055 )
...
* Refactor for simplification 2
* Update __init__.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-21 02:00:35 +02:00
20049be2e7
EMA FP32 assert classification bug fix ( #9016 )
...
* Return EMA float on classification val
* verbose val fix
* EMA check
2022-08-18 14:06:15 +02:00
d3ea0df8b9
New YOLOv5 Classification Models ( #8956 )
...
* Update
* Logger step fix: Increment step with epochs (#8654 )
* enhance
* revert
* allow training from scratch
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update --img argument from train.py
single line
* fix image size from 640 to 128
* suport custom dataloader and augmentation
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* format
* Update dataloaders.py
* Single line return, single line comment, remove unused argument
* address PR comments
* fix spelling
* don't augment eval set
* use fstring
* update augmentations.py
* new maning convention for transforms
* reverse if statement, inline ops
* reverse if statement, inline ops
* updates
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* update dataloaders
* Remove additional if statement
* Remove is_train as redundant
* Cleanup
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup2
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update classifier.py
* Update augmentations.py
* fix: imshow clip warning
* update
* Revert ToTensorV2 removal
* Update classifier.py
* Update normalize values, revert uint8
* normalize image using cv2
* remove dedundant comment
* Update classifier.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* replace print with logger
* commit steps
* [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>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Allow logging models from GenericLogger (#8676 )
* enhance
* revert
* allow training from scratch
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update --img argument from train.py
single line
* fix image size from 640 to 128
* suport custom dataloader and augmentation
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* format
* Update dataloaders.py
* Single line return, single line comment, remove unused argument
* address PR comments
* fix spelling
* don't augment eval set
* use fstring
* update augmentations.py
* new maning convention for transforms
* reverse if statement, inline ops
* reverse if statement, inline ops
* updates
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* update dataloaders
* Remove additional if statement
* Remove is_train as redundant
* Cleanup
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup2
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update classifier.py
* Update augmentations.py
* fix: imshow clip warning
* update
* Revert ToTensorV2 removal
* Update classifier.py
* Update normalize values, revert uint8
* normalize image using cv2
* remove dedundant comment
* Update classifier.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* replace print with logger
* commit steps
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* support final model logging
* update
* update
* update
* update
* remove curses
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update classifier.py
* Update __init__.py
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>
* Update
* Update
* Update
* Update
* Update dataset download
* Update dataset download
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Pass imgsz to classify_transforms()
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Cos scheduler
* Cos scheduler
* Remove unused args
* Update
* Add seed
* Add seed
* Update
* Update
* Add run(), main()
* Merge master
* Merge master
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Create YOLOv5 BaseModel class (#8829 )
* Create BaseModel
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* fix
* Hub load device fix
* Update
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
* Add experiment
* Merge master
* Attach names
* weight decay = 1e-4
* weight decay = 5e-5
* update smart_optimizer console printout
* fashion-mnist fix
* Merge master
* Update Table
* Update Table
* Remove destroy process group
* add kwargs to forward()
* fuse fix for resnet50
* nc, names fix for resnet50
* nc, names fix for resnet50
* ONNX CPU inference fix
* revert
* cuda
* if augment or visualize
* if augment or visualize
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* New smart_inference_mode()
* Update README
* Refactor into /classify dir
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* reset defaults
* reset defaults
* fix gpu predict
* warmup
* ema half fix
* spacing
* remove data
* remove cache
* remove denormalize
* save run settings
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* verbose false on initial plots
* new save_yaml() function
* Update ci-testing.yml
* Path(data) CI fix
* Separate classification CI
* fix val
* fix val
* fix val
* smartCrossEntropyLoss
* skip validation on hub load
* autodownload with working dir root
* str(data)
* Dataset usage example
* im_show normalize
* im_show normalize
* add imagenet simple names to multibackend
* Add validation speeds
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* 24-space names
* Update bash scripts
* Update permissions
* Add bash script arguments
* remove verbose
* TRT data fix
* names generator fix
* optimize if names
* update usage
* Add local loading
* Verbose=False
* update names printing
* Add Usage examples
* Add Usage examples
* Add Usage examples
* Add Usage examples
* named_children
* reshape_classifier_outputs
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* update
* update
* fix CI
* fix incorrect class substitution
* fix incorrect class substitution
* remove denormalize
* ravel fix
* cleanup
* update opt file printing
* update opt file printing
* update defaults
* add opt to checkpoint
* Add warning
* Add comment
* plot half bug fix
* Use NotImplementedError
* fix export shape report
* Fix TRT load
* cleanup CI
* profile comment
* CI fix
* Add cls models
* avoid inplace error
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Fix usage examples
* Update README
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-17 11:59:01 +02:00
dc38cd03f4
New `smart_inference_mode()` conditional decorator ( #8957 )
...
New smart_inference_mode()
2022-08-13 20:38:51 +02:00
4bb305275a
Edge TPU add `--search_delegate` fix ( #8902 )
...
Resolves https://github.com/ultralytics/yolov5/issues/8842
2022-08-08 17:51:28 +02:00
628c05ca6f
export.py replace `check_file` -> `check_yaml` ( #8852 )
...
* export.py replace `check_file` -> `check_yaml`
* [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-08-03 23:38:36 +02:00
6884da3a32
Add check_file(data) i.e. `--data coco128.yaml` ( #8851 )
...
* Add check_file(data) i.e. `--data coco128.yaml`
* [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-08-03 23:32:31 +02:00
587a3a37c5
Dynamic batch size support for TensorRT ( #8526 )
...
* Dynamic batch size support for TensorRT
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update export.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Fix optimization profile when batch size is 1
* Warn users if they use batch-size=1 with dynamic
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* More descriptive assertion error
* Fix syntax
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* pre-commit formatting sucked
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update export.py
Co-authored-by: Colin Wong <noreply@brains4drones.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-07-29 13:51:16 +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
f4b05680f8
Assert `--optimize` not used with cuda device ( #8569 )
2022-07-15 16:01:01 +02:00
c215e87393
XML export `--half` fix ( #8522 )
...
Improved error reporting for https://github.com/ultralytics/yolov5/issues/8519
2022-07-08 13:49:20 +02:00
63ba0cb18a
Add `--half` arguments to export.py Usage examples ( #8516 )
2022-07-08 00:46:56 +02:00
3e54651fca
Add `--hard-fail` argument to benchmarks for CI errors ( #8513 )
...
* Add `--hard-fail` list argument to benchmarks for CI
Will cause CI to fail on a benchmark failure for given indices.
* Update ci-testing.yml
* Attempt Failure (CI should fail)
* Update benchmarks.py
* Update export.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update benchmarks.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update ci-testing.yml
* Update benchmarks.py
* Update benchmarks.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-07-07 23:41:34 +02:00
0c1324067c
Fix ONNX `--dynamic` export on GPU ( #8378 )
...
* Fix ONNX `--dynamic` export on GPU
Patch forces --dynamic export model and image to CPU. Resolves bug raised in https://github.com/ultralytics/yolov5/issues/8377
* Update export.py
2022-06-28 15:22:15 +02:00
6dd6aea086
Fix FP32 TensorRT model export ( #8046 )
...
Fixed FP32 TRT model export
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-05-31 12:41:09 +02:00
945579699a
Remove `formats` variable to avoid `pd` conflict ( #7993 )
...
* Remove `formats` variable to avoid `pd` conflict
* Update export.py
2022-05-26 16:07:58 +02:00
Edward Yang
Randall Zhuo
Glenn Jocher
Aarni Koskela
David Strahm
Johan Bergman
paradigm
曾逸夫（Zeng Yifu）
Katteria
Colin Wong
daquexian
Sahil Chachra