Hugging Face's logo

Spaces:
xiang-wuu
/
yolov5
Runtime error

App Files Community
yolov5 / models /export.py

Commit History

Merge `develop` branch into `master` (#3518)
f3c3d2c
unverified

glenn-jocher Piotr Skalski SkalskiP Peretz Cohen tudoulei chocosaj BuildTools developer0hye Sam_S Samridha Shrestha edificewang commited on

ONNX export in .train() mode fix (#3362)
ef4d538
unverified

Chaofeng111 commited on

Check CoreML models.train() mode (#3262)
7b36e38
unverified

glenn-jocher commited on

Parameterize ONNX `--opset-version` (#3154)
9ab561d
unverified

Cristi Fati commited on

Add `--include torchscript onnx coreml` argument (#3137)
d9b4e6b
unverified

Cristi Fati glenn-jocher commited on

Add --optimize argument (#3093)
525f4f8
unverified

glenn-jocher commited on

Update export.py with --train mode argument (#3066)
e97d129
unverified

glenn-jocher commited on

Do not optimize CoreML TorchScript model (#3055)
3ef3a95
unverified

glenn-jocher commited on

Fix ONNX export using --grid --simplify --dynamic simultaneously (#2982)
b292837
unverified

whoam glenn-jocher commited on

Add support for FP16 (half) to export.py (#3010)
e2a80c6
unverified

Hodovo glenn-jocher commited on

Add yolov5/ to sys.path() for *.py subdir exec (#2949)
4200674
unverified

glenn-jocher commited on

Add file_size() function (#2911)
1b1ab4c
unverified

glenn-jocher commited on

Update export.py for 2 dry runs (#2910)
646386f
unverified

glenn-jocher commited on

Update export.py (#2909)
a2a514d
unverified

glenn-jocher commited on

Default optimize_for_mobile() on TorchScript models (#2908)
28db237
unverified

glenn-jocher commited on

Fix ONNX dynamic axes export support with onnx simplifier, make onnx simplifier optional (#2856)
1df8c6c
unverified

Tim Stokman Tim glenn-jocher commited on

ONNX Simplifier (#2815)
1f3e482
unverified

glenn-jocher commited on

YOLOv5 v5.0 Release patch 1 (#2764)
0f395b3
unverified

glenn-jocher commited on

GPU export options (#2297)
7a0a81f
unverified

Jan Hajek Jan Hajek glenn-jocher commited on

Dynamic ONNX engine generation (#2208)
95aefea
unverified

Aditya Lohia Shivam Swanrkar glenn-jocher commited on

nn.SiLU() export support (#1713)
6bd5e8b
unverified

glenn-jocher commited on

Rearrange export input after checks (#1118)
d45e349
unverified

glenn-jocher commited on

classifier, export, torch seed updates
883924d

glenn-jocher commited on

Update ci-testing.yml reflecting sys.path additions (#869)
a12698f
unverified

glenn-jocher commited on

export.py folder sys.path addition (#1052)
d55c326
unverified

PetrDvoracek commited on

Update export.py --img-size comment
1e7674d

glenn-jocher commited on

Update export.py with v3.0 Hardswish() support (#831)
4fb8cb3

glenn-jocher commited on

Update export.py with v3.0 Hardswish() support
4d7f222

glenn-jocher commited on

model.fuse() fix for export.py (#827)
a8751e5

glenn-jocher commited on

Explicit Imports (#498)
d5b6416
unverified

Jirka Borovec commited on

update export.py torchscript file extension
9006b85

glenn-jocher commited on

updated coreml conversion to do pixel scaling between 0.0 and 1.0
a9918fb

Laurentiu Diaconu commited on

FROM nvcr.io/nvidia/pytorch:20.06-py3
615d6d0

glenn-jocher commited on