Toybrick

标题: caffe-Upsample编译错误 [打印本页]

作者: 18022443868 时间: 2020-10-22 11:27
标题: caffe-Upsample编译错误
在本人模型里面存在着upsample操作，只要运行到Upsapmle操作时就会出现IndexError：list index out of range错误。
本人的API 是：1.3.2 版本

作者: 18022443868 时间: 2020-10-23 14:57
已解决，自问自答，方便一波其他有同样错误的人（如有错误，请告知）。
前提：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
  }
}

作者: jefferyzhang 时间: 2020-10-27 17:12
优秀

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