PyTorch är ett välkänt ramverk för djupinlärning som erbjuder olika verktyg och moduler för att arbeta med många bilder. Den tillhandahåller också en 'torchvision.transforms'-modul som har en uppsättning klasser och funktioner för att utföra olika transformationer på bilder, såsom beskärning, storleksändring, rotering, vändning, skalning och många fler. Användare kan använda ' Ändra storlek() ”-metoden och ange storlek, dvs höjd och bredd för att ändra storlek på önskad bild. Denna metod returnerar en ny storleksändrad bild av den angivna storleken.
Den här bloggen kommer att demonstrera metoden för att ändra storlek på en bild till en specifik storlek i PyTorch.
Hur ändrar man storlek på en bild till en specifik storlek i PyTorch?
För att ändra måtten på en bild och ändra storlek på den till en specifik storlek i PyTorch, följ stegen nedan:
- Ladda upp en bild till Google Colab
- Importera nödvändiga bibliotek
- Läs ingångsbild
- Beräkna och skriv ut storleken på ingångsbilden
- Skapa en transformation
- Använd transformering på den angivna bilden
- Visa den ändrade storleken på bilden och dess storlek
Steg 1: Ladda upp en bild till Google Colab
Öppna först Google Colab och klicka på de markerade ikonerna nedan. Välj sedan den specifika bilden från datorn och ladda upp den:
Därefter kommer bilden att laddas upp till Google Colab:
Här har vi laddat upp följande bild och vi kommer att ändra storlek på den till en specifik storlek:
Steg 2: Importera obligatoriskt bibliotek
Importera sedan de nödvändiga biblioteken. Till exempel har vi importerat följande bibliotek:
importera ficklampaimportera torchvision.transforms som Omvandla
från PIL import bild
importera matplotlib.pyplot som plt
Här:
- ' importera ficklampa ” importerar PyTorch-biblioteket.
- ' import torchvision.transformers som transformer ” importerar transformationsmodulen från torchvision som används för att förbehandla bilddata innan den matas in i ett neuralt nätverk.
- ' från PIL import bild ” öppnar och sparar olika bildfilformat.
- ' importera matplotlib.pyplot som plt ' importerar modulen 'pyplot' som används för att skapa visualiseringar och plotter:
Steg 3: Läs ingångsbilden
Efter det läser du ingångsbilden från datorn. Här läser vi ' galaxy_img.jpg ' och lagra den i ' input_img ' variabel:
input_img = Bild.öppen ( 'galaxy_img.jpg' )
Steg 4: Beräkna och skriv ut indatabildstorlek
Beräkna sedan storleken på inmatningsbilden, dvs bredd och höjd, och skriv ut den:
storlek = input_img.sizeskriva ut ( 'Storlek på originalbild (indata):' , storlek )
Det kan observeras att bredden på inmatningsbilden är '384' och höjden på bilden är '576':
Steg 5: Skapa en transformation
Definiera sedan en transformation för att ändra storlek på inmatningsbilden. Användare måste ange den nya storleken, dvs höjd och bredd för den nya bilden. Här har vi specificerat höjd '200' och bredd '400':
transform = Transform.Ändra storlek ( storlek = ( 200 , 400 ) )
Steg 6: Tillämpa transformationen på indatabilden
Använd nu transformationen ovan på den angivna inmatningsbilden för att ändra dess dimension:
new_img = transform ( input_img )
Steg 7: Visa ändrad storlek på bild och dess storlek
Se slutligen den ändrade storleksbilden genom att visa den och verifiera dess storlek:
skriva ut ( 'Ny storlek på bilden efter storleksändring:' , new_img.size )plt.imshow ( new_img )
plt.show ( )
Utdata nedan indikerar att storleken på ingångsbilden har ändrats. Nu är dess bredd '400' och dess höjd är '200':
På samma sätt kan användare ange vilken annan storlek som helst för att ändra storlek på bilden. Här kommer vi att ändra storlek på samma ingångsbild med en annan storlek, dvs höjd '250' och bredd '150':
transform = Transform.Ändra storlek ( storlek = ( 250 , 150 ) )
Detta kommer att ändra storlek på bilden till nya dimensioner:
Jämförelse
Jämförelsen mellan originalbilden och de storleksförändrade bilderna med olika dimensioner kan ses nedan:
Notera : Du kan komma åt vår Google Colab Notebook här länk .
Vi har effektivt förklarat metoden för att ändra storlek på en bild till önskad storlek i PyTorch.
Slutsats
För att ändra storlek på önskad bild till en specifik storlek i PyTorch, ladda först upp den önskade bilden till Google Colab. Importera sedan de nödvändiga biblioteken och läs inmatningsbilden. Beräkna och skriv sedan ut storleken på inmatningsbilden. Efter det använder du ' Ändra storlek() ”-metod för att definiera och tillämpa transformationen på den önskade indatabilden. Till sist, visa den nya storleken på bilden och dess storlek. Den här bloggen har demonstrerat metoden för att ändra storlek på önskad bild till en specifik storlek i PyTorch.