Come ritagliare qualsiasi immagine al centro in PyTorch?

Come Ritagliare Qualsiasi Immagine Al Centro In Pytorch



PyTorch offre un modulo 'torchvision.transforms' che ha una serie di funzioni e classi. Esegue varie trasformazioni sulle immagini, come ritaglio, ridimensionamento, rotazione, capovolgimento, ridimensionamento e molto altro. Offre il “ CenterCrop(dimensione) ' metodo per ritagliare qualsiasi immagine specifica al suo centro. Gli utenti devono specificare la dimensione di ritaglio desiderata, ovvero altezza e larghezza dell'immagine. Questo metodo produce una nuova immagine ritagliata di una dimensione specificata.

Questo articolo dimostrerà il metodo per ritagliare qualsiasi immagine al centro in PyTorch.

Come ritagliare qualsiasi immagine al centro in PyTorch?

Per ritagliare qualsiasi immagine al centro in PyTorch, controlla i passaggi forniti:







  • Carica un'immagine su Google Colab
  • Importa le librerie necessarie
  • Leggi l'immagine in ingresso
  • Definire una trasformazione per ritagliare l'immagine di input al suo centro
  • Applica la trasformazione sull'immagine
  • Visualizza l'immagine ritagliata

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:





Fatto ciò, l'immagine verrà caricata su Google Colab:





Qui, abbiamo caricato la seguente immagine e la ritaglieremo al centro:



Passaggio 2: importa la libreria necessaria
Successivamente, importa le librerie richieste. Ad esempio, abbiamo importato le seguenti librerie:

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

Qui:

  • torcia di importazione ' importa la libreria PyTorch.
  • importare torchvision.transforms come trasformazioni ' 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
Successivamente, leggi l'immagine di input dal computer. Qui stiamo leggendo il “paesaggio_img.jpg” e memorizzandolo nel “input_img” variabile:

input_img = Immagine.open ( 'paesaggio_img.jpg' )

Passaggio 4: definire una trasformazione
Quindi, definisci una trasformazione per ritagliare l'immagine di input sopra al suo centro. Gli utenti possono fornire un valore per il ritaglio quadrato o due valori per il ritaglio rettangolare. Qui abbiamo specificato un valore, ad esempio '230' per il ritaglio quadrato:

trasformazione = trasforma.CenterCrop ( ( 230 ) )

Passaggio 5: applica la trasformazione sull'immagine
Ora applica la trasformazione sopra sull'immagine di input desiderata per ritagliarla al centro:

cropped_img = trasforma ( input_img )

Passaggio 6: visualizza l'immagine ritagliata
Infine, visualizza l'immagine ritagliata visualizzandola:

ritagliato_img

L'output sopra mostra che l'immagine in input è stata ritagliata con successo al centro in una forma quadrata.

Allo stesso modo, gli utenti possono anche specificare due valori, ovvero altezza e larghezza nel file “CropCentro()” metodo per ritagliare l'immagine al centro in una forma rettangolare. Qui forniremo la seguente altezza e larghezza:

trasformazione = trasforma.CenterCrop ( ( 150 , 280 ) )

Questo ha ritagliato l'immagine in una forma rettangolare:

Confronto
Il confronto tra l'immagine originale e le immagini ritagliate può essere visto di seguito:

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

Abbiamo spiegato in modo efficiente il metodo per ritagliare un'immagine al centro in PyTorch.

Conclusione

Per ritagliare un'immagine specifica al centro in PyTorch, carica innanzitutto l'immagine desiderata su Google Colab. Quindi, importa le librerie necessarie e leggi l'immagine di input. Successivamente, utilizzare il “CropCentro()” metodo per ritagliare l'immagine di input desiderata al centro. Infine, visualizza l'immagine ritagliata visualizzandola. Questo articolo dimostra il metodo per ritagliare qualsiasi immagine al centro in PyTorch.