[혼만딥] 5주차 - 전이 학습으로 이미지 분류하기

dev·2025년 8월 10일
0

혼만딥

목록 보기
5/7
post-thumbnail

📚 혼공학습단

📖 혼자 만들면서 공부하는 딥러닝

전이 학습

  • 특정한 문제를 해결하기 위해 훈련된 모델의 지식을 다른 문제에 적용하는 머신러닝 기법
    • 이미 대규모 데이터로 학습된 모델의 가중치를 가져와 새로운 문제에 맞게 모델의 일부를 수정하거나 재학습
    • 처음부터 모델을 학습시키는 것보다 적은 데이터와 계산량으로 높은 성능을 얻을 수 있음

사전 훈련된 모델로 피스타치오 분류

모델 ImageNet으로 사전 훈련된 EfficientNet
데이터셋 Kaggle에서 제공하는 피스타치오 이미지 데이터셋

  • 결과

20 에포크 학습 후, 훈련/검증 데이터 모두에서 높은 정확도를 보임 (안정적으로 수렴)

0개의 댓글