미리 계산 된 caffe 모델의 하위 집합을 저장할 수있는 방법은 무엇입니까?

Question

저는 3 개의 레이어가있는 사전 계산 된 caffe 모델 (파이썬으로) 작업 중입니다. 이 카페 모델을 분해하고이 모델의 첫 번째 레이어와 동일한 새 모델을 만들고 싶습니다. 예를 들면 : -> conv1_1 -> conv1_2 -> conv2_1 -> conv2_2 -> conv3_1 -> conv3_2

새로운 CAFFE 모델 데이터 -

원래 CAFFE 모델 데이터> conv1_1 -> conv1_2

수 아무도 도와주지?

Answer 1

파이썬은 .caffemodel 파일 안의 데이터를 공개합니다. 배열로 액세스 할 수 있습니다. 예 :

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[...] 

이 데이터를 새 파일로 복사하여 .caffemodel 파일로 저장할 수 있습니다. this 및 this을 살펴보십시오.