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

허다혜님의 프로필 이미지

작성한 질문수

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

Oxford Pet 데이터 세트 Training 및 Object Detection 실습 04(TF Record 생성로직 Customization 및 재 Training)

pretrained model 재학습 관련 질문

해결된 질문

23.03.13 15:31 작성

·

369

·

수정됨

0

안녕하세요.

tensorflow object detection 공부 중 몇 가지 궁금증이 생겨 질문드립니다.

 

  1. pretrained model을 이용해 "사과"라는 객체를 탐지하는 A모델을 만들었을 경우, 이 A 모델에 추가적으로 "바나나"라는 객체를 학습 시켜, 최종적으로 "사과","바나나"를 탐지하는 A모델을 만들 수 있는지 궁금합니다.

  2. 만약, 1번이 가능하다면,

    1. 기존 A모델의 labelmap에 id를 추가하여 "바나나" 입력

    2. 기존 A모델의 config 파일에 num class를 2로 변경, "바나나" 학습 데이터(tfrecord) 위치로 변경

      하면 되는 걸까요?

  3. 만약, 1번이 가능하지 않다면,

    1. "사과" tfrecord와 "바나나"tfrecord를 합쳐서 새로운 tfrecord를 만들고

    2. pretrained model를 불러와 새로운 B 모델을 만들어야 할까요?

 

답변 1

1

권 철민님의 프로필 이미지
권 철민
지식공유자

2023. 03. 13. 18:52

안녕하십니까,

안타깝게도(?) 1번 방식은 동작하지 않습니다.

사과와 바나나 두개로 학습 데이터를 만들고, 기존의 pretrained 모델을 재 학습 시켜야 합니다.

감사합니다.