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

Bài tập

Xây dựng U-Net: phương thức forward

Với các lớp encoder và decoder đã được định nghĩa, bây giờ bạn có thể triển khai phương thức forward() của U-Net. Đầu vào đã được đưa qua encoder sẵn cho bạn. Tuy nhiên, bạn cần định nghĩa khối decoder cuối cùng.

Mục tiêu của decoder là upsample các feature map để đầu ra có cùng chiều cao và chiều rộng với ảnh đầu vào của U-Net. Điều này giúp bạn tạo được các mặt nạ ngữ nghĩa ở cấp độ pixel.

Hướng dẫn

100 XP
  • Định nghĩa khối decoder cuối cùng, dùng torch.cat() để tạo skip connection.