¿Cómo recortar una imagen en el centro en PyTorch?

En este artículo, discutiremos cómo recortar una imagen en el centro en PyTorch . 

Método CenterCrop()

Podemos recortar una imagen en PyTorch usando el método CenterCrop() . Este método acepta imágenes como PIL Image, Tensor Image y un lote de imágenes Tensor. La imagen del tensor es un tensor PyTorch con forma [C, H, W] , donde C representa una cantidad de canales y H, W representa la altura y el ancho respectivamente. 

Sintaxis:  torchvision.transforms.CenterCrop(tamaño)

Parámetros:

• tamaño: tamaño de recorte deseado de la imagen.

Devolver: este método devuelve la imagen recortada del tamaño de entrada dado.

Imagen utilizada para la demostración :

 

Ejemplo 1:

En este ejemplo, estamos transformando la imagen en el centro. En esto, obtendremos una imagen cuadrada como salida.

# import required libraries
import torch
import torchvision.transforms as transforms
from PIL import Image
  
# Read image
image = Image.open('a.jpg')
  
# create an transform for crop the image
transform = transforms.CenterCrop(200)
  
# use above created transform to crop 
# the image
image_crop = transform(image)
  
# display result
image_crop.show()

Producción:

 

Ejemplo 2:

En este ejemplo, estamos transformando la imagen con una altura de 180 y un ancho de 300.

# import required libraries
import torch
import torchvision.transforms as transforms
from PIL import Image
  
# Read image
image = Image.open('a.jpg')
  
# define an transform, height=180 width=300
transform = transforms.CenterCrop((180, 300))
  
# use above created transform to crop 
# the image
image_crop = transform(image)
  
# display result
image_crop.show()

Producción: