카페에서 네트워크의 가중치를 추출하십시오.

Question

python으로 최적화 된 네트워크의 가중치를 추출하고 싶습니다. .caffemodel 파일이 있고 전체 네트워크의 매개 변수를 제공하는 net.params을 얻었습니다. 문제는 내가 즉 net.params['ip2'] 그것은 나를주는 예제에 대한 첫 번째 레이어를 부르고 때입니다 : 내가 대신 포인터, 가중치의 행렬을 얻을 수있는 방법

<caffe._caffe.BlobVec object at 0x7f1cb03c8fa0>

? 사전에

감사합니다,

아프신 내가 그것을 가지고

Answer 1

얻을 수있다.

net = caffe.Net('path/to/conv.prototxt', 'path/to/conv.caffemodel', caffe.TEST) 
W = net.params['con_1'][0].data[...] 
b = net.params['con_1'][1].data[...] 

이 link에서 모양과 자세한 내용은이 link 되세요.

Answer 2

, 는 당신은 .prototxt 파일과 .caffemodel 파일을 사용하여 네트워크를 읽을 필요 weights2 = net.params['ip2'][0].data