yolov5/models
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>
2021-02-21 21:50:44 -08:00
..
hub Update plot_study() (#2112) 2021-02-01 13:51:11 -08:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py Update inference default to multi_label=False (#2252) 2021-02-19 15:39:09 -08:00
experimental.py GhostConv update (#2082) 2021-01-29 11:25:01 -08:00
export.py Dynamic ONNX engine generation (#2208) 2021-02-21 21:50:44 -08:00
yolo.py Update yolo.py channel array (#2223) 2021-02-15 21:21:53 -08:00
yolov5l.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5m.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5s.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5x.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00