b74dd4ba4f
Add `--int8` argument ( #4799 )
...
* Add `--int8` argument
* parents[0] bug fix
* Fix order
2021-09-15 11:33:46 +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
c47be26f34
Replace `path.absolute()` with `path.resolve()` ( #4763 )
2021-09-11 22:46:33 +02:00
2d9411dbb8
Scope `onnx-simplifier` requirements check ( #4730 )
...
* Changed onnx-simplifier check behavior
Export.py has been updated to check for onnx-simplifier requirement only when the --simplify argument is added.
Allows for better flexibility and one less requirement if simplify is not needed.
* Fix single-element tuples
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-09-09 16:49:10 +02:00
8b18b66304
ONNX opset 13 ( #4566 )
2021-08-27 16:00:39 +02:00
d1182c4f29
`check_requirements(('coremltools',))` ( #4478 )
...
* `check_requirements(('coremltools',))`
* Update ci-testing.yml
* Update ci-testing.yml
2021-08-18 21:16:57 +02:00
24bea5e4b7
Standardize headers and docstrings ( #4417 )
...
* Implement new headers
* Reformat 1
* Reformat 2
* Reformat 3 - math
* Reformat 4 - yaml
2021-08-14 21:17:51 +02:00
57c342a6a5
Simplify ONNX inference command ( #4405 )
2021-08-13 13:22:13 +02:00
2d99063201
Feature `python train.py --cache disk` ( #4049 )
...
* Add cache-on-disk and cache-directory to cache images on disk
* Fix load_image with cache_on_disk
* Add no_cache flag for load_image
* Revert the parts('logging' and a new line) that do not need to be modified
* Add the assertion for shapes of cached images
* Add a suffix string for cached images
* Fix boundary-error of letterbox for load_mosaic
* Add prefix as cache-key of cache-on-disk
* Update cache-function on disk
* Add psutil in requirements.txt
* Update train.py
* Cleanup1
* Cleanup2
* Skip existing npy
* Include re-space
* Export return character fix
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-08-02 18:47:24 +02:00
388016e9e3
Update 'results saved to' string ( #4275 )
2021-08-02 15:48:53 +02:00
6e4358f3f3
Add `export.py` ONNX inference suggestion ( #4146 )
2021-07-25 13:46:16 +02:00
264be1a616
Rename `opset_version` to `opset` ( #4135 )
2021-07-24 13:19:12 +02:00
3bef77f5cb
Addition refactor `export.py` ( #4089 )
...
* Addition refactor `export.py`
* Update export.py
2021-07-20 18:42:27 +02:00
442a7abdf2
Refactor `export.py` ( #4080 )
...
* Refactor `export.py`
* cleanup
* Update check_requirements()
* Update export.py
2021-07-20 13:21:52 +02:00
0e2d0d54d7
Fix typo ( #3729 )
2021-06-22 13:33:38 +02:00
1f69d12591
Update 4 main ops for paths and .run() ( #3715 )
...
* Add yolov5/ to path
* rename functions to run()
* cleanup
* rename fix
* CI fix
* cleanup find models/export.py
2021-06-21 17:25:04 +02:00
Glenn Jocher
Zegorax
junji hashimoto
fcakyon