¿Cómo rotar una imagen en un ángulo específico en PyTorch?

Como Rotar Una Imagen En Un Angulo Especifico En Pytorch



PyTorch proporciona el módulo 'torchvision.transforms' que tiene muchas transformaciones esenciales para realizar diversas tareas. Este módulo también tiene el “ Rotación aleatoria() 'Método que aplica una rotación aleatoria a una imagen dentro de un rango específico de ángulos en grados. Esta transformación/método devuelve una nueva imagen rotada.

Este artículo ilustrará el método para rotar una imagen particular en un ángulo específico en PyTorch.

¿Cómo rotar una imagen deseada en un ángulo específico en PyTorch?

Para rotar una imagen deseada en un ángulo específico en PyTorch, consulte los pasos que se enumeran a continuación:







Paso 1: sube una imagen a Google Colab

Primero, abra Google Colab y haga clic en los íconos resaltados a continuación. Luego, elige la imagen específica de la computadora y cárgala:





Posteriormente, la imagen se subirá a Google Colab:





Aquí, hemos subido la siguiente imagen y la rotaremos en un ángulo específico:



Paso 2: Importar la biblioteca requerida

Después de eso, importe las bibliotecas necesarias. Por ejemplo, hemos importado las siguientes bibliotecas:

importar antorcha
importar torchvision.transforms como t
desde la imagen de importación PIL

Aquí:

  • importar antorcha ”importa la biblioteca PyTorch.
  • importar torchvision.transforms como T ”importa el módulo de transformaciones de torchvision que se utiliza para preprocesar datos de imágenes antes de introducirlos en una red neuronal.
  • “De la imagen de importación PIL” se utiliza para abrir y guardar diferentes formatos de archivos de imagen:

Paso 3: leer la imagen de entrada

Luego, lea la imagen de entrada desde la computadora. Aquí estamos leyendo el “mi_img.jpg” y almacenarlo en el “entrada_img” variable:

entrada_img = Imagen. abierto ( 'mi_img.jpg' )

Paso 4: definir una transformación

A continuación, defina una transformación para rotar la imagen de entrada. Los usuarios deben especificar el rango deseado de grados (mínimo, máximo). Aquí, hemos especificado (30, 60) grados y la nueva imagen se rotará con cualquier ángulo aleatorio seleccionado de este rango:

transformar = T. Rotación aleatoria ( grados = ( 30 , 60 ) )

Paso 5: aplicar la transformación en la imagen de entrada

Ahora, gire la imagen de entrada en ángulos específicos usando la transformación definida anteriormente:

rotado_img = transformar ( entrada_img )

Paso 6: Mostrar imagen girada

Finalmente, vea la imagen rotada mostrándola:

rotado_img

El resultado anterior muestra que la imagen de entrada se ha girado correctamente en ángulos aleatorios en el rango especificado de 30 y 60 grados.

De manera similar, los usuarios también pueden especificar otros rangos de grados para rotar la imagen. Aquí, especificaremos el siguiente rango para ver la diferencia:

transformar = T. Rotación aleatoria ( grados = ( 90 , 120 ) )

Esto generará una nueva imagen que se rotará con cualquier ángulo aleatorio seleccionado del rango especificado de 90 y 120 grados:

Comparación

La comparación entre la imagen original y las imágenes rotadas en diferentes ángulos se puede ver a continuación:

Nota : Puede acceder a nuestro Google Colab Notebook en este enlace .

Hemos explicado de manera eficiente el método para rotar una imagen en diferentes ángulos en PyTorch.

Conclusión

Para rotar la imagen deseada en un ángulo específico en PyTorch, primero, cargue la imagen deseada en Google Colab. Luego, importe las bibliotecas necesarias y lea la imagen de entrada. Después de eso, defina la transformación usando el botón ' Rotación aleatoria() 'Método y aplicarlo a la imagen de entrada. Finalmente, muestre la nueva imagen rotada. Este artículo ha ilustrado el método para rotar la imagen deseada en un ángulo específico en PyTorch.