새 배열을 .caffemodel 파일로 덤프하십시오.

Question

커널 크기, 패드 정보 등과 같은 특정 매개 변수가있는 사용자 정의 된 레이어가 있습니다.이 중 bool/int 유형의 배열 매개 변수가 있습니다. 처리 후 .caffemodel 파일에이 배열 매개 변수를 추가해야합니다. 배열을 카페인 모델 파일에 링크/덤프하는 방법?

아래와 같이 .proto 파일에 배열 매개 변수를 포함 시켰습니다.

{ 
... //other parameters 
repeated bool/int <varibale_name> [packed = true] 
} 

bool 형식의 blob을 만들 수 있습니까?

Answer 1

당신은 PY-CAFFE net surgery을 통해 그것을 할 수 있습니다 -

net.params[learnable_layer_name].add_blob() 
new_blob_index = len(net.params[learnable_layer_name]) - 1 
net.params[learnable_layer_name][new_blob_index].reshape(desired_shape) 
net.params[learnable_layer_name][new_blob_index].data[:] = new_data_to_insert 

HTH.