¿Cómo rellenar una imagen en todos los lados en PyTorch?

En este artículo, discutiremos cómo rellenar una imagen en todos los lados en PyTorch . 

método transforms.pad()

Los rellenos se utilizan para crear algo de espacio alrededor de la imagen, dentro de cualquier borde definido. Podemos configurar diferentes rellenos para lados individuales como (arriba, derecha, abajo, izquierda). El método transforms.Pad() se utiliza para rellenar una imagen. Este método acepta imágenes como PIL Image y Tensor Image. La imagen del tensor es un tensor PyTorch con forma [C, H, W], donde C es el número de canales y H, W es la altura y el ancho respectivamente. La siguiente sintaxis se utiliza para rellenar una imagen.

Sintaxis: transforms.Pad(N)

Parámetro:

• N: Relleno en cada borde

Retorno: este método devuelve una imagen con relleno.

Requisito del paquete

pip install torchvision
pip install Pillow

Imagen utilizada para la demostración:

 

Ejemplo 1:

En este ejemplo, veremos cómo rellenar N a la izquierda/derecha y M al relleno superior/inferior para todos los lados.

# import required libraries
import torch
import torchvision.transforms as transforms
from PIL import Image
 
# Read the image from your computer
img = Image.open('geekslogo.png')
 
# get width and height of image
w, h = img.size
 
# pad 100 to left/right and 50 to top/bottom
transform = transforms.Pad((100, 50))
 
# add padding to image
img = transform(img)
 
# resize the image to original dimension
img = img.resize((w, h))
 
# display output
img.show()

Producción:

 

Ejemplo 2:

En este ejemplo, veremos cómo agregar un espacio de relleno único en todos los lados.

# import required libraries
import torch
import torchvision.transforms as transforms
from PIL import Image
 
# Read the image from your computer
img = Image.open('geekslogo.png')
 
# get width and height of image
w, h = img.size
 
# pad 10 to left, 20 to top, 30 to right,
# 50 bottom
transform = transforms.Pad((10, 20, 50, 50))
 
# add padding to image
img = transform(img)
 
# resize the image to original dimension
img = img.resize((w, h))
 
# display output
img.show()

Producción: