Создание бинарных масок

Изображения для задач сегментации обычно размечаются с помощью масок на уровне пикселей. Рассмотрим изображение кошки породы египетская мау.

cat image

В этом и следующем упражнении вы будете использовать соответствующую маску, чтобы выделить кошку на изображении. Сначала необходимо загрузить маску и преобразовать её в бинарную.

Image из PIL, transforms из torchvision и torch уже импортированы для вас.

Это упражнение является частью курса

Глубокое обучение для работы с изображениями на PyTorch

Посмотреть курс

Инструкции к упражнению

Загрузите изображение маски, хранящееся в annotations/Egyptian_Mau_123.png, и присвойте его переменной mask.
Создайте binary_mask на основе mask_tensor: каждому пикселю со значением 1/255 присвойте тензорное значение 1.0, остальным пикселям — тензорное значение 0.0.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# Load mask image
mask = ____

# Transform mask to tensor
transform = transforms.Compose([transforms.ToTensor()])
mask_tensor = transform(mask)

# Create binary mask
binary_mask = ____(
    ____, 
    ____,
    ____,
)

# Print unique mask values
print(binary_mask.unique())

Редактировать и запускать код