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!
Acest exercițiu face parte din cursul
Deep Learning intermediar cu PyTorch
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
class Net(nn.Module):
def __init__(self, num_classes):
super().__init__()
# Define feature extractor
self.feature_extractor = nn.Sequential(
nn.Conv2d(3, 32, kernel_size=3, padding=1),
nn.ELU(),
nn.MaxPool2d(kernel_size=2),
____,
____,
____,
____,
)