Toybrick

yolov5-6.0加分割头推理报错

allenyang1027

新手上路

积分
13
楼主
发表于 2023-9-3 19:43:42    查看: 2229|回复: 1 | [复制链接]    打印 | 只看该作者
本帖最后由 allenyang1027 于 2023-9-3 20:00 编辑

我用的模型是yolov5-6.0加分割头模型。
共四个输出头,三个检测头,一个分割头。

* 用rknn-toolkit2-1.5.0export里yolo的test.py可以正常转换也可以正常推理。
* 在用C++部署到rknn时,检测头可以正常输出,但是分割头输出层的上一个3*3卷积层总是报错。报错如下:
# ============================================================================================================================= #
E RKNN: [19:42:29.039] failed to submit!, op id: 140, op name: Conv:/model.33/Conv, flags: 0x1, task start: 365, task number: 213, run task counter: 197, int status: 0
rknn_run fail! ret=-1
tensor name: onnx::Sigmoid_430
tensor name: onnx::Sigmoid_633
tensor name: onnx::Sigmoid_835
tensor name: 1065

# ============================================================================================================================= #
麻烦帮忙看一下!!!谢谢!!!

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

allenyang1027

新手上路

积分
13
沙发
 楼主| 发表于 2023-9-3 19:44:49 | 只看该作者
C++部署在RK3588
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表