Toybrick

1.3.2 rknn upsmaple issue

gssg123

注册会员

积分
111
发表于 2020-7-8 15:39:39    查看: 5894|回复: 2 | [复制链接]    打印 | 显示全部楼层
按照你们demo中upsample 的格式改造了代码,代码如下,模型能够正常住转换成onnx模型,但是从onnx 转rknn 遇到not match upsample 的问题,我的环境是pytorch 1.2.0 onnx 1.4.1 还是报错, 你们rknn 支持 IR_version 4 吗?
  1. def forward(self, x, skip, concat=True):
  2.         if x.size()[1] != 320:
  3.             # out = F.interpolate(
  4.             #         x,
  5.             #         size=(skip.size(2), skip.size(3)),
  6.             #         mode="bilinear",
  7.             #         align_corners=False)
  8.             upsample = nn.Upsample(size=(skip.size(2), skip.size(3)), mode='nearest')
  9.             out = upsample(x)
  10.         else:
  11.             out = self.deconv(x)
  12.         # self.deconv.weight.data(3, 3, 4)
  13.         # out = self.deconv(x)
  14.         print('after upsample', out.size(),skip.size(2), skip.size(3))
  15.         if concat:                           
  16.           out = torch.cat([out, skip], 1)
  17.           print('upsample concat', out.size(), skip.size(2), skip.size(3))
  18.         return out
复制代码



回复

使用道具 举报

Zen

中级会员

积分
201
发表于 2020-7-8 16:59:12 | 显示全部楼层
你试试直接pytorch导出pt模型,然后转rknn?
upsample用nearest我记得是没啥问题的
回复

使用道具 举报

gssg123

注册会员

积分
111
 楼主| 发表于 2020-7-9 10:28:51 | 显示全部楼层
Zen 发表于 2020-7-8 16:59
你试试直接pytorch导出pt模型,然后转rknn?
upsample用nearest我记得是没啥问题的 ...

pt模型直接转 rknn 模型吗?
回复

使用道具 举报

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

本版积分规则

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


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