(비공식) DragGAN 모델 및 코드 공개

소개

일전에 DragGAN(Drag-your-GAN) 모델을 소개해드렸었는데요,

DragGAN의 공식 GitHub에는 아직 코드가 안 올라온 상태입니다. (6월쯤 공개 예정이라고 합니다)

Code will be released in June.


이 와중에, DragGAN을 (논문 저자가 아닌 제 3자가) 구현한 비공식 구현체가 공개되었습니다!

(아래 영상을 보시면 엄청 빠르게 생성이 되지는 않고, 조금씩 생성을 하면서 새로고침되는 형태로 구현이 되어 있습니다.)

소개 영상 1

소개 영상 2

온라인 데모와 함께 코드, PyPI 패키지 등이 모두 공개되어 있으니 관심있으신 분들께서는 직접 설치해서 사용해보셔도 좋을 것 같습니다. :smiley:

GitHub 저장소

프로젝트 홈페이지

데모 (Google Colab)

로컬 배포

설치 (pip)

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

설치(conda)

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

실행(GPU)

python -m draggan.web

실행(CPU)

python -m draggan.web --device cpu

로컬 데모 (Web)

http://127.0.0.1:7860 주소로 접근하여 확인


1개의 좋아요