yolov8系列模型转换步骤
1. pt->onnx
- 在官方给出的airockchip/ultralytics_yolov8下,配置cfg/default.yaml中task,mode,model路径:
task: segment # (str) YOLO task, i.e. detect, segment, classify, pose
mode: export # (str) YOLO mode, i.e. train, val, predict, export, track, benchmark
model: xx/yolov8n-seg.pt(自己训练的模型路径) # (str, optional) path to model file, i.e. yolov8n.pt, yolov8n.yaml
...
format: rknn
- run engine/exporter.py转换为官方算子适应的onnx
PyTorch: starting from '' with input shape (16, 3