在使用YOLOV8-seg训练出来的pt模型转为onnx的时候,利用以下仓库地址转。
git clone https://github.com/airockchip/ultralytics_yolov8.git
在修改ultralytics/cfg/default.yaml中的task,mode为model为自己需要的内容后,
执行以下语句
cd ultralytics_yolov8
export PYTHONPATH=./
python ./ultralytics/engine/export.py
出现错误 AttributeError: 'Segment' object has no attribute 'detect'
一开始以为库的环境安装不对,所以不断卸载重装,后面发现是因为这个onnx转rknn模型的仓库使用的是ultralytics8.0.151版本,而训练的时候由于下载的是最新版8.2.52,两个结构可能有不一致的地方,所以我们可以在训练的时候intall官方8.0.151版本的ultralytics去训练,最后成功转如下图所示。