Створення бінарних масок

Зображення для задач сегментації зазвичай анотують масками на рівні пікселів. Розгляньте це зображення кота породи єгипетський мау.

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())

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