RK3588部署YOLOV8-seg的问题

Source

         在使用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去训练,最后成功转如下图所示。