Toybrick

标题: 1.3.2 rknn upsmaple issue [打印本页]

---

作者: gssg123    时间: 2020-7-8 15:39
标题: 1.3.2 rknn upsmaple issue
按照你们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    时间: 2020-7-8 16:59
你试试直接pytorch导出pt模型，然后转rknn？
upsample用nearest我记得是没啥问题的

---

作者: gssg123    时间: 2020-7-9 10:28

Zen 发表于 2020-7-8 16:59
你试试直接pytorch导出pt模型，然后转rknn？
upsample用nearest我记得是没啥问题的 ...

pt模型直接转 rknn 模型吗？

---

欢迎光临 Toybrick (https://t.rock-chips.com/)

Powered by Discuz! X3.3