인프런 커뮤니티 질문&답변
작성한 질문수
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번 방법은 기존 학습 데이터가 확연히 많다면 사용하면 좋을 것 같습니다.
감사합니다.