b292837e36
Fix ONNX export using --grid --simplify --dynamic simultaneously ( #2982 )
...
* Update yolo.py
* Update export.py
* fix export grid
* Update export.py, remove detect export attribute
* rearrange if order
* remove --grid, default inplace=False
* rename exp_dynamic to onnx_dynamic, comment
* replace bs with 1 in anchor_grid[i] index 0
* Update export.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-03 19:01:29 +02:00
e2a80c6c0f
Add support for FP16 (half) to export.py ( #3010 )
...
* Added support for fp16 (half) to export.py
* minimize code additions
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-02 22:42:33 +02:00
4200674a13
Add yolov5/ to sys.path() for *.py subdir exec ( #2949 )
...
* Add yolov5/ to sys.path() for *.py subdir exec
* Update export.py
2021-04-27 17:02:07 +02:00
1b1ab4cca2
Add file_size() function ( #2911 )
...
* Add file_size() function
* Update export.py
2021-04-24 01:31:11 +02:00
646386ff09
Update export.py for 2 dry runs ( #2910 )
...
* Update export.py for 2 dry runs
* Update export.py
2021-04-24 00:10:38 +02:00
a2a514dec8
Update export.py ( #2909 )
2021-04-23 23:50:02 +02:00
28db237639
Default optimize_for_mobile() on TorchScript models ( #2908 )
...
Per https://pytorch.org/tutorials/recipes/script_optimized.html this should improve performance on torchscript models (and maybe coreml models also since coremltools operates on a torchscript model input, though this still requires testing).
2021-04-23 21:21:58 +02:00
1df8c6c963
Fix ONNX dynamic axes export support with onnx simplifier, make onnx simplifier optional ( #2856 )
...
* Ensure dynamic export works succesfully, onnx simplifier optional
* Update export.py
* add dashes
Co-authored-by: Tim <tim.stokman@hal24k.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-20 13:54:03 +02:00
1f3e482bce
ONNX Simplifier ( #2815 )
...
* ONNX Simplifier
Add ONNX Simplifier to ONNX export pipeline in export.py. Will auto-install onnx-simplifier if onnx is installed but onnx-simplifier is not.
* Update general.py
2021-04-16 14:03:27 +02:00
0f395b3e3b
YOLOv5 v5.0 Release patch 1 ( #2764 )
...
* torch.jit.trace(model, img, strict=False)
* Update check_file()
* Update hubconf.py
* Update README.md
2021-04-11 23:11:43 +02:00
7a0a81fd1d
GPU export options ( #2297 )
...
* option for skip last layer and cuda export support
* added parameter device
* fix import
* cleanup 1
* cleanup 2
* opt-in grid
--grid will export with grid computation, default export will skip grid (same as current)
* default --device cpu
GPU export causes ONNX and CoreML errors.
Co-authored-by: Jan Hajek <jan.hajek@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-03-06 12:02:10 -08:00
95aefea493
Dynamic ONNX engine generation ( #2208 )
...
* add: dynamic onnx export
* delete: test onnx inference
* fix dynamic output axis
* Code reduction
* fix: dynamic output axes, dynamic input naming
* Remove fixed axes
Co-authored-by: Shivam Swanrkar <ss8464@nyu.edu>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-02-21 21:50:44 -08:00
6bd5e8bca7
nn.SiLU() export support ( #1713 )
2020-12-16 17:55:57 -08:00
d45e349a1e
Rearrange export input after checks ( #1118 )
...
img size checks are warnings rather than errors, so current implementation allows improperly formed model inputs.
2020-10-11 16:23:36 +02:00
883924d9dc
classifier, export, torch seed updates
2020-10-06 14:54:02 +02:00
a12698f4c9
Update ci-testing.yml reflecting sys.path additions ( #869 )
...
* Update ci-testing.yml
* Update yolo.py
* Update ci-testing.yml
* Update export.py
* Update yolo.py
2020-10-04 18:50:32 +02:00
d55c326dda
export.py folder sys.path addition ( #1052 )
2020-10-04 17:42:05 +02:00
d68179953d
Update export.py
2020-09-02 13:23:29 -07:00
1e7674d4dd
Update export.py --img-size comment
2020-08-25 23:07:22 -07:00
4fb8cb353f
Update export.py with v3.0 Hardswish() support ( #831 )
2020-08-24 21:59:26 -07:00
4d7f222f73
Update export.py with v3.0 Hardswish() support
2020-08-24 21:47:49 -07:00
a8751e50de
model.fuse() fix for export.py ( #827 )
2020-08-24 19:27:54 -07:00
b42e8a531b
optimize imports
2020-08-20 21:17:40 -07:00
0892c44bc4
Fix Logging ( #719 )
...
* Add logging setup
* Fix fusing layers message
* Fix logging does not have end
* Add logging
* Change logging to use logger
* Update yolo.py
I tried this in a cloned branch, and everything seems to work fine
* Update yolo.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-12 14:18:19 -07:00
d5b6416c87
Explicit Imports ( #498 )
...
* expand imports
* optimize
* miss
* fix
2020-08-02 15:47:36 -07:00
9006b85d21
update export.py torchscript file extension
2020-07-10 11:56:01 -07:00
a9918fbf2b
updated coreml conversion to do pixel scaling between 0.0 and 1.0
2020-07-08 09:43:33 +03:00
659ad748c8
update get_voc.sh
2020-07-04 17:13:43 -07:00
7bd7b2c090
update export.py
2020-07-03 20:05:50 -07:00
a9d20eba3e
export.py update
2020-07-03 11:50:59 -07:00
569fbc63c8
export.py update
2020-07-03 11:29:53 -07:00
a62a1c2c67
export.py update
2020-07-01 16:14:49 -07:00
f1d67f4110
update export.py
2020-07-01 15:46:15 -07:00
615d6d0cfa
FROM nvcr.io/nvidia/pytorch:20.06-py3
2020-06-29 14:00:13 -07:00
jylink
Hodovo
Glenn Jocher
Tim Stokman
Jan Hajek
Aditya Lohia
PetrDvoracek
NanoCode012
Jirka Borovec
Laurentiu Diaconu