StudioGAN은 포항공대 CVLab 강민국님께서 공개하신 pytorch 기반의 오픈소스 라이브러리로서 다양한 GAN 알고리즘들이 구현되어 있습니다. 포함되어 있는 GAN 알고리즘에는 DCGAN, LSGAN, WGAN 등 주요 알고리즘들이 다수 포함되어 있어 상호 성능 비교 등 연구 목적으로 매우 유용할 것으로 생각됩니다.
코드 설명에 따르면 각 알고리즘들의 원래 구현체보다 메모리 요구량이나 복잡도 면에서 최적화가 이루어졌다고 하며 pre-trained model 또한 공개되어 있습니다. 그 밖에도 multi-GPU, mixed precision, synchronized batch normalization 등 최신 기술들도 반영되어 있어서 GAN 관련 연구를 위한 최고의 시작점 중 하나가 아닐까 생각합니다. 다음은 StudioGAN의 github 저장소 링크입니다.
POSTECH-CVLab/PyTorch-StudioGAN
StudioGAN is a Pytorch library providing implementations of representative Generative Adversarial Networks (GANs) for conditional/unconditional image generation. – POSTECH-CVLab/PyTorch-StudioGAN
