1. Learn
    2. /
  2. Kurser
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

övning

Páteřní síť (backbone) předtrénovaného modelu

Nastal čas postavit architekturu R-CNN! Jako základ pro extrakci příznaků použiješ páteřní síť (backbone) předtrénovaného modelu vgg16. Nezapomeň si také uložit výstupní tvar páteřní sítě – ten poslouží jako vstupní tvar pro následující bloky: klasifikátor a regressor ohraničujících rámečků.

torch, torchvision a torch.nn jako nn jsou již importovány. Model byl importován jako vgg16 s váhami uloženými v VGG16_Weights.

Instruktioner

100 XP
  • Načti předtrénované váhy VGG16.
  • Pomocí .children() vytvoř sekvenční blok z classifieru, extrahuj in_features z jeho první vrstvy a ulož je jako input_dim.
  • Vytvoř páteřní síť jako sekvenční blok pomocí features a .children().
  • Vypiš model páteřní sítě.