Computer Vision/Image Classification

[ResNet50] 전이 학습 기법을 이용한 CNN 이미지 분류 모델 생성

저번에 공부했던 ResNet 이론을 실습으로 옮겨 PyTorch 환경에서 ResNet50 사전학습 모델을 불러와 STL10 데이터셋을 전이학습시키고, 이미지 분류를 할 수 있는 모델을 생성해보자. 데이터셋 설명 및 준비에 앞서, 데이터 증강에 대해 간단히 알아보자. 데이터 증강(Data Augmentation)이란, 갖고 있는 데이터셋을 여러가지 방법으로 증강시켜 실질적인 학습 데이터셋의 규모를 키우는 방법이다. 데이터가 다양하게 많아진다는것은 학습 성능을 향상시키고 과적합을 줄일 수 있다는 것을 의미하기 때문에 데이터셋 규모가 작을때 데이터 증강은 어쩌면 필수라고 할 수 있다. 조금 더 쉽게 설명하면 위 사진에서, 원본 데이터를 d라고 하고 증강된 데이터를 Aug_d 라고 하자. 빨간색 x 표시가 Pr..