This is a modified version of CycleGAN for the paper "CycleGAN Face-off" by Xiaohan Jin, Ye Qi and Shangxuan Wu.
Paper. Note that this is not the repo for the video in here.
This repo contains the code which adds:
- SSIM loss. Usage: adding
--with_ssimin your training script. - Better testing script for generating a whole video. Usage:
python generate_fake_sequence.py - Training scripts for some of the experiments like Shangxuan <-> Russ etc.
- For weighted loss of facial mask, please use this repo.
- For double_d setting, please use this_repo.
- Adding make_video folder for making a comparison video (such as make_video/jin/output_with_sould.mkv). Usage:
python jin.py. The result video is here. - Adding draw_plot folder for analysing plots for the log files. Usage:
python draw_plot.py. The result plot is here.
