Toybrick

标题: yolov5-6.0加分割头推理报错 [打印本页]

作者: allenyang1027 时间: 2023-9-3 19:43
标题: yolov5-6.0加分割头推理报错
本帖最后由 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

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

作者: allenyang1027 时间: 2023-9-3 19:44
C++部署在RK3588

欢迎光临 Toybrick (https://t.rock-chips.com/)