PyTorch er en velkendt deep-learning-ramme, der tilbyder forskellige værktøjer og moduler til at arbejde med adskillige billeder. Det giver også et 'torchvision.transforms'-modul, der har et sæt klasser og funktioner til at udføre forskellige transformationer på billeder, såsom beskæring, ændring af størrelse, rotation, vending, skalering og mange flere. Brugere kan bruge ' Ændre størrelse() ”-metoden og angiv størrelsen, dvs. højde og bredde for at ændre størrelsen på ethvert ønsket billede. Denne metode returnerer et nyt ændret billede af den angivne størrelse.
Denne blog vil demonstrere metoden til at ændre størrelsen på et billede til en bestemt størrelse i PyTorch.
Hvordan ændrer man størrelsen på et billede til en bestemt størrelse i PyTorch?
For at ændre dimensionerne på et billede og ændre størrelsen på det til en bestemt størrelse i PyTorch, følg nedenstående trin:
- Upload et billede til Google Colab
- Importer nødvendige biblioteker
- Læs inputbillede
- Beregn og udskriv inputbilledstørrelse
- Opret en transformation
- Anvend transformation på det angivne billede
- Vis det ændrede størrelse billede og dets størrelse
Trin 1: Upload et billede til Google Colab
Åbn først Google Colab og klik på de nedenfor fremhævede ikoner. Vælg derefter det specifikke billede fra computeren og upload det:
Efterfølgende vil billedet blive uploadet til Google Colab:
Her har vi uploadet følgende billede, og vi vil ændre størrelsen på det til en bestemt størrelse:
Trin 2: Importer påkrævet bibliotek
Importer derefter de nødvendige biblioteker. For eksempel har vi importeret følgende biblioteker:
import lommelygteimport torchvision.transformers som Transform
fra PIL import billede
importer matplotlib.pyplot som plt
Her:
- “ import lommelygte ” importerer PyTorch-biblioteket.
- “ import torchvision.transformers som transformer ” importerer transformationsmodulet fra torchvision, der bruges til at forbehandle billeddata, før det føres ind i et neuralt netværk.
- “ fra PIL import billede ” åbner og gemmer forskellige billedfilformater.
- “ importer matplotlib.pyplot som plt ' importerer 'pyplot'-modulet, der bruges til at skabe visualiseringer og plots:
Trin 3: Læs inputbilledet
Læs derefter inputbilledet fra computeren. Her læser vi ' galaxy_img.jpg ' og gemme den i ' input_img variabel:
input_img = Image.open ( 'galaxy_img.jpg' )
Trin 4: Beregn og udskriv inputbilledstørrelse
Beregn derefter størrelsen på inputbilledet, dvs. bredde og højde, og udskriv det:
størrelse = input_img.sizePrint ( 'Størrelse på originalt (input) billede:' , størrelse )
Det kan observeres, at bredden af inputbilledet er '384' og højden af billedet er '576':
Trin 5: Opret en transformation
Derefter skal du definere en transformation for at ændre størrelsen på inputbilledet. Brugere skal angive den nye størrelse, dvs. højde og bredde for det nye billede. Her har vi specificeret højde '200' og bredde '400':
transform = Transform.Resize ( størrelse = ( 200 , 400 ) )
Trin 6: Anvend transformationen på inputbilledet
Anvend nu ovenstående transformation på det angivne inputbillede for at ændre dets dimension:
new_img = transform ( input_img )
Trin 7: Vis ændret størrelse billede og dets størrelse
Til sidst kan du se det ændrede størrelsesbillede ved at vise det og kontrollere dets størrelse:
Print ( 'Ny størrelse på billede efter ændring af størrelse:' , new_img.size )plt.imshow ( new_img )
plt.vis ( )
Nedenstående output indikerer, at inputbilledet er blevet ændret. Nu er dens bredde '400' og dens højde er '200':
På samme måde kan brugere angive enhver anden størrelse for at ændre størrelsen på billedet. Her vil vi ændre størrelsen på det samme inputbillede med en anden størrelse, dvs. højde '250' og bredde '150':
transform = Transform.Resize ( størrelse = ( 250 , 150 ) )
Dette vil ændre størrelsen på billedet til nye dimensioner:
Sammenligning
Sammenligningen mellem det originale billede og de ændrede billeder med forskellige dimensioner kan ses nedenfor:
Bemærk : Du kan få adgang til vores Google Colab Notebook her link .
Vi har effektivt forklaret metoden til at ændre størrelsen på et billede til en ønsket størrelse i PyTorch.
Konklusion
For at ændre størrelsen på det ønskede billede til en bestemt størrelse i PyTorch skal du først uploade det ønskede billede til Google Colab. Importer derefter de nødvendige biblioteker og læs inputbilledet. Beregn og udskriv derefter størrelsen på inputbilledet. Brug derefter ' Ændre størrelse() ” metode til at definere og anvende transformationen på det ønskede inputbillede. Til sidst skal du vise det nye billedstørrelse og dets størrelse. Denne blog har demonstreret metoden til at ændre størrelsen på det ønskede billede til en bestemt størrelse i PyTorch.