torchvision에서 제공하는 transform 함수

transforms.Resize

이미지의 사이즈 변환
transforms.Resize((200,200))(im) # 이미지 크기를 200*200으로 변환

 

transforms.RandomCrop

이미지를 임의의 위치에서 자름
transforms.RandomCrop((100,100))(im)

 

transforms.RandomRotation

이미지를 임의의 각도만큼 회전시킴
transforms.RandomRotation(30)(im) # 30도 회전시킴

 

transforms.CenterCrop

중앙에서 주어진 이미지를 자릅니다.
transforms.CenterCrop(size) # size=(h,w), int로 주어지면 정사각형 크롭이 만들어짐

 

transforms.RandomVerticalFlip

이미지를 상하로 뒤집어줌
transforms.RandomVerticalFlip(p) #p=뒤집힐 확률, 기본값=0.5

 

transforms.ToTensor

PIL 이미지 또는 ndarray를 텐서로 변환하고 그에 따라 값을 조정
transforms.ToTensor()(im)

 

transforms.Compose

여러 transforms들을 하나로 묶어서 처리
transforms.Compose([transforms.Resize((224,224)),                    
                                   transforms.RandomVerticalFlip(0.5),                    
                                   transforms.CenterCrop(150)])(im)

 

transforms.Normalize

평균 및 표준편차로 텐서 이미지를 정규화
transforms.Normalize(mean,std)

 

등등...

<다양한 transform 제공하는 라이브러리>

https://github.com/albumentations-team/albumentations

GitHub - albumentations-team/albumentations: Fast image augmentation library and an easy-to-use wrapper around other libraries.