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

연습 문제

합성곱 기반 Generator

지난 영상에서 다룬 DCGAN 가이드라인에 따라 합성곱 기반 generator를 정의하세요.

편의를 위해 torch.nn은 이미 nn으로 임포트되어 있어요. 또한 사용자 정의 함수 dc_gen_block()을 제공해 드렸으며, 이 함수는 전치 합성곱, 배치 정규화, ReLU 활성화를 묶은 블록을 반환합니다. 이 함수는 합성곱 generator를 구성하기 위한 기본 구성 요소로 사용돼요. 아래에서 dc_gen_block()의 정의를 확인해 보세요. 

def dc_gen_block(in_dim, out_dim, kernel_size, stride):
    return nn.Sequential(
        nn.ConvTranspose2d(in_dim, out_dim, kernel_size, stride=stride),
        nn.BatchNorm2d(out_dim),
        nn.ReLU()
    )

지침

100 XP
  • 마지막 generator 블록을 추가하여 feature map의 크기를 256으로 매핑하세요.
  • 출력 채널 크기가 3인 전치 합성곱을 추가하세요.
  • tanh 활성화를 추가하세요.