Phân đoạn ảnh bằng mặt nạ
Với mặt nạ nhị phân đã sẵn sàng, bạn có thể dùng nó để tách đối tượng — con mèo — ra khỏi ảnh.
Để làm điều đó, bạn sẽ cần tải ảnh gốc và chuyển nó thành một tensor. Tiếp theo, bạn sẽ tạo tensor đối tượng bằng cách áp mặt nạ lên ảnh gốc. Cuối cùng, bạn sẽ hiển thị kết quả.
transforms từ torchvision đã được nhập, và binary_mask bạn đã tạo ở bài tập trước hiện có sẵn cho bạn.
Bài tập này là một phần của khóa học
Deep Learning cho Ảnh với PyTorch
Hướng dẫn bài tập
- Tạo
image_tensorbằng cách áp dụng phép biến đổiToTensor(), đã được định nghĩa làtransform, lên ảnh thô. - Phân đoạn ảnh bằng cách áp mặt nạ
binary_masklênimage_tensor, gán kết quả choobject_tensor. - Áp dụng phép biến đổi
to_pil_imageđã định nghĩa sẵn lênobject_tensorđể hiển thị nó.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# 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()