top | item 23039240

(no title)

walrus | 5 years ago

The following instructions are for Linux or macOS. It may work on Windows too, but I'm not very experienced with Windows. No special hardware is required (you can run it on a CPU; no GPU needed).

Install pyenv[0] and pyenv-virtualenv[1]. Clone the repo and set up an environment:

  git clone https://github.com/taki0112/UGATIT
  cd UGATIT
  pyenv install 3.6.10  # [2]
  pyenv virtualenv 3.6.10 UGATIT
  pyenv local UGATIT
  pyenv activate
  pip install opencv-python==4.2.0.34 tensorflow==1.14.0

Download the pretrained weights from https://github.com/taki0112/UGATIT/issues/50#issuecomment-53.... Extract them:

  tar xf ugatit100.tar.xz
  mkdir checkpoint
  mv UGATIT_selfie2anime* checkpoint
  mkdir -p dataset/selfie2anime/{train,test}{A,B}

Crop your images in a 1:1 aspect ratio so that they contain only the head. Place them in the dataset/selfie2anime/testA/ directory. Run the program:

  python main.py --dataset selfie2anime --phase test

Open results/*/index.html in your browser to see the results.

[0] https://github.com/pyenv/pyenv#installation

[1] https://github.com/pyenv/pyenv-virtualenv#installation

[2] Other versions may work, but this is the version mentioned in https://github.com/taki0112/UGATIT#requirements

discuss

bentcorner|5 years ago

Thank you so much! I'll give this a crack on windows and see how it turns out.

bentcorner|5 years ago

I got it working via WSL on windows. Thanks!