PyTorch ialah rangka kerja pembelajaran mendalam yang terkenal yang menawarkan pelbagai alat dan modul untuk bekerja dengan banyak imej. Ia juga menyediakan modul 'torchvision.transforms' yang mempunyai set kelas dan fungsi untuk melakukan pelbagai transformasi pada imej, seperti memangkas, mengubah saiz, memutar, membalikkan, menskala dan banyak lagi. Pengguna boleh menggunakan ' Ubah saiz() ” kaedah dan nyatakan saiz iaitu, tinggi dan lebar untuk mengubah saiz mana-mana imej yang dikehendaki. Kaedah ini mengembalikan imej bersaiz baharu bagi saiz yang ditentukan.
Blog ini akan menunjukkan kaedah mengubah saiz imej kepada saiz tertentu dalam PyTorch.
Bagaimana untuk Mengubah Saiz Imej kepada Saiz Tertentu dalam PyTorch?
Untuk menukar dimensi imej dan mengubah saiznya kepada saiz tertentu dalam PyTorch, ikuti langkah yang disenaraikan di bawah:
- Muat naik imej ke Google Colab
- Import perpustakaan yang diperlukan
- Baca imej input
- Kira dan cetak saiz imej input
- Buat transformasi
- Gunakan transformasi pada imej yang ditentukan
- Paparkan imej yang diubah saiz dan saiznya
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 mengubah saiznya kepada saiz tertentu:
Langkah 2: Import Pustaka Diperlukan
Seterusnya, import perpustakaan yang diperlukan. Sebagai contoh, kami telah mengimport perpustakaan berikut:
obor importimport torchvision.transforms sebagai Berubah
daripada PIL import Image
import matplotlib.pyplot sebagai plt
di sini:
- “ obor import ” mengimport perpustakaan PyTorch.
- “ import torchvision.transforms as transforms ” mengimport modul ubah daripada torchvision yang digunakan untuk pra-memproses data imej sebelum memasukkannya ke dalam rangkaian saraf.
- “ daripada PIL import Image ” membuka dan menyimpan format fail imej yang berbeza.
- “ import matplotlib.pyplot sebagai plt ' mengimport modul 'pyplot' yang digunakan untuk membuat visualisasi dan plot:
Langkah 3: Baca Imej Input
Selepas itu, baca imej input dari komputer. Di sini, kita sedang membaca ' galaxy_img.jpg 'dan menyimpannya dalam' input_img ” pembolehubah:
input_img = Imej.terbuka ( 'galaxy_img.jpg' )
Langkah 4: Kira dan Cetak Saiz Imej Input
Kemudian, kira saiz imej input iaitu, lebar dan tinggi, dan cetaknya:
saiz = input_img.sizecetak ( 'Saiz imej (input) asal:' , saiz )
Ia boleh diperhatikan bahawa lebar imej input ialah '384' dan ketinggian imej ialah '576':
Langkah 5: Buat Transformasi
Seterusnya, tentukan transformasi untuk mengubah saiz imej input. Pengguna perlu menentukan saiz baharu iaitu, tinggi dan lebar untuk imej baharu. Di sini, kami telah menentukan ketinggian '200' dan lebar '400':
transform = Transform.Resize ( saiz = ( 200 , 400 ) )
Langkah 6: Gunakan Transformasi pada Imej Input
Sekarang, gunakan transformasi di atas pada imej input yang ditentukan untuk menukar dimensinya:
new_img = transform ( input_img )
Langkah 7: Paparkan Imej Bersaiz Semula dan Saiznya
Akhir sekali, lihat imej yang diubah saiz dengan memaparkannya dan mengesahkan saiznya:
cetak ( 'Saiz imej baharu selepas mengubah saiz:' , new_img.size )plt.imshow ( new_img )
plt.show ( )
Output di bawah menunjukkan bahawa imej input telah berjaya diubah saiznya. Kini, lebarnya ialah '400' dan ketinggiannya ialah '200':
Begitu juga, pengguna boleh menentukan sebarang saiz lain untuk mengubah saiz imej. Di sini, kami akan mengubah saiz imej input yang sama dengan saiz lain iaitu, ketinggian '250' dan lebar '150':
transform = Transform.Resize ( saiz = ( 250 , 150 ) )
Ini akan mengubah saiz imej kepada dimensi baharu:
Perbandingan
Perbandingan antara imej asal dan imej diubah saiz dengan dimensi berbeza boleh dilihat di bawah:
Catatan : Anda boleh mengakses Buku Nota Google Colab kami di sini pautan .
Kami telah menerangkan dengan cekap kaedah untuk mengubah saiz imej kepada saiz yang dikehendaki dalam PyTorch.
Kesimpulan
Untuk mengubah saiz imej yang dikehendaki kepada saiz tertentu dalam PyTorch, mula-mula, muat naik imej yang dikehendaki ke Google Colab. Kemudian, import perpustakaan yang diperlukan dan baca imej input. Seterusnya, kira dan cetak saiz imej input. Selepas itu, gunakan ' Ubah saiz() ” kaedah untuk menentukan dan menggunakan transformasi pada imej input yang dikehendaki. Akhir sekali, paparkan imej bersaiz baharu dan saiznya. Blog ini telah menunjukkan kaedah untuk mengubah saiz imej yang dikehendaki kepada saiz tertentu dalam PyTorch.