1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Bài tập

Khối bộ phân loại

Nhiệm vụ tiếp theo của bạn là tạo một khối bộ phân loại để thay thế bộ phân loại VGG16 gốc. Bạn quyết định dùng một khối gồm hai lớp kết nối đầy đủ với một kích hoạt ReLU ở giữa.

vgg_model và input_dim mà bạn đã định nghĩa ở bài trước có sẵn trong không gian làm việc của bạn, và torch cùng torchvision.models đã được nhập.

Hướng dẫn

100 XP
  • Tạo biến num_classes với số lượng lớp, giả sử bạn chỉ phát hiện mèo và chó.
  • Tạo một khối tuần tự dùng nn.Sequential.
  • Tạo một lớp tuyến tính với in_features đặt là input_dim.
  • Thêm số đặc trưng đầu ra cho lớp cuối cùng của bộ phân loại.