Comment faire pivoter une image selon un angle spécifique dans PyTorch ?

Comment Faire Pivoter Une Image Selon Un Angle Specifique Dans Pytorch



PyTorch fournit le module « torchvision.transforms » qui propose de nombreuses transformations essentielles pour effectuer diverses tâches. Ce module a également le « Rotation aléatoire() ' méthode qui applique une rotation aléatoire à une image dans une plage spécifiée d'angles en degrés. Cette transformation/méthode renvoie une nouvelle image pivotée.

Cet article illustrera la méthode pour faire pivoter une image particulière d'un angle spécifique dans PyTorch.

Comment faire pivoter une image souhaitée selon un angle spécifique dans PyTorch ?

Pour faire pivoter une image souhaitée selon un angle spécifique dans PyTorch, consultez les étapes ci-dessous :







Étape 1 : Télécharger une image sur Google Colab

Tout d’abord, ouvrez Google Colab et cliquez sur les icônes en surbrillance ci-dessous. Ensuite, choisissez l'image spécifique sur l'ordinateur et téléchargez-la :





Par la suite, l'image sera téléchargée sur Google Colab :





Ici, nous avons téléchargé l'image suivante et nous allons la faire pivoter d'un angle spécifique :



Étape 2 : Importer la bibliothèque requise

Après cela, importez les bibliothèques nécessaires. Par exemple, nous avons importé les bibliothèques suivantes :

importer une torche
importer torchvision.transforms comme T
à partir de l'image d'importation PIL

Ici:

  • ' importer une torche » importe la bibliothèque PyTorch.
  • ' importer torchvision.transforms en tant que T ' importe le module de transformation de torchvision qui est utilisé pour prétraiter les données d'image avant de les introduire dans un réseau neuronal.
  • « à partir de l'image d'importation PIL » est utilisé pour ouvrir et enregistrer différents formats de fichiers image :

Étape 3 : Lire l'image d'entrée

Ensuite, lisez l’image d’entrée depuis l’ordinateur. Ici, nous lisons le 'mon_img.jpg' et le stocker dans le 'entrée_img' variable:

entrée_img = Image. ouvrir ( 'mon_img.jpg' )

Étape 4 : définir une transformation

Ensuite, définissez une transformation pour faire pivoter l'image d'entrée. Les utilisateurs doivent spécifier la plage souhaitée de degrés (min, max). Ici, nous avons spécifié (30, 60) degrés et la nouvelle image sera pivotée avec n'importe quel angle aléatoire sélectionné dans cette plage :

transformer = T. Rotation aléatoire ( degrés = ( 30 , 60 ) )

Étape 5 : appliquer la transformation sur l'image d'entrée

Maintenant, faites pivoter l'image d'entrée selon les angles spécifiés à l'aide de la transformation définie ci-dessus :

rotation_img = transformer ( entrée_img )

Étape 6 : Afficher l'image pivotée

Enfin, visualisez l'image pivotée en l'affichant :

rotation_img

Le résultat ci-dessus montre que l'image d'entrée a été pivotée avec succès selon des angles aléatoires dans la plage spécifiée de 30 et 60 degrés.

De même, les utilisateurs peuvent également spécifier d'autres plages de degrés pour faire pivoter l'image. Ici, nous préciserons la plage suivante pour voir la différence :

transformer = T. Rotation aléatoire ( degrés = ( 90 , 120 ) )

Cela générera une nouvelle image qui sera pivotée avec n'importe quel angle aléatoire sélectionné dans la plage spécifiée de 90 et 120 degrés :

Comparaison

La comparaison entre l’image originale et les images pivotées sous différents angles est visible ci-dessous :

Note  : Vous pouvez accéder à notre bloc-notes Google Colab à cette adresse lien .

Nous avons expliqué efficacement la méthode pour faire pivoter une image sous différents angles dans PyTorch.

Conclusion

Pour faire pivoter l'image souhaitée selon un angle spécifique dans PyTorch, commencez par télécharger l'image souhaitée sur Google Colab. Ensuite, importez les bibliothèques nécessaires et lisez l’image d’entrée. Après cela, définissez la transformation en utilisant le ' Rotation aléatoire() » et appliquez-la à l’image d’entrée. Enfin, affichez la nouvelle image pivotée. Cet article a illustré la méthode pour faire pivoter l'image souhaitée d'un angle spécifique dans PyTorch.