인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

인프런 커뮤니티 질문&답변

HAHA님의 프로필 이미지
HAHA

작성한 질문수

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

안녕하세요 강사님 matterport부분 질문있습니다

작성

·

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들이 설정되어 있으면 오류를 발생 시킬 수 있는 것 같습니다. 

감사합니다.

HAHA님의 프로필 이미지
HAHA

작성한 질문수

질문하기