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

Bài tập

Phân đoạn với Mask R-CNN đã được huấn luyện sẵn

Trong bài tập này, bạn sẽ dùng mô hình Mask R-CNN đã được huấn luyện sẵn để thực hiện instance segmentation trên bức ảnh sau gồm hai con mèo.

two cats image

Mô hình bạn sẽ dùng đã được huấn luyện trước trên COCO dataset, bộ dữ liệu chứa ảnh các vật thể phổ biến, bao gồm cả động vật. Nhờ vậy, mô hình có thể nhận diện mèo ngay lập tức mà không cần fine-tune.

Nhiệm vụ của bạn là tải mô hình và ảnh hai con mèo, chuẩn bị ảnh, rồi truyền vào mô hình để lấy dự đoán. Image từ PIL, torch, transforms từ torchvision, và maskrcnn_resnet50_fpn đã được nhập sẵn cho bạn.

Hướng dẫn

100 XP
  • Tải Mask R-CNN pretrained vào model bằng maskrcnn_resnet50_fpn().
  • Biến đổi ảnh hai con mèo thành tensor và thêm một chiều (unsqueeze).
  • Thực hiện suy luận bằng cách truyền ảnh vào mô hình và gán đầu ra cho prediction.