caffe-Upsample编译错误

[复制链接] · 发表于 2020-10-22 11:27:29

在本人模型里面存在着upsample操作，只要运行到Upsapmle操作时就会出现IndexError：list index out of range错误。
本人的API 是：1.3.2 版本

只看该作者 · 发表于 2020-10-23 14:57:46

已解决，自问自答，方便一波其他有同样错误的人（如有错误，请告知）。
前提：API:1.4.0 DRV：1.4.0
caffe转rknn文件，其中存在upsample操作。
目前，rknn貌似不支持upsample操作，不过，其有一个备选的是Resize层，将upsample的参数修改为Resize参数格式即可，具体如下：
layer {
  name: "upsample1"
  type: "Resize"
  bottom: "relu_blob41"
  top: "upsample_blob1"
  resize_param {
height: 16
width: 16
interp_mode: 1
  }
}

只看该作者 · 发表于 2020-10-27 17:12:24

优秀