ideaplexus / caingan Goto Github PK

Pose Manipulation with Identity Preservation. Andrei-Timotei Ardelean, Lucian Mircea Sasu in International Journal of Computers Communications & Control

Official PyTorch implementation of CainGAN, used to perform few-shot image generation. Starting from one or more source pictures, the model can synthesize face images in novel poses while preserving the person's identity.

Copyright (C) 2019 Andrei-Timotei Ardelean, Lucian Mircea Sasu

All rights reserved. Licensed under the CC BY-NC-SA 4.0 (Attribution-NonCommercial-ShareAlike 4.0 International)

A ready to use conda environment is provided. To create the python environment run:

conda env create -f environment.yaml
conda activate fewshot

To train the model you can use the train script.

python train.py --K 8 --cuda_visible 0 --data_root vox2selection/mp4 

This will train an 8-shot model on GPU 0. Change data_root path to the directory of your dataset. The format of the directory is expected to be similar to the one used in VoxCeleb2 dataset.

Special thanks go to Xperi Corporation that provided the environment and physical resources that made this work possible.

Should you find this work useful for your research, please cite:

@article{IJCCC3862,
	author = {Andrei-Timotei Ardelean and Lucian Sasu},
	title = {Pose Manipulation with Identity Preservation},
	journal = {International Journal of Computers Communications & Control},
	volume = {15},
	number = {2},
	year = {2020},
	issn = {1841-9844},
	doi = {10.15837/ijccc.2020.2.3862},
	url = {http://univagora.ro/jour/index.php/ijccc/article/view/3862}
}