Table of Contents

분류기

분류기(Classifier) 학습하기

데이터는 어떻게 하나요?

일반적으로 이미지나 텍스트,오디오나 비디오 데이터를 다룰 때는 표준 Python 패키지를 이용하여 NumPy 배열로 불러오면 됩니다. 그 후 그 배열을 torch.*Tensor로 변환합니다.

이미지 분류기 학습하기

다음과 같은 단계로 진행해보겠습니다.

  1. torchvision을 사용하여 CIFAR10의 학습용/시험용 데이터넷을 불러오고, 정규화 (nomalizing)합니다.
  2. 합성곱 신경망 (Convolution Neural Network)을 정의합니다.
  3. 손실 함수를 정의합니다.
  4. 학습용 데이터를 사용하여 신경망을 학습합니다.
  5. 시험용 데이터를 사용하여 신경망을 검사합니다.