2017-04-13 3 views
0

나는 train.prototxt을 파이썬 코드로 생성하고 loss 레이어를 제거하여 deploy.prototxt을 자동으로 생성하려고합니다. 그러나 다음과 같은 정수로 레이어를 제거하는 방법 만 알고 있습니다.파이썬을 사용하여 카페에서 .prototxt의 이름으로 레이어를 제거하는 방법

net_param = deploy_net.to_proto() 
del net_param.layer[0] 

이름으로 레이어를 제거 할 가능성이 있습니까? Python API 문서는 어디에 있습니까? 나는 그것을 정말로 발견 할 수 없다. C++ 코드를보고 파이썬 코드로 변환하려고합니까?

내가 가진 그물을 초기화하는하고

편집 할 수 있습니다.

net = caffe.NetSpec() 

답변

2

net.layer_dict은 모든 레이어 사전입니다. 그래서 당신이 할 수있는 삭제 : 당신은 파이썬 API에 대한 자세한 내용은 pycaffe.py로 볼 수

del net.layer_dict['layer_name'];

.

+0

내가이 오류 받고 있어요 : __getattr__ 반환 self.tops에'델 net.layer_dict [ '손실'] 파일 "/../net_spec.py", 라인 (176) [이름] KeyError를 'layer_dict' '대답에 빠지셨습니까? – thigi

+0

prototxt를로드했다고 생각했습니다. 커스텀 사용의 경우'del net.tops [ 'loss'];'. – lnman

+0

감사합니다! @ 인 만 – thigi

관련 문제