Segmentasi gambar dengan mask

Dengan mask biner yang sudah siap, Anda dapat menggunakannya untuk melakukan segmentasi objek—yaitu kucing—dari gambar.

Untuk melakukannya, Anda perlu memuat gambar asli dan mengubahnya menjadi tensor. Selanjutnya, Anda akan membuat tensor objek dengan melakukan masking pada gambar asli. Terakhir, Anda akan menampilkan hasilnya.

transforms dari torchvision telah diimpor, dan binary_mask yang Anda buat pada latihan sebelumnya tersedia untuk digunakan.

Latihan ini adalah bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Petunjuk latihan

Buat image_tensor dengan menerapkan transform ToTensor(), yang didefinisikan sebagai transform, pada gambar mentah.
Lakukan segmentasi gambar dengan melakukan masking image_tensor menggunakan binary_mask, lalu simpan hasilnya ke object_tensor.
Terapkan transform to_pil_image yang sudah didefinisikan pada object_tensor untuk menampilkannya.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Load image and transform to tensor
image = Image.open("images/Egyptian_Mau_123.jpg")
transform = transforms.Compose([transforms.ToTensor()])
image_tensor = ____

# Segment object out of the image
object_tensor = ____

# Convert segmented object to image and display
to_pil_image = ____
object_image = to_pil_image(object_tensor)
plt.imshow(object_image)
plt.show()

Edit dan Jalankan Kode