인공지능/DeepLearning
2021. 1. 11.
[Computer Vision] 확장된 Image를 이용한 훈련 모델 사용
%matplotlib inline from d2l import torch as d2l import torch import torchvision from torch import nn 사용될 모듈 항목들이다. 학습에 사용될 모델들은 torchvision에서 제공하는 image 데이터셋인 CIFAR-10을 사용한다. all_images = torchvision.datasets.CIFAR10(train=True, root="../data", download=True) d2l.show_images([all_images[i][0] for i in range(32)], 4, 8, scale=0.8); train : train 데이터셋을 받아올지 test용 데이터셋을 받아올지 선택 root : 데이터셋을 받아올 경로를..