Toybrick

pytorch的上采样不支持转rknn模型?

wjzhoumm

注册会员

积分
84
发表于 2021-4-17 10:04:17    查看: 455|回复: 1 | [复制链接]    打印 | 显示全部楼层
  1. nn.Upsample(size=(x1.size()[2]*2, x1.size()[3]*2), mode='bilinear', align_corners=True)
复制代码
或者
  1. F.interpolate(x1, size=(x1.size()[2]*2, x1.size()[3]*2), mode='bilinear', align_corners=True)
复制代码

使用以上两种上采样的pytorch模型总是无法转换成rknn模型,在转换模型的过程中会报如下错误:
  1. --> Config model
  2. done
  3. --> Loading model
  4. ./unet.pt ********************
  5. E Catch exception when loading pytorch model: ./unet.pt!
  6. E Traceback (most recent call last):
  7. E   File "rknn\api\rknn_base.py", line 339, in rknn.api.rknn_base.RKNNBase.load_pytorch
  8. E   File "rknn\base\RKNNlib\RK_nn.py", line 146, in rknn.base.RKNNlib.RK_nn.RKnn.load_pytorch
  9. E   File "rknn\base\RKNNlib\app\importer\import_pytorch.py", line 128, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run
  10. E   File "rknn\base\RKNNlib\converter\convert_pytorch_new.py", line 2255, in rknn.base.RKNNlib.converter.convert_pytorch_new.convert_pytorch.load
  11. E   File "rknn\base\RKNNlib\converter\convert_pytorch_new.py", line 2370, in rknn.base.RKNNlib.converter.convert_pytorch_new.convert_pytorch.parse_nets
  12. E   File "rknn\base\RKNNlib\converter\convert_pytorch_new.py", line 2059, in rknn.base.RKNNlib.converter.convert_pytorch_new.PyTorchOpConverter.convert_operators
  13. E   File "rknn\base\RKNNlib\converter\convert_pytorch_new.py", line 1439, in rknn.base.RKNNlib.converter.convert_pytorch_new.PyTorchOpConverter.size
  14. E   File "rknn\base\RKNNlib\core\shape.py", line 42, in rknn.base.RKNNlib.core.shape.Shape.reset_shape_format
  15. E   File "rknn\base\RKNNlib\core\shape.py", line 86, in rknn.base.RKNNlib.core.shape.Shape.format
  16. E TypeError: '>=' not supported between instances of 'Tensor' and 'int'
  17. Load Pytorch model failed!
复制代码
请问大佬们这个错误该如何解决,或者这个上采样的函数有可替换的吗?
回复

使用道具 举报

brunolin

中级会员

积分
441
发表于 2021-4-29 10:26:39 | 显示全部楼层
試試先轉成ONNX,再轉RKNN
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表