데이터 로드1 [Pytorch] 데이터 불러오기 및 처리 Data Loading and Processing Tutorial Transforms 대부분 뉴럴넷은 정해진 크기의 이미지를 입력으로 받는다. 그래서 preprocessing 코드가 필요하다. Rescale RandomCrop : 임의로 이미지를 자른다. (data augmentation) ToTensor : numpy 배열의 이미지를 torch 텐서로 바꾸어준다.(we need to swap axes!) torch 는 효율적인 연산을 위해서 numpy array를 tensor로 바꾸고 모델에 입력한다. ToTensor가 이를 돕는데, 내부 코드는 아래와 같다. class ToTensor(object): def __call__(self, sample): # axis를 바꾼다. # numpy array: .. 2019. 7. 11. 728x90 이전 1 다음
