1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

cvičení

Segmentace pomocí předtrénovaného Mask R-CNN

V tomto cvičení použiješ předtrénovaný model Mask R-CNN k provední instance segmentace na následujícím obrázku dvou koček.

two cats image

Model, který budeš používat, byl předtrénovaný na datové sadě COCO, která obsahuje obrázky běžných objektů včetně zvířat. Díky tomu by měl model rozpoznat kočky rovnou bez nutnosti dalšího dotrénování.

Tvým úkolem je načíst model a obrázek dvou koček, obrázek připravit a předat ho modelu, aby vrátil predikce. Image z PIL, torch, transforms z torchvision a maskrcnn_resnet50_fpn jsou již naimportované.

Pokyny

100 XP
  • Načti předtrénovaný Mask R-CNN do proměnné model pomocí maskrcnn_resnet50_fpn().
  • Transformuj obrázek dvou koček na tensor a přidej mu dimenzi pomocí unsqueeze.
  • Proveď inferenci tím, že předáš obrázek modelu, a výstup ulož do proměnné prediction.