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ạy semantic segmentation

Làm tốt lắm khi thiết kế U-Net! Bạn đã có sẵn một mô hình đã được huấn luyện trước, rất giống với mô hình bạn vừa xây dựng. Mô hình này được huấn luyện trên một tập ảnh lớn và có thêm một vài chi tiết trong kiến trúc, như các lớp batch norm.

Bạn có thể khởi tạo mô hình bằng UNet(), mô hình sẽ tự động dùng các trọng số đã huấn luyện. Nhiệm vụ của bạn là dùng nó để tạo mặt nạ phân đoạn cho bức ảnh chiếc xe dưới đây.

car image

Image từ PIL đã được nhập sẵn cho bạn.

Hướng dẫn

100 XP
  • Khởi tạo UNet() vào một biến tên model.
  • Tải ảnh tại car.jpg vào một biến tên image.
  • Tạo mặt nạ phân đoạn bằng cách truyền ảnh vào mô hình và squeeze(0) đầu ra.