인프런 커뮤니티 질문&답변
작성한 질문수
size mismatch 현상
작성
·
432
0
안녕하세요 교수님. 양질의 강의를 제공해주셔서 감사합니다.
다름이 아니라 VOC Dataset은 CLASS가 20개이고 COCO Dataset은 80개의 클래스를 가지고 있는 것으로 알고 있습니다.
config_file과 checkpoint_file 또한 coco로 맞추어져 있는데 train_detector를 실행하면
위의 사진과 같이 size mismatch라는 출력이 나오게 됩니다.
우려와는 다르게(?) 결과는 잘 나오는데 혹시나 문제가 되지 않을까 하여 질문 드립니다.
감사합니다.
답변 1
1
권 철민
지식공유자
안녕하십니까,
80개의 class로 된 coco 데이터 세트로 pretrained된 checkpoint 된 파일에서 RPN의 Weight가 customized된 학습 데이터로 학습하려는 RPN의 Weight와 size가 맞지 않아서 나오는 warning message 같습니다.
음,,, 성능에는 큰 영향이 없어보입니다. 저도 해보면 해당 warning이 나옵니다만 성능에는 문제는 없어 보입니다.
감사합니다.