작성
·
197
0
안녕하세요!
Segmentation -> matterport에서 load_weights 함수를 작동시킬때, by_name=True라는 변수를 넣어주지 않을때 에러가 발생합니다.
혹시 by_name 변수가 하는 역할을 알려주실수 있을까요?
감사합니다!
답변 1
0
안녕하십니까,
load_weights()는 pretrained 모델의 weight를 미리 mask rcnn 모델로 로딩합니다. by_name=True로 설정하면 전체 모델중에 특정 이름을 가진 layer들만 weight를 로딩합니다. 여기에 exclude=["mrcnn_class_logits", "mrcnn_bbox_fc","mrcnn_bbox", "mrcnn_mask"] 하면 exclude layer 만 제외한 다른 layer들의 weight를 로딩합니다.
by_name=False이면, 특정 이름의 layer가 아닌 모델의 tensorflow graph에 기재된 순서대로 layer의 weight를 로딩합니다. 그런데 이때 exclude할 layer들이 설정되어 있으면 오류를 발생 시킬 수 있는 것 같습니다.
감사합니다.