csinva/gan-vae-pretrained-pytorch
📦 오픈 소스 프로젝트csinva
MNIST 및 CIFAR 데이터셋을 위한 PyTorch 기반 사전 학습된 GAN, VAE 및 분류기 모음.
이 저장소는 PyTorch 워크플로우에 생성 모델을 통합하려는 개발자와 연구자를 위한 가교 역할을 합니다. 컴퓨터 비전 벤치마크에 최적화된 심층 합성곱 생성적 적대 신경망(DCGAN)과 변분 오토인코더(VAE) 구현체를 포함하고 있습니다. 사용자가 사전 학습된 가중치를 로드하여 추론을 수행하거나, 잠재 공간을 시각화하거나, 분류기를 하위 작업을 위한 특징 추출기로 활용할 수 있도록 구조화되어 있습니다. 모든 코드는 Jupyter Notebook 형식으로 제공되어 교육 목적과 빠른 프로토타이핑에 매우 적합합니다. 이러한 구현을 표준화함으로써 사용자가 GAN과 VAE의 아키텍처적 특성을 이해하고, 생성 딥러닝 연구를 위한 안정적인 베이스라인을 확보할 수 있도록 돕습니다.
💡하이라이트
- ├─사전 학습된 DCGAN 및 VAE 모델 제공
- ├─MNIST 및 CIFAR 데이터셋 지원
- └─Jupyter Notebook 기반 구현
🎯대상
- ├─AI 연구원
- └─딥러닝 학생