Hur roterar man en bild efter specifik vinkel i PyTorch?

Hur Roterar Man En Bild Efter Specifik Vinkel I Pytorch



PyTorch tillhandahåller modulen 'torchvision.transforms' som har många viktiga transformationer för att utföra olika uppgifter. Denna modul har också ' RandomRotation() ” metod som tillämpar en slumpmässig rotation på en bild inom ett specificerat intervall av vinklar i grader. Denna transform/metod returnerar en ny roterad bild.

Den här artikeln kommer att illustrera metoden för att rotera en viss bild med en specifik vinkel i PyTorch.

Hur roterar man en önskad bild efter specifik vinkel i PyTorch?

För att rotera en önskad bild med en specifik vinkel i PyTorch, kolla in stegen nedan:







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 rotera den med en specifik vinkel:



Steg 2: Importera obligatoriskt bibliotek

Efter det, importera de nödvändiga biblioteken. Till exempel har vi importerat följande bibliotek:

importera ficklampa
importera torchvision.transforms som T
från PIL import bild

Här:

  • ' importera ficklampa ” importerar PyTorch-biblioteket.
  • ' import torchvision.transforms som T ” 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' används för att öppna och spara olika bildfilformat:

Steg 3: Läs ingångsbilden

Läs sedan ingångsbilden från datorn. Här läser vi 'my_img.jpg' och förvara den i 'input_img' variabel:

input_img = Bild. öppen ( 'my_img.jpg' )

Steg 4: Definiera en transformation

Definiera sedan en transformation för att rotera inmatningsbilden. Användare måste ange önskat intervall på (min, max) grader. Här har vi specificerat (30, 60) grader och den nya bilden kommer att roteras med valfri slumpmässig vinkel vald från detta intervall:

omvandla = T. RandomRotation ( grader = ( 30 , 60 ) )

Steg 5: Använd transformationen på inmatningsbilden

Rotera nu ingångsbilden med angivna vinklar med den ovan definierade transformationen:

rotated_img = omvandla ( input_img )

Steg 6: Visa roterad bild

Se slutligen den roterade bilden genom att visa den:

rotated_img

Ovanstående utdata visar att ingångsbilden framgångsrikt har roterats med slumpmässiga vinklar inom det specificerade området 30 och 60 grader.

På samma sätt kan användare också ange andra grader för att rotera bilden. Här kommer vi att specificera följande intervall för att se skillnaden:

omvandla = T. RandomRotation ( grader = ( 90 , 120 ) )

Detta kommer att generera en ny bild som kommer att roteras med valfri slumpmässig vinkel vald från det angivna området 90 och 120 grader:

Jämförelse

Jämförelsen mellan originalbilden och roterade bilder i olika vinklar 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 rotera en bild med olika vinklar i PyTorch.

Slutsats

För att rotera den önskade bilden med en specifik vinkel i PyTorch, ladda först upp den önskade bilden till Google Colab. Importera sedan de nödvändiga biblioteken och läs inmatningsbilden. Därefter definierar du transformationen med hjälp av ' RandomRotation() ”-metoden och tillämpa den på inmatningsbilden. Visa slutligen den nya roterade bilden. Den här artikeln har illustrerat metoden för att rotera den önskade bilden med en specifik vinkel i PyTorch.