Hiển thị soft mask
Trong bài trước, bạn đã thấy rằng hai đối tượng có khả năng cao nhất mà mô hình Mask R-CNN phân đoạn được đều là mèo. Bây giờ, bạn sẽ hiển thị các mask của hai chú mèo này đè lên ảnh gốc để kiểm tra trực quan độ chính xác. Bạn sẽ cần lặp qua hai mask, và với mỗi mask, vẽ ảnh gốc rồi phủ lên trên một lớp mask bán trong suốt.
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
- Trích xuất các mask và nhãn từ
prediction, gán lần lượt chomasksvàlabels. - Bên trong vòng lặp for, hiển thị mask thứ
ilên ảnh bằng cách truyềnmask[i, 0]cho hàm vẽ, sử dụng bản đồ màu"jet"và đặt tham số độ trong suốt là0.5.
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.
# Extract masks and labels from prediction
masks = ____
labels = ____
# Plot image with two overlaid masks
for i in range(2):
plt.imshow(image)
# Overlay the i-th mask on top of the image
plt.imshow(____, ____, ____)
plt.title(f"Object: {class_names[labels[i]]}")
plt.show()