StarGAN V2 텐서플로우 모델 학습하기

 

StarGAN v2 참고자료

https://github.com/clovaai/stargan-v2-tensorflow

GitHub - clovaai/stargan-v2-tensorflow: StarGAN v2 - Official Tensorflow Implementation (CVPR 2020)

 

 

개발환경

• Intel I7 CPU + Nvidia GTX1070(nvidia-470 driver installed)
• Ubuntu 20.04
• Docker(텐서플로우)

 

Requirement

• Python 3.6이상
• Tensorflow 2.1.0
• pip: Tensorflow-addons 0.9.1
• pip: OpenCV-python
• pip: Pillow
• pip: tqdm

 

* 도커 설치 및 도커환경의 텐서플로우 설치는 아래 링크 참고

https://swiftcam.tistory.com/562

도커에서 텐서플로우 GPU를 사용하기 위한 환경설정

 

 

1. 도커 실행후 필수 프로그램 설치

$ apt-get update && apt-get install -y \
    tmux \
    curl \
    wget \
    vim \
    git \
    sudo \
    libgl1-mesa-glx \
    libgl1-mesa-dri \
    mesa-utils \
    unzip \
    locales \
    whois \
    net-tools \
    sudo

 

2. python 업데이트

$ pip install --upgrade pip
$ python --version       
Python 3.6.9
$ pip --version
pip 21.3.1 from /usr/local/lib/python3.6/dist-packages/pip (python 3.6)

 

3. tensorflow-addons 설치

tensorflow-addons는 텐서플로우와 호환되는 버전으로 사용해야 한다. 호환성 매트릭스는 아래와 같다.

 

Python Op 호환성 매트릭스

TensorFlow 애드온TensorFlow파이썬

tfa-nightly	2.3	3.6, 3.7, 3.8
tensorflow-addons-0.11.2	2.2, 2.3	3.5, 3.6, 3.7, 3.8
tensorflow-addons-0.10.0	2.2	3.5, 3.6, 3.7, 3.8
tensorflow-addons-0.9.1	2.1, 2.2	3.5, 3.6, 3.7
tensorflow-addons-0.8.3	2.1	3.5, 3.6, 3.7
tensorflow-addons-0.7.1	2.1	2.7, 3.5, 3.6, 3.7
tensorflow-addons-0.6.0	2.0	2.7, 3.5, 3.6, 3.7

$ pip install tensorflow-addons==0.9.1

 

 

4. Opencv-python 설치

$ pip install opencv-python

 

 

5. 기타 파이썬 라이브러리 설치

$ pip install pillow

$ pip install tqdm

 

 

6. 소스코드 다운로드

root@Eagle:/$ mkdir -p $HOME/git/
root@Eagle:/$ cd /$HOME/git
root@Eagle://root/git$ git clone https://github.com/clovaai/stargan-v2-tensorflow.git
root@Eagle://root/git/stargan-v2-tensorflow$ ll
total 124
drwxr-xr-x 5 root root  4096 Jan 28 22:29 ./
drwxr-xr-x 3 root root  4096 Jan 28 22:29 ../
-rw-r--r-- 1 root root  6148 Jan 28 22:29 .DS_Store
drwxr-xr-x 8 root root  4096 Jan 28 22:29 .git/
-rw-r--r-- 1 root root  1799 Jan 28 22:29 .gitignore
-rw-r--r-- 1 root root 19407 Jan 28 22:29 LICENSE
-rw-r--r-- 1 root root  2991 Jan 28 22:29 README.md
-rw-r--r-- 1 root root 27414 Jan 28 22:29 StarGAN_v2.py
drwxr-xr-x 2 root root  4096 Jan 28 22:29 assets/
drwxr-xr-x 4 root root  4096 Jan 28 22:29 dataset/
-rw-r--r-- 1 root root  4998 Jan 28 22:29 main.py
-rw-r--r-- 1 root root  7848 Jan 28 22:29 networks.py
-rw-r--r-- 1 root root 15500 Jan 28 22:29 ops.py
-rw-r--r-- 1 root root  7521 Jan 28 22:29 utils.py

 

 

 

7. 학습과정

$ python main.py --dataset afhq --phase train