파이토치로 이미지 패딩하기 torchvision padding

파이토치로 이미지 패딩하기 torchvision padding¶

• 패딩을 위한 이미지를 PIL 패키지를 활용하여 불러옵니다

In [1]:

from PIL import Image
# 이미지 불러오기
img = Image.open("test2.png").convert('RGB')
img

Out[1]:

• 위, 아래 부분만 패딩을 추가합니다.

In [2]:

from torchvision.transforms.functional import pad

# pad(패팅을 넣을 이미지, (왼쪽 패딩양, 위쪽 패딩양, 오른쪽패딩양, 아래쪽패딩양), fill=패딩색상)
pad(img, (0, 30, 0, 30), fill=135)

Out[2]:

• 위, 아래, 왼쪽, 오른쪽 모두 패딩을 추가합니다

In [3]:

pad(img, (30, 30, 30, 30), fill=135)

Out[3]:

• 딥러닝을 이용하여 이미지를 학습시킬 때 전처리 방식으로 패딩을 줍니다.