yolov5/models
Jiacong Fang 3beb871ba4
Multiple TF export improvements (#4824)
* Add fused conv support

* Set all saved_model values to non trainable

* Fix TFLite fp16 model export

* Fix int8 TFLite conversion
2021-09-16 15:27:22 +02:00
..
hub Fix: add P2 layer 21 to yolov5-p2.yaml `Detect()` inputs (#4608) 2021-08-30 15:46:33 +02:00
__init__.py
common.py Allow `multi_label` option for NMS with PyTorch Hub (#4728) 2021-09-09 15:32:04 +02:00
experimental.py Add TensorFlow and TFLite export (#1127) 2021-08-17 13:18:16 +02:00
tf.py Multiple TF export improvements (#4824) 2021-09-16 15:27:22 +02:00
yolo.py Refactor `forward()` method profiling (#4816) 2021-09-16 09:47:34 +02:00
yolov5l.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
yolov5m.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
yolov5s.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00
yolov5x.yaml Standardize headers and docstrings (#4417) 2021-08-14 21:17:51 +02:00