Tạo mặt nạ nhị phân

Ảnh dùng cho bài toán phân đoạn thường được gán nhãn bằng các mặt nạ ở cấp độ pixel. Hãy xem bức ảnh một chú mèo Ai Cập Mau dưới đây.

cat image

Trong bài này và bài tiếp theo, bạn sẽ dùng mặt nạ tương ứng để tách chú mèo ra khỏi ảnh. Đầu tiên, bạn cần tải mặt nạ và chuyển nó về dạng nhị phân.

Image từ PIL, transforms từ torchvision, và torch đã được nhập sẵn cho bạn.

Tải ảnh mặt nạ được lưu tại annotations/Egyptian_Mau_123.png và gán vào mask.
Tạo binary_mask từ mask_tensor sao cho mỗi pixel bằng 1/255 được gán giá trị tensor 1.0, và các pixel còn lại được gán giá trị tensor 0.0.