RKNN3588——YOLOv8的PT模型转RKNN模型

一：PT转ONNX

1. 首先克隆rknn修改后的ultralytics版本项目到本地

https://github.com/airockchip/ultralytics_yolov8

cd ultralytics-main
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install -e .

主要是修改了源码的ultralytics/nn/modules/head.py和ultralytics/engine/exporter.py两个文件。

2. 使用修改后的ultralytics对pt模型进行模型转换，此处的format=rknn代表支持rknn后续的转换，而不是用onnx，一定要注意！！！

yolo export model=/your_path/best.pt format=rknn

二：ONNX转RKNN

1. 在linux x86上配置Toolkit环境，具体步骤如下链接：