Jan Hajek
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>
3 роки тому
Aditya Lohia
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>
3 роки тому
Glenn Jocher
6bd5e8bca7
nn.SiLU() export support ( #1713 )
3 роки тому
Glenn Jocher
d45e349a1e
Rearrange export input after checks ( #1118 )
img size checks are warnings rather than errors, so current implementation allows improperly formed model inputs.
4 роки тому
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
4 роки тому
Glenn Jocher
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
4 роки тому
PetrDvoracek
d55c326dda
export.py folder sys.path addition ( #1052 )
4 роки тому
Glenn Jocher
d68179953d
Update export.py
4 роки тому
Glenn Jocher
1e7674d4dd
Update export.py --img-size comment
4 роки тому
Glenn Jocher
4fb8cb353f
Update export.py with v3.0 Hardswish() support ( #831 )
4 роки тому
Glenn Jocher
4d7f222f73
Update export.py with v3.0 Hardswish() support
4 роки тому
Glenn Jocher
a8751e50de
model.fuse() fix for export.py ( #827 )
4 роки тому
Glenn Jocher
b42e8a531b
optimize imports
4 роки тому
NanoCode012
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>
4 роки тому
Jirka Borovec
d5b6416c87
Explicit Imports ( #498 )
* expand imports
* optimize
* miss
* fix
4 роки тому
Glenn Jocher
9006b85d21
update export.py torchscript file extension
4 роки тому
Laurentiu Diaconu
a9918fbf2b
updated coreml conversion to do pixel scaling between 0.0 and 1.0
4 роки тому
Glenn Jocher
659ad748c8
update get_voc.sh
4 роки тому
Glenn Jocher
7bd7b2c090
update export.py
4 роки тому
Glenn Jocher
a9d20eba3e
export.py update
4 роки тому
Glenn Jocher
569fbc63c8
export.py update
4 роки тому
Glenn Jocher
a62a1c2c67
export.py update
4 роки тому
Glenn Jocher
f1d67f4110
update export.py
4 роки тому
Glenn Jocher
615d6d0cfa
FROM nvcr.io/nvidia/pytorch:20.06-py3
4 роки тому