인프런 커뮤니티 질문&답변
작성한 질문수
custom데이터로 학습한 yolo 전이학습
작성
·
960
0
안녕하세요!
이번에 yolov7-tiny로 머리만 학습하여 사람들의 머리를 감지하는 모델을 만들었는데요,
학습 시에 프리트레인된 yolov7-tiny.pt를 받아서 학습을 했어요! 근데 이게 전이학습으로 이해를 하고 있습니다.
그런데 아무래도 tiny모델이라 성능이 좋지않아 학습된 모델을 이용하여 다시 전이학습을 해보고싶습니다.
그렇다면 이 상황은 전이학습을 두번하는것이라고 이해하고있는데 그래도 되나요?
처음 학습시킨 vscrowd데이터셋은 이미지가 4~5만 장 사이 되고 새로 학습 시키는 데이터셋은 3천장 미만입니다.
답변 1
0
안녕하세요, 인프런 AI 인턴이에요.
전이학습은 미리 학습된 모델을 가져와서 새로운 데이터셋을 학습시키는 것이기 때문에, 여러번 사용 가능합니다. 따라서, 당신이 이야기한 바와 같이, 성능을 더 개선하기 위해 기존에 학습시킨 모델을 다시 전이학습 하는 것은 가능합니다.
다만, 새로운 데이터셋이 매우 적은 경우 오버피팅(overfitting)이 발생할 가능성이 크기 때문에 이에 대한 조치가 필요합니다. 예를 들어, 데이터셋을 더 확보하거나, augmentation 기법을 사용하여 이미지를 다양하게 변형시키는 등의 방식으로 오버피팅을 방지할 수 있습니다.
감사합니다.