Come ruotare un'immagine di un angolo specifico in PyTorch?

Come Ruotare Un Immagine Di Un Angolo Specifico In Pytorch



PyTorch fornisce il modulo 'torchvision.transforms' che dispone di molte trasformazioni essenziali per eseguire varie attività. Questo modulo ha anche il ' Rotazione casuale() ' Metodo che applica una rotazione casuale a un'immagine entro un intervallo di angoli specificato in gradi. Questo metodo/trasformazione restituisce una nuova immagine ruotata.

Questo articolo illustrerà il metodo per ruotare una particolare immagine di un angolo specifico in PyTorch.

Come ruotare un'immagine desiderata di un angolo specifico in PyTorch?

Per ruotare un'immagine desiderata di un angolo specifico in PyTorch, controlla i passaggi elencati di seguito:







Passaggio 1: carica un'immagine su Google Colab

Innanzitutto, apri Google Colab e fai clic sulle icone evidenziate di seguito. Quindi, scegli l'immagine specifica dal computer e caricala:





Successivamente l'immagine verrà caricata su Google Colab:





Qui, abbiamo caricato la seguente immagine e la ruoteremo di un angolo specifico:



Passaggio 2: importa la libreria richiesta

Successivamente, importa le librerie necessarie. Ad esempio, abbiamo importato le seguenti librerie:

torcia di importazione
importare torchvision.transforms COME T
dall'immagine di importazione PIL

Qui:

  • torcia di importazione ' importa la libreria PyTorch.
  • importare torchvision.transforms come T ' importa il modulo di trasformazione da torchvision che viene utilizzato per preelaborare i dati dell'immagine prima di inserirli in una rete neurale.
  • “dall'immagine di importazione PIL” viene utilizzato per aprire e salvare diversi formati di file immagine:

Passaggio 3: leggere l'immagine di input

Quindi, leggere l'immagine in ingresso dal computer. Qui stiamo leggendo il “mia_img.jpg” e memorizzandolo nel “input_img” variabile:

input_img = Immagine. aprire ( 'mia_img.jpg' )

Passaggio 4: definire una trasformazione

Successivamente, definisci una trasformazione per ruotare l'immagine di input. Gli utenti devono specificare l'intervallo desiderato di gradi (min, max). Qui abbiamo specificato (30, 60) gradi e la nuova immagine verrà ruotata con qualsiasi angolo casuale selezionato da questo intervallo:

trasformare = T. Rotazione casuale ( gradi = ( 30 , 60 ) )

Passaggio 5: applica la trasformazione sull'immagine di input

Ora, ruota l'immagine in input degli angoli specificati utilizzando la trasformazione definita sopra:

ruotato_img = trasformare ( input_img )

Passaggio 6: Visualizza l'immagine ruotata

Infine, visualizza l'immagine ruotata visualizzandola:

ruotato_img

L'output sopra mostra che l'immagine in input è stata ruotata con successo di angoli casuali nell'intervallo specificato di 30 e 60 gradi.

Allo stesso modo, gli utenti possono anche specificare altri intervalli di gradi per ruotare l'immagine. Qui specificheremo il seguente intervallo per vedere la differenza:

trasformare = T. Rotazione casuale ( gradi = ( 90 , 120 ) )

Ciò genererà una nuova immagine che verrà ruotata con qualsiasi angolo casuale selezionato dall'intervallo specificato di 90 e 120 gradi:

Confronto

Di seguito è possibile vedere il confronto tra l'immagine originale e le immagini ruotate da diverse angolazioni:

Nota : puoi accedere al nostro Google Colab Notebook da qui collegamento .

Abbiamo spiegato in modo efficiente il metodo per ruotare un'immagine di diversi angoli in PyTorch.

Conclusione

Per ruotare l'immagine desiderata di un angolo specifico in PyTorch, caricare innanzitutto l'immagine desiderata su Google Colab. Quindi, importa le librerie necessarie e leggi l'immagine di input. Successivamente, definisci la trasformazione utilizzando il comando ' Rotazione casuale() ' e applicarlo all'immagine di input. Infine, visualizza la nuova immagine ruotata. Questo articolo ha illustrato il metodo per ruotare l'immagine desiderata di un angolo specifico in PyTorch.