Toybrick

caffe-Upsample编译错误

18022443868

中级会员

积分
228
发表于 2020-10-22 11:27:29    查看: 383|回复: 2 | [复制链接]    打印 | 显示全部楼层
在本人模型里面存在着upsample操作,只要运行到Upsapmle操作时就会出现IndexError:list index out of range错误。
本人的API 是:1.3.2 版本
回复

使用道具 举报

18022443868

中级会员

积分
228
 楼主| 发表于 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
  }
}
回复

使用道具 举报

jefferyzhang

超级版主

积分
7364
发表于 2020-10-27 17:12:24 | 显示全部楼层
优秀
回复

使用道具 举报

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

本版积分规则

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


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