"Meet unsupported split" on RK3588 NPU

固件类型:官方提供的固件
固件文件名称:ITX-3588J_Ubuntu20.04-Gnome-r21164_v1.0.0c_220622
固件下载地址:BaiduPan
Log日志:empty.txt

在rk3588上运行yolov6https://github.com/meituan/YOLOv6/releases/download/0.1.0/yolov6s.onnx
将其转为rknn文件之后,在simulator上能正常跑完。但是在rk3588 npu上运行会有如下错误

E RKNN: [02:25:03.732] Meet unsupported split
Aborted (core dumped)

出错的算子

ID OpType DataType Target InputShape OutputShape
57 Split INT8 CPU (1,1,80,80,85) (1,1,80,80,2),(1,1,80,80,2),(1,1,80,80,81)

请问目前是否是不支持改运算?有计划何时会支持么?
tchip_askquestions
empty.txt (68 Bytes)

请问您解决了么

这因为你输出时,没有设置输出节点

没解决呢。
这个split属于中间层,还没到输出层。为什么和没设置输出节点有关系呢。
看转换的log,好像这几层都是用CPU操作的,是不是因为3588的NPU上没有实现CPU的算子
yolov6s.zip (8.18 KB)

sjwlaoda 发表于 2022-7-13 15:20
没解决呢。
这个split属于中间层,还没到输出层。为什么和没设置输出节点有关系呢。
看转换的log,好像这 …

你输出时,要设置输出节点,输出3个头