1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Bài tập

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.

Hướng dẫn

100 XP
  • Trích xuất các mask và nhãn từ prediction, gán lần lượt cho masks và labels.
  • Bên trong vòng lặp for, hiển thị mask thứ i lên ảnh bằng cách truyền mask[i, 0] cho hàm vẽ, sử dụng bản đồ màu "jet" và đặt tham số độ trong suốt là 0.5.