YOLOv5 PyTorch模型转TFLite格式

1 环境准备

确保自己采用YOLOv5 v6.0以及以上版本进行训练，因为v6.0及以上版本才有export.py，模型转换脚本。后面会补充v5.0以及以下版本训练的模型转换方法。

GitHub - ultralytics/yolov5 at v6.0

2 使用export脚本

2.1 确认模型方法

 python .\models\tf.py --weights best.pt --imgsz 640

2.2 export脚本使用

核心代码如下，采用是tf.lite.TFLiteConverter加载pt模型文件，构造转换器。

def export_tflite(keras_model, im, file, int8, data, ncalib, prefix=colorstr('TensorFlow Lite:')):
    # YOLOv5 TensorFlow Lite export
    try:
        import tensorflow as