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

add "__init__.py" file in <models> folder

2020-06-15 14:37:07 +08:00

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