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

Connected

Bài tập

Chồng lớp các instance mask

Làm rất tốt khi tạo được semantic mask ở bài trước! Giờ bạn sẽ ghi đè nó bằng các instance mask tại những vị trí mà mô hình instance segmentation đã nhận diện được đối tượng.

Bạn sẽ dùng MaskRCNN đã được huấn luyện sẵn trong workspace để tạo ra các mặt nạ instance segmentation. Sau đó, bạn sẽ lặp qua các mặt nạ này và với mỗi cái, bạn sẽ chồng những phần có độ chắc chắn cao rằng có đối tượng lên trên semantic mask.

torch đã được import sẵn cho bạn.

Hướng dẫn

100 XP
  • Khởi tạo panoptic_mask bằng cách sao chép semantic_mask.
  • Định nghĩa vòng lặp for để lặp qua các instance mask, đặt tên biến vòng lặp là mask.
  • Với mỗi instance mask, tại những vị trí lớn hơn 0.5, ghi đè panoptic mask bằng instance_id hiện tại.