Transforms.Normalize()의 역할
파이썬에서 tensor로 이미지를 처리할 때 주로 transform.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))로 처리를 한다고 한다. 이는 이미지의 경우 픽셀 값 하나 하나가 0 ~ 255 값을 갖는데, tensor로 타입 변환 시 0 ~ 1 사이의 값으로 바뀌게 된다. 이를 -1 ~ 1 사이의 값으로 normalized 시켜서 이미지를 동일한 환경으로 어느정도 맞춰주어 학습을 진행하기 위해 사용하는 것이라고 한다! transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) 이렇게 한 번 Normalize 시켜주었던 이미지를 다시 복구시켜 확인하고 싶으면 아래처럼 값을 조정해주어야 한다. plt.imshow(transforms.T..
[PyTorch] DataLoader의 역할 및 사용법
DataLoader는 데이터를 미니 배치 단위로 나누어서 제공해주는 역할을 합니다. 학습을 하기 위해서 데이터를 읽어올 때 사용하게 됩니다. dataset 인자에는 pytorch Dataset 객체를 넣어주면 됩니다. DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, num_workers=0, collate_fn=None, pin_memory=False, drop_last=False, timeout=0, worker_init_fn=None, *, prefetch_factor=2, persistent_workers=False) 인자를 하나씩 살펴보겠습니다. - batch_size : 말 그대로 batch_si..
