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
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),
____,
____,
____,
____,
)