怎么读取py-faster-rcnn训练出来的caffemodel中的权重值

普通的网络可以通过Python接口加载solver文件和caffemodel,然后将caffemodel中的值保存为文本文件,请问py-faster-rcnn训练好的模型怎么读取其权重,按照常规方法好像不行
1.jpg
已邀请:

PangDeGui - 人脸、主动学习、职场新人

赞同来自:

提示是缺rpn的那个层啊,是不是文件丢了。
 
你可以试试

import caffe.proto.caffe_pb2 as caffe_pb2 
m = caffe_pb2.NetParameter() 
with open('./Pnet.caffemodel', 'rb') as f: 
     m.ParseFromString(f.read()) 
 
m对象里面就是整个训练好的模型,包括网络结构和参数
 

要回复问题请先登录注册