IT_World

[PYTHON] 파이썬 이미지 Crop - 2- 본문

Artificial intelligence, AI/TensorFlow

[PYTHON] 파이썬 이미지 Crop - 2-

engine 2021. 4. 6. 18:03

랜덤으로 이미지 crop 하기

 

하나의 이미지만 랜덤으로 잘라내 보기를 해봤다.

 

아래 crop 같은 경우 크롭 위치를 지정해주지만,  RandomHorizontalFlip 랜덤으로 크롭 해준다.

from PIL import Image
from torchvision import transforms
from torchvision.utils import save_image

a= 1
path = f"/home/Documents/0.jpg"
#path 이름 중 반복적으로 특정 명을 넣어줘야할 경우 f를 앞에 작성하고 path를 적으면 
#a = '변경명'으로 다중 path에서 특정 name을 변경할 수 있다.

save_path = f"/home/{a}/"

img = Image.open(path)
transcrop = img(transforms.RandomHorizontalFlip(p=0.5))
# pillow_resize = transforms.Compose([transforms.Resize(size)])

save_image(tensor_interpolate,'/home/09.jpg')
Comments