PyTorch és un marc d'aprenentatge profund conegut que ofereix diverses eines i mòduls per treballar amb nombroses imatges. També proporciona un mòdul 'torchvision.transforms' que té un conjunt de classes i funcions per realitzar diverses transformacions a les imatges, com ara retallar, canviar la mida, girar, girar, escalar i moltes més. Els usuaris poden utilitzar el ' Canvia la mida () ” i especifiqueu la mida, és a dir, l'alçada i l'amplada per canviar la mida de la imatge desitjada. Aquest mètode retorna una nova imatge redimensionada de la mida especificada.
Aquest bloc mostrarà el mètode per canviar la mida d'una imatge a una mida específica a PyTorch.
Com canviar la mida d'una imatge a una mida específica a PyTorch?
Per canviar les dimensions d'una imatge i canviar-la a una mida específica a PyTorch, seguiu els passos que s'indiquen a continuació:
- Penja una imatge a Google Colab
- Importa les biblioteques necessàries
- Llegir la imatge d'entrada
- Calcula i imprimeix la mida de la imatge d'entrada
- Crea una transformació
- Aplica la transformació a la imatge especificada
- Mostra la imatge redimensionada i la seva mida
Pas 1: carregueu una imatge a Google Colab
Primer, obriu Google Colab i feu clic a les icones destacades a continuació. A continuació, trieu la imatge específica de l'ordinador i pengeu-la:
Posteriorment, la imatge es penjarà a Google Colab:
Aquí, hem penjat la següent imatge i la canviarem a una mida específica:
Pas 2: importa la biblioteca necessària
A continuació, importeu les biblioteques necessàries. Per exemple, hem importat les biblioteques següents:
torxa d'importacióimportar torchvision.transforms com Transformar
de la imatge d'importació PIL
importar matplotlib.pyplot com plt
Aquí:
- “ torxa d'importació ” importa la biblioteca PyTorch.
- “ importar torchvision.transforms com a transformacions 'importa el mòdul de transformacions de torchvision que s'utilitza per preprocessar les dades d'imatge abans d'alimentar-les a una xarxa neuronal.
- “ de la imatge d'importació PIL ” obre i desa diferents formats de fitxer d'imatge.
- “ importar matplotlib.pyplot com a plt ” importa el mòdul “pyplot” que s'utilitza per crear visualitzacions i gràfics:
Pas 3: llegiu la imatge d'entrada
Després d'això, llegiu la imatge d'entrada des de l'ordinador. Aquí estem llegint el ' galaxy_img.jpg ” i emmagatzemar-lo al “ entrada_img ” variable:
input_img = Imatge.obert ( 'galaxy_img.jpg' )
Pas 4: Calculeu i imprimiu la mida de la imatge d'entrada
A continuació, calculeu la mida de la imatge d'entrada, és a dir, l'amplada i l'alçada, i imprimiu-la:
mida = input_img.sizeimprimir ( 'Mida de la imatge original (d'entrada):' , mida )
Es pot observar que l'amplada de la imatge d'entrada és '384' i l'alçada de la imatge és '576':
Pas 5: creeu una transformació
A continuació, definiu una transformació per canviar la mida de la imatge d'entrada. Els usuaris han d'especificar la nova mida, és a dir, l'alçada i l'amplada de la nova imatge. Aquí, hem especificat l'alçada '200' i l'amplada '400':
transformar = Transformar.Canviar la mida ( mida = ( 200 , 400 ) )
Pas 6: apliqueu la transformació a la imatge d'entrada
Ara, apliqueu la transformació anterior a la imatge d'entrada especificada per canviar-ne la dimensió:
new_img = transformar ( entrada_img )
Pas 7: Mostra la imatge redimensionada i la seva mida
Finalment, visualitzeu la imatge redimensionada mostrant-la i comproveu-ne la mida:
imprimir ( 'Nova mida de la imatge després de canviar la mida:' , new_img.size )plt.imshow ( nou_img )
plt.show ( )
La sortida següent indica que la imatge d'entrada s'ha redimensionat correctament. Ara, la seva amplada és '400' i la seva alçada és '200':
De la mateixa manera, els usuaris poden especificar qualsevol altra mida per canviar la mida de la imatge. Aquí, canviarem la mida de la mateixa imatge d'entrada amb una altra mida, és a dir, l'alçada '250' i l'amplada '150':
transformar = Transformar.Canviar la mida ( mida = ( 250 , 150 ) )
Això canviarà la mida de la imatge a noves dimensions:
Comparació
La comparació entre la imatge original i les imatges redimensionades amb diferents dimensions es pot veure a continuació:
Nota : Podeu accedir al nostre quadern Google Colab aquí enllaç .
Hem explicat de manera eficient el mètode per canviar la mida d'una imatge a la mida desitjada a PyTorch.
Conclusió
Per canviar la mida de la imatge desitjada a una mida específica a PyTorch, primer, carregueu la imatge desitjada a Google Colab. A continuació, importeu les biblioteques necessàries i llegiu la imatge d'entrada. A continuació, calculeu i imprimiu la mida de la imatge d'entrada. Després d'això, utilitzeu ' Canvia la mida () ” per definir i aplicar la transformació a la imatge d'entrada desitjada. Finalment, mostra la nova imatge redimensionada i la seva mida. Aquest bloc ha demostrat el mètode per canviar la mida de la imatge desitjada a una mida específica a PyTorch.