더북(TheBook)

torchvision.transfrom은 이미지 데이터를 변환하여 모델(네트워크)의 입력으로 사용할 수 있게 변환해 줍니다. 이때 사용되는 파라미터는 다음과 같습니다.

Resize: 이미지의 크기를 조정. 즉, 256×256 크기로 이미지 데이터를 조정합니다.

RandomResizedCrop: 이미지를 랜덤한 크기 및 비율로 자릅니다.

ResizeRandomResizedCrop 모두 이미지를 자르는 데 사용하지만 그 용도는 다릅니다. Resize가 합성곱층을 통과하기 위해 이미지 크기를 조정하는 전처리 과정이라면, RandomResizedCrop은 데이터 확장 용도로 사용됩니다. RandomResizedCrop은 이미지를 랜덤한 비율로 자른 후 데이터 크기를 조정합니다.

RandomHorizontalFlip: 이미지를 랜덤하게 수평으로 뒤집습니다.

ToTensor: 이미지 데이터를 텐서로 변환합니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.