밑바닥부터 시작하는 딥러닝 5

178

밑바닥부터 시작하는 딥러닝 5

그림

6-11

학습 후 신경망

그림과 같이

sin

함수의 곡선을 잘 표현하고 있습니다. 이처럼 선형 회귀가 수행한 선형 변환에

활성화 함수를 덧씌우면 비선형 관계도 제대로 학습할 수 있습니다.

6.5

토치비전과 데이터셋

이번 절에서는 파이토치용 이미지 처리 라이브러리인 토치비전

torchvision

에 대해 설명합니다. 토

치비전은 데이터셋 로딩과 이미지 전처리 등의 기능을 제공합니다. 또한

MNIST

와

CIFAR

-

10

같은 유명한 데이터셋도 손쉽게 이용할 수 있게 해줍니다.

6.5.1

토치비전 설치

토치비전은 다음 명령어로 설치할 수 있습니다.

$ pip3 install torchvision

올바르게 설치되면 다음 코드처럼 임포트할 수 있습니다.