将YOLOv5转换为TFLite后检测出现类别名称错误问题求助

问题原因 #

导出TFLite模型时，默认不会自动关联自定义数据集的类别配置文件，运行detect.py时会自动使用YOLOv5自带的COCO类别列表，导致检测出的类别名称和自定义训练的不匹配。

解决步骤 #

• 先确认你有自定义数据集的配置文件（比如custom_data.yaml），文件里必须包含names字段，清晰列出所有训练时用的类别名称。
• 方法一：运行检测命令时显式指定配置文件

  !python yolov5/detect.py --weights best-fp16.tflite --source images.jpg --data custom_data.yaml
  

• 方法二：导出TFLite模型时就绑定配置文件，让模型自带类别信息

  !python yolov5/export.py --weights best.pt --include tflite --data custom_data.yaml
  

• 验证：重新运行检测命令，查看输出的类别名称是否和自定义类别一致。

问题截图 #

内容的提问来源于stack exchange，提问作者Mouad Baaziz