yolov5/models
DavidB 596de6d5a0
Default FP16 TensorRT export (#6798)
* Assert engine precision #6777

* Default to FP32 inputs for TensorRT engines

* Default to FP16 TensorRT exports #6777

* Remove wrong line #6777

* Automatically adjust detect.py input precision #6777

* Automatically adjust val.py input precision #6777

* Add missing colon

* Cleanup

* Cleanup

* Remove default trt_fp16_input definition

* Experiment

* Reorder detect.py if statement to after half checks

* Update common.py

* Update export.py

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-03-06 21:21:16 +01:00
..
hub Fix yolov3.yaml remove list (#6655) 2022-02-16 10:13:05 +01:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py Default FP16 TensorRT export (#6798) 2022-03-06 21:21:16 +01:00
experimental.py precommit: isort (#5493) 2021-11-04 17:24:25 +01:00
tf.py Fix 6 Flake8 issues (#6541) 2022-02-05 15:22:59 +01:00
yolo.py Init tensor directly on device (#6068) 2021-12-23 13:49:00 +01:00
yolov5l.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5m.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5n.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5s.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5x.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00