yolov5-6.0加分割头推理报错

[复制链接] · 发表于 2023-9-3 19:43:42

本帖最后由 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

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

只看该作者 · 发表于 2023-9-3 19:44:49

C++部署在RK3588

yolov5-6.0加分割头推理报错

本帖子中包含更多资源