1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch로 배우는 이미지 딥러닝
Connected

연습 문제

U-Net 만들기: forward 메서드

인코더와 디코더 레이어가 정의되어 있으므로 이제 U-Net의 forward() 메서드를 구현할 수 있어요. 입력은 이미 인코더를 거쳐 제공됩니다. 이제 마지막 디코더 블록을 정의해야 합니다.

디코더의 목표는 피처 맵을 업샘플링하여 출력의 높이와 너비가 U-Net 입력 이미지와 같아지도록 만드는 것입니다. 이렇게 하면 픽셀 단위의 시맨틱 마스크를 얻을 수 있어요.

지침

100 XP
  • torch.cat()을 사용해 스킵 커넥션을 구성하면서 마지막 디코더 블록을 정의하세요.