RK3588部署YOLOV8-seg的问题

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

相关推荐

  1. rknn部署rk3588

    2024-07-12 10:46:05       20 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-12 10:46:05       66 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-12 10:46:05       70 阅读
  3. 在Django里面运行非项目文件

    2024-07-12 10:46:05       57 阅读
  4. Python语言-面向对象

    2024-07-12 10:46:05       68 阅读

热门阅读

  1. 引入时间概念的分布式系统浅谈

    2024-07-12 10:46:05       22 阅读
  2. Vue3框架搭建4:配置说明-eslint配置

    2024-07-12 10:46:05       21 阅读
  3. linux 查看 io使用率iotop

    2024-07-12 10:46:05       24 阅读
  4. 【番外】Springboot集成推荐配置及十问RocketMQ

    2024-07-12 10:46:05       27 阅读
  5. 软设之职责链模式

    2024-07-12 10:46:05       20 阅读
  6. vue中ref()与reactive(的区别)

    2024-07-12 10:46:05       19 阅读
  7. Flask发布一个及时止损(止盈)服务(二)

    2024-07-12 10:46:05       20 阅读
  8. NoSQL之Redis数据库配置与优化

    2024-07-12 10:46:05       27 阅读