[Infra] #18의 필수 가중치 획득

[oMFDOo]

#18 에서 stylegan2_ffhq.pt, albedogan.pt, disp_GAN.pt를 요구하게 된다.
이 파일들을 찾아오자

    ## Initialize StyleGAN2

    stylegan2_gen = Generator(1024, 512, 8, channel_multiplier=2)
    stylegan2_gen.to(args.device)
    ckpt_path = 'stylegan2/training_logs/checkpoint/stylegan2_ffhq.pt'
    ckpt = torch.load(ckpt_path, map_location=lambda storage, loc: storage)
    stylegan2_gen.load_state_dict(ckpt["g_ema"], strict=True)
    # mean_latent = stylegan2_gen.mean_latent(4096)


    ## Initialize AlbedoGAN

    albedogan = Generator(1024, 512, 8, channel_multiplier=2)
    albedogan = torch.nn.DataParallel(albedogan)
    albedogan.to(args.device)
    ckpt_path = 'stylegan2/training_logs/checkpoint/albedogan.pt'
    ckpt = torch.load(ckpt_path, map_location=lambda storage, loc: storage)
    albedogan.load_state_dict(ckpt["g"], strict=True)

    ## Initialize Displacement Generator

    disp_GAN = Generator(256, 512, 8, channel_multiplier=2)
    disp_GAN = torch.nn.DataParallel(disp_GAN)
    disp_GAN.to(args.device)
    ckpt_path = 'stylegan2/training_logs/checkpoint/disp_GAN.pt'
    ckpt = torch.load(ckpt_path, map_location=lambda storage, loc: storage)
    disp_GAN.load_state_dict(ckpt["g"], strict=True)

[oMFDOo]

stylegan2_ffhq.pt

: StyleGAN2의 FFHQ 가중치 다운로드

• stylegan2_ffhq.pt : StyleGAN2가 학습한 FFHQ(Flickr-Faces-HQ) 데이터셋의 가중치

공식 문서에서 제공한다.

다운 받은 이후에는 이렇게 생겼다.

이름을 바꿔주어 stylegan2/training_logs/checkpoint/stylegan2_ffhq.pt 경로에 넣어준다.

[oMFDOo]

가중치 획득 불가 판정

: 기존 가중치의 문제가 있어 재학습 후 업로드 됨이 고지되었음 테스트 만을 위해서라도 임시 가중치 요청을 하였음.

• 문제 정의
• 남긴 문의