search

rockchip-linux / rknn-toolkit

BSD 3-Clause "New" or "Revised" License
771 stars 175 forks source link

跪求！只有输出前一层的resize报错 Load模型时Try match layer: Resize_/Resize_1:out0 failed! #420

Open xianglongJ opened 7 months ago

xianglongJ commented 7 months ago

使用nn.UpsamplingNearest2d和 F.interpolate（model=nearest）都会在output前的resize层报错，其余的resize使用前面两个OP不会报错。 在ret = rknn.load_onnx(model=ONNX_MODEL,inputs=['images'],outputs=['output'])会报这个错误 直接ret = rknn.load_onnx(model=ONNX_MODEL)不会报错 1701865400594

xianglongJ commented 7 months ago

rknn-tool 1.7.5版本 目标平台rv1126

eRaul commented 6 months ago

我这里测了下，torch模型可以正常转rknn。请提供下复现问题的模型。 torch模型： torch_resize_end 转出来的rknn模型： rknn_resize_end 如果是onnx框架的，试下在导出的时候把opset_version设到12或以下，我这里测opset version 12也可以正常转换。