開始使用免費開始

建立卷積式網路

你正在一個團隊中建置天氣預報系統。系統的一部分會在多個地點安裝相機,拍攝天空的影像。你的任務是建立一個模型,將影像中的不同雲層類型加以分類,協助偵測接近中的鋒面。

你決定建立一個卷積式影像分類器。模型會由兩個部分組成:

  • 特徵擷取器,從輸入影像中學習一個特徵向量;
  • 分類器,根據學到的特徵預測影像的類別。

torchtorch.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),
            ____,
            ____,
            ____,
            ____,
        )
編輯並執行程式碼