AIAIAIAI

AI project) AnimeGAN - Code 관련

summerorange 2022. 5. 2. 19:26
반응형

이전에 animeGAN에 관심이 있어서 여러 번 돌려봤던 적이 있었다. 그 때 열심히 구글링하다가 받은 코드인데

까먹을 것 같아서 여기에 끄적끄적. 사용환경은 코랩이었고, 이 코드를 참고해서 돌렸었다. 지금은 사실 기억이 가물가물

!git clone https://github.com/TachibanaYoshino/AnimeGAN


import os
os.chdir('AnimeGAN')
print(os.getcwd())


download_staffs.sh

URL=https://github.com/TachibanaYoshino/AnimeGAN/releases/download/Haoyao-style_V1.0/Haoyao-style.zip
ZIP_FILE=./checkpoint/Haoyao-style.zip
TARGET_DIR=./checkpoint/saved_model

mkdir -p ./checkpoint
wget -N $URL -O $ZIP_FILE
mkdir -p $TARGET_DIR
unzip $ZIP_FILE -d $TARGET_DIR
rm $ZIP_FILE

DatesetURL=https://github.com/TachibanaYoshino/AnimeGAN/releases/download/dataset-1/dataset.zip
ZIP_FILE=./dataset.zip
TARGET_DIR=./dataset

rm -rf dataset
wget -N $DatesetURL -O $ZIP_FILE
unzip $ZIP_FILE -d $TARGET_DIR
rm $ZIP_FILE

VGG_FILE=./vgg19_weight/vgg19.npy
wget --load-cookies /tmp/cookies.txt "https://docs.google.com/uc?export=download&confirm=$(wget --quiet --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=1U5HCRpZWAbDVLipNoF8t0ZHpwCRX7kdF' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/\1\n/p')&id=1U5HCRpZWAbDVLipNoF8t0ZHpwCRX7kdF" -O $VGG_FILE && rm -rf /tmp/cookies.txt


!bash download_staffs.sh


!nvidia-smi


!python main.py --phase train --dataset Hayao --epoch 1 --init_epoch 1


!pip install tensorflow==1.15


!python test.py --checkpoint_dir checkpoint/saved_model --test_dir dataset/test/real --style_name H

위에랑 아래랑 같은데 돌리다보니 필요한 코드가 있어서 잘 믹씽해서 만들었었다.

%cd /content/drive/MyDrive

!git clone https://github.com/TachibanaYoshino/AnimeGAN
 
import os
os.chdir('AnimeGAN')
print(os.getcwd())
 
# ------------------------------------------------
 
download_staffs.sh
 
URL=https://github.com/TachibanaYoshino/AnimeGAN/releases/download/Haoyao-style_v1.0/Haoyao-style.zip
ZIP_FILE=./checkpoint/Haoyao-style.zip
TARGET_DIR=./checkpoint/saved_model
 
mkdir -p ./checkpoint
wget -N $URL -O $ZIP_FILE
mkdir -p $TARGET_DIR
unzip $ZIP_FILE -d $TARGET_DIR
rm $ZIP_FILE
 

 
# ------------------------------------------------
 
!bash download_staffs.sh
 
!nvidia-smi
 
!pip install tensorflow==1.15
 
!python test.py --checkpoint_dir checkpoint/saved_model --test_dir dataset/test/real --style_name H

돌아갈 때도 있고 안 돌아갈 때도 있어서 좀...  고생했었는데. 이 건 나중에 포스팅하기루.

하단의 왼쪽이 산책하다 찍은 원본 사진이고 오른쪽이 GAN 돌렸을 때. 미야자키 하야오의 그림을 학습한 걸 사진에 자동으로 적용시킨다.

미야자키 하야오 풍으로 만들어준다.

참고로 맥북에서 돌렸다. 그런데 사용환경이 어쨌거나 어차피 난 코랩을 쓰니 상관없지만. 코랩 pro GPU 썼었다. 

반응형