Wie drehe ich ein Bild in PyTorch um einen bestimmten Winkel?

Wie Drehe Ich Ein Bild In Pytorch Um Einen Bestimmten Winkel



PyTorch stellt das Modul „torchvision.transforms“ bereit, das viele wichtige Transformationen zur Ausführung verschiedener Aufgaben enthält. Dieses Modul verfügt außerdem über „ RandomRotation() „Methode, die eine zufällige Drehung auf ein Bild innerhalb eines angegebenen Winkelbereichs in Grad anwendet. Diese Transformation/Methode gibt ein neues gedrehtes Bild zurück.

In diesem Artikel wird die Methode zum Drehen eines bestimmten Bilds um einen bestimmten Winkel in PyTorch erläutert.

Wie drehe ich ein gewünschtes Bild in PyTorch um einen bestimmten Winkel?

Um ein gewünschtes Bild in PyTorch um einen bestimmten Winkel zu drehen, sehen Sie sich die unten aufgeführten Schritte an:







Schritt 1: Laden Sie ein Bild auf Google Colab hoch

Öffnen Sie zunächst Google Colab und klicken Sie auf die unten hervorgehobenen Symbole. Wählen Sie dann das spezifische Bild vom Computer aus und laden Sie es hoch:





Anschließend wird das Bild auf Google Colab hochgeladen:





Hier haben wir das folgende Bild hochgeladen und werden es um einen bestimmten Winkel drehen:



Schritt 2: Erforderliche Bibliothek importieren

Anschließend importieren Sie die erforderlichen Bibliotheken. Beispielsweise haben wir die folgenden Bibliotheken importiert:

Taschenlampe importieren
Torchvision.transforms importieren als T
aus PIL-Importbild

Hier:

  • Taschenlampe importieren ” importiert die PyTorch-Bibliothek.
  • importiere Torchvision.transforms als T „Importiert das Transformationsmodul von Torchvision, das zur Vorverarbeitung von Bilddaten verwendet wird, bevor sie in ein neuronales Netzwerk eingespeist werden.
  • „aus PIL-Importbild“ wird zum Öffnen und Speichern verschiedener Bilddateiformate verwendet:

Schritt 3: Lesen Sie das Eingabebild

Lesen Sie dann das Eingabebild vom Computer. Hier lesen wir die „mein_img.jpg“ und speichern Sie es in der „input_img“ Variable:

input_img = Bild. offen ( 'my_img.jpg' )

Schritt 4: Definieren Sie eine Transformation

Definieren Sie als Nächstes eine Transformation, um das Eingabebild zu drehen. Benutzer müssen den gewünschten Gradbereich (min., max.) angeben. Hier haben wir (30, 60) Grad angegeben und das neue Bild wird mit einem beliebigen zufälligen Winkel aus diesem Bereich gedreht:

verwandeln = T. RandomRotation ( Grad = ( 30 , 60 ) )

Schritt 5: Wenden Sie die Transformation auf das Eingabebild an

Drehen Sie nun das Eingabebild mithilfe der oben definierten Transformation um bestimmte Winkel:

rotierte_img = verwandeln ( input_img )

Schritt 6: Gedrehtes Bild anzeigen

Sehen Sie sich abschließend das gedrehte Bild an, indem Sie es anzeigen:

rotierte_img

Die obige Ausgabe zeigt, dass das Eingabebild erfolgreich um zufällige Winkel im angegebenen Bereich von 30 und 60 Grad gedreht wurde.

Ebenso können Benutzer auch andere Gradbereiche angeben, um das Bild zu drehen. Hier geben wir den folgenden Bereich an, um den Unterschied zu erkennen:

verwandeln = T. RandomRotation ( Grad = ( 90 , 120 ) )

Dadurch wird ein neues Bild erstellt, das mit einem beliebigen zufälligen Winkel aus dem angegebenen Bereich von 90 und 120 Grad gedreht wird:

Vergleich

Der Vergleich zwischen dem Originalbild und um verschiedene Winkel gedrehten Bildern ist unten zu sehen:

Notiz : Hier können Sie auf unser Google Colab Notebook zugreifen Verknüpfung .

Wir haben die Methode zum Drehen eines Bildes um verschiedene Winkel in PyTorch effizient erklärt.

Abschluss

Um das gewünschte Bild in PyTorch um einen bestimmten Winkel zu drehen, laden Sie zunächst das gewünschte Bild auf Google Colab hoch. Importieren Sie dann die erforderlichen Bibliotheken und lesen Sie das Eingabebild. Definieren Sie anschließend die Transformation mit „ RandomRotation() ”-Methode und wenden Sie sie auf das Eingabebild an. Zeigen Sie abschließend das neue gedrehte Bild an. In diesem Artikel wurde die Methode zum Drehen des gewünschten Bildes um einen bestimmten Winkel in PyTorch veranschaulicht.