1. Learn
    2. /
  2. Cursuri
    3. /
  3. Deep Learning intermediar cu PyTorch
Connected

exercițiu

Construirea rețelelor convoluționale

Faci parte dintr-o echipă care dezvoltă un sistem de prognoză meteo. Ca parte a acestui sistem, camere vor fi instalate în diverse locații pentru a fotografia cerul. Sarcina ta este să construiești un model care să clasifice diferite tipuri de nori din aceste imagini, ajutând astfel la detectarea fronturilor meteo.

Decizi să construiești un clasificator convoluțional de imagini. Modelul va fi alcătuit din două componente:

  • Un extractor de caracteristici care învață un vector de caracteristici din imaginea de intrare,
  • Un clasificator care prezice clasa imaginii pe baza caracteristicilor învățate.

Atât torch, cât și torch.nn as nn au fost deja importate, așa că hai să începem!

Instrucțiuni 1/3

undefined XP
    1
    2
    3
  • Definește componenta feature_extractor a modelului adăugând un al doilea strat convoluțional cu 64 hărți de caracteristici de ieșire, activarea ELU și un strat de max pooling cu o fereastră de dimensiune doi; la final, aplatizează ieșirea.