This repo is based on pose-gan [https://github.com/AliaksandrSiarohin/pose-gan], which is an excellent work for person pose transfer.
- python2.7
- Numpy
- Scipy
- Skimage
- Pandas
- Tensorflow
- Keras
- keras-contrib
- tqdm
- pip install -r requirements.txt
- install keras-contrib by running: pip install git+https://www.github.com/keras-team/keras-contrib.git
In order to train a model:
-
Download VeRi dataset https://github.com/VehicleReId/VeRidataset. Put it in data folder. Rename this folder to data/VeRi-dataset. Rename image_train and image_test with test and train.
-
Download view-synthesis-for-vehicles.zip and unzip the contents to the main folder.
-
Training the full system by running the train.sh: bash train.sh
bash test.sh