Hoe roteer je een afbeelding volgens een specifieke hoek in PyTorch?

Hoe Roteer Je Een Afbeelding Volgens Een Specifieke Hoek In Pytorch



PyTorch biedt de module “torchvision.transforms” die veel essentiële transformaties bevat om verschillende taken uit te voeren. Deze module heeft ook de “ Willekeurige rotatie() '-methode die een willekeurige rotatie toepast op een afbeelding binnen een opgegeven bereik van hoeken in graden. Deze transformatie/methode retourneert een nieuwe geroteerde afbeelding.

Dit artikel illustreert de methode om een ​​bepaalde afbeelding met een specifieke hoek in PyTorch te roteren.

Hoe roteer ik een gewenste afbeelding volgens een specifieke hoek in PyTorch?

Om een ​​gewenste afbeelding met een specifieke hoek in PyTorch te roteren, bekijk je de onderstaande stappen:







Stap 1: Upload een afbeelding naar Google Colab

Open eerst Google Colab en klik op de hieronder gemarkeerde pictogrammen. Kies vervolgens de specifieke afbeelding op de computer en upload deze:





Vervolgens wordt de afbeelding geüpload naar Google Colab:





Hier hebben we de volgende afbeelding geüpload en we zullen deze in een specifieke hoek draaien:



Stap 2: Importeer de vereiste bibliotheek

Importeer daarna de benodigde bibliotheken. We hebben bijvoorbeeld de volgende bibliotheken geïmporteerd:

fakkel importeren
importeer torchvision.transforms als T
van PIL-importafbeelding

Hier:

  • fakkel importeren ” importeert de PyTorch-bibliotheek.
  • importeer torchvision.transforms als T ” importeert de transformatiemodule van torchvision die wordt gebruikt om beeldgegevens voor te verwerken voordat deze in een neuraal netwerk worden ingevoerd.
  • “uit iPIL-importafbeelding” wordt gebruikt voor het openen en opslaan van verschillende afbeeldingsbestandsformaten:

Stap 3: Lees de invoerafbeelding

Lees vervolgens het invoerbeeld van de computer. Hier lezen we de “mijn_img.jpg” en het opslaan in de “invoer_img” variabele:

invoer_img = Afbeelding. open ( 'mijn_img.jpg' )

Stap 4: Definieer een transformatie

Definieer vervolgens een transformatie om de invoerafbeelding te roteren. Gebruikers moeten het gewenste bereik van (min, max) graden opgeven. Hier hebben we (30, 60) graden gespecificeerd en de nieuwe afbeelding wordt geroteerd met elke willekeurige hoek die uit dit bereik is geselecteerd:

transformeren = T. Willekeurige rotatie ( graden = ( 30 , 60 ) )

Stap 5: Pas de transformatie toe op de invoerafbeelding

Roteer nu de invoerafbeelding met gespecificeerde hoeken met behulp van de hierboven gedefinieerde transformatie:

geroteerd_img = transformeren ( invoer_img )

Stap 6: Geef het geroteerde beeld weer

Bekijk ten slotte de geroteerde afbeelding door deze weer te geven:

geroteerd_img

Uit de bovenstaande uitvoer blijkt dat het invoerbeeld met succes is geroteerd met willekeurige hoeken binnen het opgegeven bereik van 30 en 60 graden.

Op dezelfde manier kunnen gebruikers ook andere gradenbereiken opgeven om de afbeelding te roteren. Hier zullen we het volgende bereik specificeren om het verschil te zien:

transformeren = T. Willekeurige rotatie ( graden = ( 90 , 120 ) )

Hierdoor wordt een nieuwe afbeelding gegenereerd die wordt geroteerd met elke willekeurige hoek geselecteerd uit het opgegeven bereik van 90 en 120 graden:

Vergelijking

De vergelijking tussen de originele afbeelding en geroteerde afbeeldingen vanuit verschillende hoeken kunt u hieronder zien:

Opmerking : Hier heeft u toegang tot ons Google Colab Notebook koppeling .

We hebben op efficiënte wijze de methode uitgelegd om een ​​afbeelding vanuit verschillende hoeken te roteren in PyTorch.

Conclusie

Om de gewenste afbeelding met een specifieke hoek in PyTorch te roteren, uploadt u eerst de gewenste afbeelding naar Google Colab. Importeer vervolgens de benodigde bibliotheken en lees de invoerafbeelding. Definieer daarna de transformatie met behulp van de “ Willekeurige rotatie() ”-methode en pas deze toe op de invoerafbeelding. Geef ten slotte de nieuwe geroteerde afbeelding weer. Dit artikel heeft de methode geïllustreerd om de gewenste afbeelding met een specifieke hoek in PyTorch te roteren.