Segmentace obrázku pomocí masky

Teď, když máš binární masku připravenou, můžeš ji použít k segmentaci objektu – v tomto případě kočky – z obrázku.

Nejdřív načteš původní obrázek a převedeš ho na tensor. Poté vytvoříš tensor objektu maskováním původního obrázku. Nakonec zobrazíš výsledek.

transforms z torchvision jsou již naimportované a binary_mask, kterou jsi vytvořil/a v předchozím cvičení, máš k dispozici.

Vytvoř image_tensor aplikováním transformace ToTensor(), definované jako transform, na původní obrázek.
Segmentuj obrázek maskováním image_tensor pomocí binary_mask a výsledek ulož do object_tensor.
Aplikuj již definovanou transformaci to_pil_image na object_tensor, aby se obrázek zobrazil.