Xây dựng mạng tích chập
Bạn đang ở trong một nhóm xây dựng hệ thống dự báo thời tiết. Là một phần của hệ thống, camera sẽ được lắp đặt tại nhiều vị trí để chụp ảnh bầu trời. Nhiệm vụ của bạn là xây dựng một mô hình phân loại các loại mây khác nhau trong những bức ảnh này, giúp phát hiện các dải thời tiết đang đến gần.
Bạn quyết định xây dựng một bộ phân loại ảnh dùng tích chập. Mô hình sẽ gồm hai phần:
- Một bộ trích xuất đặc trưng học ra một vector đặc trưng từ ảnh đầu vào,
- Một bộ phân loại dự đoán lớp của ảnh dựa trên các đặc trưng đã học.
Cả torch và torch.nn as nn đã được nhập sẵn cho bạn, hãy bắt đầu thôi!
Bài tập này là một phần của khóa học
Deep Learning nâng cao với PyTorch
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
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),
____,
____,
____,
____,
)