pytorch 的 nn.Upsample 和 nn.MaxPool2d 是否支持转成 rknn?

NPU Team 答复：
请提供下保存下来的pt模型和onnx模型，pytorch版本建议1.2.0，onnx建议1.4.1

updated:
这两个op现在rknn toolkit都是支持的，但是upsample支持的不是很完整，目前只支持resize参数，不支持scale_factor参数，我们会在下一个版本完善对这个OP的支持。
upsample可以参考附件upsample_demo.py的写法来定义网络。maxpool转onnx后也是可以的，写法参考maxpool_demo.py。请确保pytorch版本是1.2.0，onnx版本是1.4.1.
附：以上脚本在RKNN Toolkit 1.3.2上测试。
（250082）



（#250082）

我也遇到这个问题了，pytorch->onnx>rknn。torch.nn.Upsample报错。请问哪个现在哪个版本支持，如果不支持，我们自己改如何改进。

奥古师弟 发表于 2020-6-15 17:45
我也遇到这个问题了，pytorch->onnx>rknn。torch.nn.Upsample报错。请问哪个现在哪个版本支持，如果不支持 ...

之前没看到最新回复。

jefferyzhang 发表于 2020-5-9 08:22
NPU Team 答复：
请提供下保存下来的pt模型和onnx模型，pytorch版本建议1.2.0，onnx建议1.4.1

安装要求修改了unsample，但是还是报了这个错误。  pytorch->onnx->rknn.load_onnx()

E Traceback (most recent call last):
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 1071, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 980, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx._onnx_push_ready_tensor
E TypeError: 'NoneType' object is not iterable
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 513, in rknn.api.rknn_base.RKNNBase.load_onnx
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 1077, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
E   File "rknn/api/rknn_log.py", line 312, in rknn.api.rknn_log.RKNNLog.e
E ValueError: Try match Upsample_270ut0 failed, catch exception!

jefferyzhang 发表于 2020-4-24 21:40
1.3.2正式版在这里：https://github.com/rockchip-linux/rknn-toolkit
其他beta版本有些是根据客户bug临时 ...

您好，我也想问下rknn什么时候支持pytorch的upsample，我这边的pt文件想通过onnx转rknn，最终也是报不支持upsample的错。。。。。

kkkaaa 发表于 2020-4-29 16:45
试了 1.3.2, 好像还是不支持 upsample

请问解决了么？同样问题。。。。