Como girar uma imagem em um ângulo específico no PyTorch?

Como Girar Uma Imagem Em Um Angulo Especifico No Pytorch



PyTorch fornece o módulo “torchvision.transforms” que possui muitas transformações essenciais para realizar diversas tarefas. Este módulo também possui o “ Rotação aleatória() ”Método que aplica uma rotação aleatória a uma imagem dentro de um intervalo especificado de ângulos em graus. Esta transformação/método retorna uma nova imagem girada.

Este artigo ilustrará o método para girar uma imagem específica em um ângulo específico no PyTorch.

Como girar uma imagem desejada em um ângulo específico no PyTorch?

Para girar uma imagem desejada em um ângulo específico no PyTorch, verifique as etapas listadas abaixo:







Etapa 1: faça upload de uma imagem para o Google Colab

Primeiro, abra o Google Colab e clique nos ícones destacados abaixo. Em seguida, escolha a imagem específica do computador e carregue-a:





Posteriormente, a imagem será carregada no Google Colab:





Aqui, carregamos a seguinte imagem e iremos girá-la em um ângulo específico:



Etapa 2: importar a biblioteca necessária

Depois disso, importe as bibliotecas necessárias. Por exemplo, importamos as seguintes bibliotecas:

importar tocha
importar torchvision.transforms como T
da imagem de importação PIL

Aqui:

  • importar tocha ”Importa a biblioteca PyTorch.
  • importar torchvision.transforms como T ”importa o módulo de transformação do torchvision que é usado para pré-processar dados de imagem antes de alimentá-los em uma rede neural.
  • “da imagem de importação PIL” é usado para abrir e salvar diferentes formatos de arquivo de imagem:

Etapa 3: leia a imagem de entrada

Em seguida, leia a imagem de entrada do computador. Aqui, estamos lendo o “meu_img.jpg” e armazená-lo no “entrada_img” variável:

entrada_img = Imagem. abrir ( 'meu_img.jpg' )

Etapa 4: definir uma transformação

A seguir, defina uma transformação para girar a imagem de entrada. Os usuários precisam especificar o intervalo desejado de graus (min, max). Aqui, especificamos (30, 60) graus e a nova imagem será girada com qualquer ângulo aleatório selecionado neste intervalo:

transformar = T. Rotação aleatória ( graus = ( 30 , 60 ) )

Etapa 5: aplicar a transformação na imagem de entrada

Agora, gire a imagem de entrada em ângulos especificados usando a transformação definida acima:

rotacionado_img = transformar ( entrada_img )

Etapa 6: exibir imagem girada

Finalmente, visualize a imagem girada exibindo-a:

rotacionado_img

A saída acima mostra que a imagem de entrada foi girada com sucesso em ângulos aleatórios no intervalo especificado de 30 e 60 graus.

Da mesma forma, os usuários também podem especificar outras faixas de graus para girar a imagem. Aqui, especificaremos o seguinte intervalo para ver a diferença:

transformar = T. Rotação aleatória ( graus = ( 90 , 120 ) )

Isso irá gerar uma nova imagem que será girada com qualquer ângulo aleatório selecionado no intervalo especificado de 90 e 120 graus:

Comparação

A comparação entre a imagem original e as imagens giradas por diferentes ângulos pode ser vista abaixo:

Observação : Você pode acessar nosso Google Colab Notebook neste link .

Explicamos com eficiência o método para girar uma imagem em diferentes ângulos no PyTorch.

Conclusão

Para girar a imagem desejada em um ângulo específico no PyTorch, primeiro carregue a imagem desejada no Google Colab. Em seguida, importe as bibliotecas necessárias e leia a imagem de entrada. Depois disso, defina a transformação usando o “ Rotação aleatória() ”E aplique-o à imagem de entrada. Finalmente, exiba a nova imagem girada. Este artigo ilustrou o método para girar a imagem desejada em um ângulo específico no PyTorch.