1. Learn
    2. /
  2. Courses
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Exercise

Thiết lập semantic masks

Một cách phổ biến để thực hiện panoptic segmentation là kết hợp đầu ra của semantic và instance segmentation. Hãy xem xét hình ảnh sau về một con phố ở New York.

street image

Nhiệm vụ của bạn là phân đoạn ảnh theo kiểu panoptic, sao cho mỗi chiếc taxi (cab) được nhận diện là một đối tượng riêng, đồng thời phân biệt nền đường nhựa và các tòa nhà.

Để làm được điều này, bạn sẽ bắt đầu bằng cách tạo một semantic mask với U-Net đã được huấn luyện sẵn, có sẵn dưới dạng UNet(). Hy vọng mô hình sẽ phân biệt được hai loại nền (nhưng không phân biệt các taxi cụ thể)!

Instructions

100 XP
  • Khởi tạo mô hình U-Net dưới tên model.
  • Tạo semantic_masks bằng cách truyền tensor ảnh đầu vào qua mô hình.
  • Tạo một semantic mask đơn bằng cách chọn lớp có xác suất cao nhất cho mỗi pixel.