Toybrick

标题: Caffe转rknn失败 [打印本页]

---

作者: xiaqing10    时间: 2019-5-14 10:39
标题: Caffe转rknn失败
--> 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    时间: 2019-5-17 18:09
是的，caffe版本太旧了，有一个命令可以升级caffe模型版本的，具体还要查下

---

作者: 求大佬指教    时间: 2019-9-24 14:41
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. }

复制代码

---

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

Powered by Discuz! X3.3