MulaiMulai sekarang secara gratis

Membangun jaringan konvolusional

Anda berada dalam tim yang membangun sistem prakiraan cuaca. Sebagai bagian dari sistem, kamera akan dipasang di berbagai lokasi untuk mengambil foto langit. Tugas Anda adalah membangun model untuk mengklasifikasikan berbagai tipe awan pada foto-foto tersebut, yang akan membantu mendeteksi front cuaca yang mendekat.

Anda memutuskan untuk membangun pengklasifikasi citra berbasis konvolusi. Model akan terdiri dari dua bagian:

  • Sebuah pengekstrak fitur yang mempelajari vektor fitur dari citra masukan,
  • Sebuah pengklasifikasi yang memprediksi kelas citra berdasarkan fitur yang dipelajari.

Baik torch maupun torch.nn as nn sudah diimpor untuk Anda, jadi mari kita mulai!

Latihan ini adalah bagian dari kursus

Deep Learning Lanjutan dengan PyTorch

Lihat Kursus

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

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),
            ____,
            ____,
            ____,
            ____,
        )
Edit dan Jalankan Kode