1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Intermediate Deep Learning with PyTorch
Connected

cvičení

Sestavení konvolučních sítí

Pracuješ v týmu, který vyvíjí systém pro předpověď počasí. Součástí systému jsou kamery rozmístěné na různých místech, které pořizují snímky oblohy. Tvým úkolem je sestavit model pro klasifikaci různých typů mraků na těchto snímcích – to pomůže odhalit blížící se atmosférické fronty.

Rozhodneš se postavit konvoluční klasifikátor obrázků. Model se skládá ze dvou částí:

  • Extraktor příznaků, který se z vstupního obrázku naučí vektor příznaků,
  • Klasifikátor, který na základě naučených příznaků předpoví třídu obrázku.

torch i torch.nn as nn jsou už naimportované, takže se můžeme pustit do práce!

Pokyny 1/3

undefined XP
    1
    2
    3
  • Definuj část modelu feature_extractor přidáním další konvoluční vrstvy s 64 výstupními mapami příznaků, aktivační funkcí ELU a vrstvou max poolingu s oknem velikosti dva; na konci výstup zploštěte.