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

cvičení

Blok klasifikátoru

Tvým dalším úkolem je vytvořit blok klasifikátoru, který nahradí původní klasifikátor VGG16. Použiješ blok se dvěma plně propojenými vrstvami s aktivací ReLU mezi nimi.

Proměnné vgg_model a input_dim, které jsi definoval/a v předchozím cvičení, jsou dostupné v tvém pracovním prostoru. Knihovny torch a torchvision.models jsou již naimportovány.

Pokyny

100 XP
  • Vytvoř proměnnou num_classes s počtem tříd — pracuješ pouze s detekcí koček a psů.
  • Vytvoř sekvenční blok pomocí nn.Sequential.
  • Přidej lineární vrstvu s in_features nastaveným na input_dim.
  • Doplň výstupní features do poslední vrstvy klasifikátoru.