Joint Bilateral Learning

This repository is an unofficial implementation in PyTorch for the paper: "Joint Bilateral Learning for Real-time Universal Photorealistic Style Transfer" (ECCV2020)

[paper]

Dependencies

• Python 3.7.2
• PyTorch 1.2
• CUDA10.0 and cuDNN

Train

$ python main.py --cont_img_path <path/to/cont_img> --style_img_path <path/to/style_img> --batch_size 8

Test

$ python test.py --cont_img_path <path/to/single_cont_img> --style_img_path <path/to/single_style_img> --model_checkpoint <path/to/model checkpoint>