Toybrick

Caffe转rknn失败

xiaqing10

高级会员

积分
572
发表于 2019-5-14 10:39:59    查看: 6437|回复: 2 | [复制链接]    打印 | 显示全部楼层
--> config model
done
--> Loading model
E Not supported caffenet model version(v0 layer or v1 layer).                    Please use the newest model version.
E Catch exception when loading caffe model: ./gender.prototxt!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 253, in rknn.api.rknn_base.RKNNBase.load_caffe
T   File "rknn/base/RKNNlib/converter/caffeloader.py", line 984, in rknn.base.RKNNlib.converter.caffeloader.CaffeLoader.load_blobs
T   File "rknn/base/RKNNlib/converter/caffeloader.py", line 888, in rknn.base.RKNNlib.converter.caffeloader.CaffeLoader.parse_blobs
T   File "rknn/base/RKNNlib/RKNNlog.py", line 105, in rknn.base.RKNNlib.RKNNlog.RKNNLog.e
T ValueError: Not supported caffenet model version(v0 layer or v1 layer).                    Please use the newest model version.
Load model failed!

提示的是caffe版本问题,请问这个需要更新新的caffe训练后才能转吗?


回复

使用道具 举报

jefferyzhang

版主

积分
12844
发表于 2019-5-17 18:09:52 | 显示全部楼层
是的,caffe版本太旧了,有一个命令可以升级caffe模型版本的,具体还要查下
回复

使用道具 举报

求大佬指教

中级会员

积分
241
发表于 2019-9-24 14:41:17 | 显示全部楼层
prototxt版本旧 ,可以手动修改代码格式解决
  1. name: "name"       #根据自己代码进行修改
  2. input: "data"
  3. input_dim: 1
  4. input_dim: 3
  5. input_dim: 128
  6. input_dim: 128
复制代码
修改为
  1. layer {
  2.   name: "name"
  3.   type: "Input"
  4.   top: "data"
  5.   input_param {
  6.     shape {
  7.       dim: 1
  8.       dim: 3
  9.       dim: 128
  10.       dim: 128
  11.     }
  12.   }
  13. }
复制代码

回复

使用道具 举报

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

本版积分规则

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


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