Bagaimana untuk Memutar Imej mengikut Sudut Tertentu dalam PyTorch?

Bagaimana Untuk Memutar Imej Mengikut Sudut Tertentu Dalam Pytorch



PyTorch menyediakan modul 'torchvision.transforms' yang mempunyai banyak perubahan penting untuk melaksanakan pelbagai tugas. Modul ini juga mempunyai ' RandomRotation() ” kaedah yang menggunakan putaran rawak pada imej dalam julat sudut tertentu dalam darjah. Transformasi/kaedah ini mengembalikan imej diputar baharu.

Artikel ini akan menggambarkan kaedah untuk memutar imej tertentu dengan sudut tertentu dalam PyTorch.

Bagaimana untuk Memutar Imej yang Diingini mengikut Sudut Tertentu dalam PyTorch?

Untuk memutar imej yang dikehendaki mengikut sudut tertentu dalam PyTorch, lihat langkah yang disenaraikan di bawah:







Langkah 1: Muat Naik Imej ke Google Colab

Mula-mula, buka Google Colab dan klik pada ikon yang diserlahkan di bawah. Kemudian, pilih imej tertentu daripada komputer dan muat naiknya:





Selepas itu, imej akan dimuat naik ke Google Colab:





Di sini, kami telah memuat naik imej berikut dan kami akan memutarkannya mengikut sudut tertentu:



Langkah 2: Import Pustaka Diperlukan

Selepas itu, import perpustakaan yang diperlukan. Sebagai contoh, kami telah mengimport perpustakaan berikut:

obor import
import torchvision.transforms sebagai T
daripada PIL import Image

di sini:

  • obor import ” mengimport perpustakaan PyTorch.
  • import torchvision.transforms as T ” mengimport modul ubah daripada torchvision yang digunakan untuk pra-memproses data imej sebelum memasukkannya ke dalam rangkaian saraf.
  • “dari PIL import Image” digunakan untuk membuka dan menyimpan format fail imej yang berbeza:

Langkah 3: Baca Imej Input

Kemudian, baca imej input dari komputer. Di sini, kita sedang membaca “my_img.jpg” dan menyimpannya dalam “input_img” pembolehubah:

input_img = Gambar. buka ( 'my_img.jpg' )

Langkah 4: Tentukan Transformasi

Seterusnya, tentukan transformasi untuk memutar imej input. Pengguna perlu menentukan julat (min, maks) darjah yang dikehendaki. Di sini, kami telah menentukan (30, 60) darjah dan imej baharu akan diputar dengan mana-mana sudut rawak yang dipilih daripada julat ini:

mengubah = T. RandomRotation ( darjah = ( 30 , 60 ) )

Langkah 5: Gunakan Transformasi pada Imej Input

Sekarang, putar imej input mengikut sudut tertentu menggunakan transformasi yang ditakrifkan di atas:

rotated_img = mengubah ( input_img )

Langkah 6: Paparkan Imej Diputar

Akhir sekali, lihat imej yang diputar dengan memaparkannya:

rotated_img

Output di atas menunjukkan bahawa imej input telah berjaya diputar dengan sudut rawak dalam julat 30, dan 60 darjah yang ditentukan.

Begitu juga, pengguna juga boleh menentukan julat darjah lain untuk memutar imej. Di sini, kami akan menentukan julat berikut untuk melihat perbezaannya:

mengubah = T. RandomRotation ( darjah = ( 90 , 120 ) )

Ini akan menghasilkan imej baharu yang akan diputar dengan mana-mana sudut rawak yang dipilih daripada julat 90 dan 120 darjah yang ditentukan:

Perbandingan

Perbandingan antara imej asal dan imej diputar mengikut sudut yang berbeza boleh dilihat di bawah:

Catatan : Anda boleh mengakses Buku Nota Google Colab kami di sini pautan .

Kami telah menerangkan dengan cekap kaedah untuk memutar imej mengikut sudut yang berbeza dalam PyTorch.

Kesimpulan

Untuk memutar imej yang dikehendaki mengikut sudut tertentu dalam PyTorch, mula-mula, muat naik imej yang dikehendaki ke Google Colab. Kemudian, import perpustakaan yang diperlukan dan baca imej input. Selepas itu, tentukan transformasi menggunakan ' RandomRotation() ” dan gunakannya pada imej input. Akhir sekali, paparkan imej baru yang diputar. Artikel ini telah menggambarkan kaedah untuk memutar imej yang dikehendaki mengikut sudut tertentu dalam PyTorch.