best.pt转best.onnx

Yolov8在RK3588上进行自定义目标检测(一)已经进行了配置文件修改。接下来可以直接进行模型的转换。

下面是两种转换方法：

1.命令行

yolo export model=best.pt format=rknn

2.转换脚本 convert_to_onnx.py

from ultralytics import YOLO
 
model = YOLO(r'C:\Users\WYX\Desktop\code\yolov8-main\runs\detect\train\weights\best.pt')
model.export(format="rknn")

直接运行python convert_to_onnx.py。

3.执行结果

将会在yolov8-main\runs\detect\train\weights目录生成onnx模型文件。