建立卷積式網路
你正在一個團隊中建置天氣預報系統。系統的一部分會在多個地點安裝相機,拍攝天空的影像。你的任務是建立一個模型,將影像中的不同雲層類型加以分類,協助偵測接近中的鋒面。
你決定建立一個卷積式影像分類器。模型會由兩個部分組成:
- 特徵擷取器,從輸入影像中學習一個特徵向量;
- 分類器,根據學到的特徵預測影像的類別。
torch 與 torch.nn as nn 已為你匯入。開始動手做吧!
本練習屬於課程
Intermediate Deep Learning with PyTorch
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
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),
____,
____,
____,
____,
)