반응형
rescale
-
Pytorch 사용자 정의 Transforms.Compose 만들기카테고리 없음 2021. 5. 15. 02:46
- 우리가 원하는 대로 데이터를 핸들링 하려면 Pytorch 에서 제공해주는 Compose 기능을 사용할 수 있다. - 간단히 Cifar 10 의 데이터를 가지고 우리가 원하는 대로 rescale, randomCrop , 그리고 이를 합친 Compose([rescale,randomcrop]) 을 구현해보겠다. - 우선 원하는 모듈을 로딩시켜주고, import torch import torch.nn as nn import numpy as np import torch.optim as optim import torchvision import torchvision.transforms as transforms import matplotlib.pyplot as plt %matplotlib inline - Cuda가..