StyleGAN2 Pytorch 사용법

https://github.com/rosinality/stylegan2-pytorch/tree/bb459e012489eb4629004d486f05f65fde554ba7

GitHub - rosinality/stylegan2-pytorch: Implementation of Analyzing and Improving the Image Quality of StyleGAN (StyleGAN 2) in P

다들 논문분석 혹은 결과물 분석만하고 막상 초보들을 위한 사용법이 없었다. github에 나와있긴한데 이런 분야에 약하다면 방법이 익숙치 않아 어려울 수 있다. 나도 처음 이런 오픈 AI 모델을 사용할 때 삽질한 경험이 있어서 올려놓고자 한다.

 

환경 : Google Colab

 

0. 필요한 라이브러리 StyleGAN2는 mdb파일로 이미지를 변환하고 학습하는데 mdb 파일을 열고닫으려면 이 라이브러리가 필요하다. 

!pip install lmdb

 

1. git clone

%cd /content/drive/MyDrive

!git clone https://github.com/rosinality/stylegan2-pytorch.git

 

2. stylegan2-pytorch로 이동 후 데이터셋 준비 실행

%cd /content/drive/MyDrive/stylegan2-pytorch/

!python prepare_data.py --out=/content/drive/MyDrive/dataset --size=256 --n_worker=2 /content/drive/MyDrive/data

여기서 사이즈는 원하는 크기, n_worker는 코어의 수이다. 그리고 마지막 경로가 jpg들이 존재하는 위치이다. 이 때 file명/class명/jpg,png등 이렇게 있어야한다. stylegan인데 왜 이래야하는건지는 모르겠다.

 

3.  본격적인 학습

%cd /content/drive/MyDrive/stylegan2-pytorch/

 

!python train.py --batch=1 --iter=5000 --size=512 /content/drive/MyDrive/dataset/

배치사이즈, 반복횟수, 학습할 이미지의 사이즈를 설장하고 2번에서 만든 mdb 파일 경로를 설정해주면 된다.

 

이러면 학습중인것. 중간중간 sample 폴더에 샘플 파일이 저장될 것이다. 

train.py 파일의 이 부분에서 sample과 save를 얼마나 자주할 지 결정할 수 있다.