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

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

fish2631님의 프로필 이미지
fish2631

작성한 질문수

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

yolo 전이학습 관련

작성

·

164

0

안녕하세요 선생님. 현재 custom데이터를 학습하여 yolo모델을 잘 사용하고 있습니다.

현재 만든 custom모델에 추가로 다양한 학습데이터를 학습시키려고 하는 방법에 고민이 있습니다.

1. 처음부터 다시 제가 가진 모든 데이터를 학습시키는 방법

  -> 예를 들면 custom학습하기 전 500개의 데이터와 추가의 데이터 600개를 합쳐 1100개를 custom학습 시키는 방법

2. 만들어진 h5파일을 가지고 새로운 데이터만 전이학습 시키는 방법

  -> 예를 들면 500개를 전이학습 시켜놓은 모델(h5)에 다시 600개를 추가로 전이학습 시키는 방법

cf) 500개짜리와 600개짜리는 동일한 래이블 입니다.

1.과 2.번의 방법중 어느 방법이 더 괜찮을지 조언을 듣고싶습니다.

답변 1

1

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

안녕하십니까,

일단 기존 모델의 학습 데이터가 그렇게 많아 보이지 않아서 1번 방법을 더 추천 드립니다.

2번 방법은 기존 학습 데이터가 확연히 많다면 사용하면 좋을 것 같습니다.

감사합니다.

fish2631님의 프로필 이미지
fish2631

작성한 질문수

질문하기