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 ảnh bằng mặt nạ

Với mặt nạ nhị phân đã sẵn sàng, bạn có thể dùng nó để tách đối tượng — con mèo — ra khỏi ảnh.

Để làm điều đó, bạn sẽ cần tải ảnh gốc và chuyển nó thành một tensor. Tiếp theo, bạn sẽ tạo tensor đối tượng bằng cách áp mặt nạ lên ảnh gốc. Cuối cùng, bạn sẽ hiển thị kết quả.

transforms từ torchvision đã được nhập, và binary_mask bạn đã tạo ở bài tập trước hiện có sẵn cho bạn.

Hướng dẫn

100 XP
  • Tạo image_tensor bằng cách áp dụng phép biến đổi ToTensor(), đã được định nghĩa là transform, lên ảnh thô.
  • Phân đoạn ảnh bằng cách áp mặt nạ binary_mask lên image_tensor, gán kết quả cho object_tensor.
  • Áp dụng phép biến đổi to_pil_image đã định nghĩa sẵn lên object_tensor để hiển thị nó.