如何使用Yolo 5转换模型

Question

Yolo 5是一个对象检测模型，可以导出到几个不同的框架，包括TensorFlow和Core。https://github.com/ultralytics/yolov5

我已经能够训练Yolo 5模型，并将其导出到TensorFlow (TF1图def，或TF2 savemodel)，并尝试了Apple。

我一直未能找到Yolo 5的任何例子，关于如何使用这些模型一旦出口。即如何获取图像文件并获取检测到的对象/标签/坐标

我尝试过类似于TF1对象检测的python代码，但是导出的模型似乎不兼容，https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/tensorflow-1.14/。

TF2，https://www.tensorflow.org/hub/tutorials/object_detection都没有

或TFlite，https://www.tensorflow.org/lite/examples/object_detection/overview

Answer 1

使用YOLOv5在训练后计算的权重。他们通常在这条路上：

yolov5/runs/train/your_yolo_model/weights/best.pt