/TGIM

Text-Guided Image Manipulation

Primary LanguagePython

Text Guided Image Manipulation

This project is under working: TaGAN + Transformer + auxiliary Loss

Setup and Dependencies

This code is implemented using PyTorch v1.6.0, and provides out of the box support with CUDA 10.1

Anaconda / Miniconda is the recommended to set up this codebase.

Anaconda or Miniconda

Clone this repository and create an environment:

git clone https://www.github.com/ecpark4545/TGIM
conda create -n TGIM python=3.7

# activate the environment and install all dependencies
conda activate TGIM
cd TGIM

# https://pytorch.org
pip install torch==1.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
pip install -r requirements.txt

And set data_root and save_path in cfg/birds.yml

Preparing Data

We provide preprocessed dataset birds.

Training

you can simply run the model with this code

python main.py --gpu 0 --batch_size 32
  • This code borrows heavily from TaGAN repository. Many thanks.