이번 예제에서 사용할 fashion_mnist 데이터셋은 토치비전으로 내려받을 수 있습니다.

코드 5-3 fashion_mnist 데이터셋 내려받기

train_dataset = torchvision.datasets.FashionMNIST("../chap05/data", download=True,
                       transform=transforms.Compose([transforms.ToTensor()])) ------ ①
test_dataset = torchvision.datasets.FashionMNIST("../chap05/data", download=True,
                train=False, transform=transforms.Compose([transforms.ToTensor()])) ------ 앞에서 훈련 데이터셋을 내려받았다면 여기에서는 테스트 데이터셋을 내려받습니다.

① torchvision.datasets는 torch.utils.data.Dataset의 하위 클래스로 다양한 데이터셋(CIFAR, COCO, MNIST, ImageNet 등)을 포함합니다. torchvision.datasets에서 사용하는 주요한 파라미터는 다음과 같습니다.

ⓐ 첫 번째 파라미터: FashionMNIST를 내려받을 위치를 지정합니다.

ⓑ download: download를 True로 변경해 주면 첫 번째 파라미터의 위치에 해당 데이터셋이 있는지 확인한 후 내려받습니다.

ⓒ transform: 이미지를 텐서(0~1)로 변경합니다.