/Joint-Bilateral-Learning

An unofficial implementation of Joint Bilateral Learning for Real-time Universal photorealistic Style Transfer

Primary LanguagePythonMIT LicenseMIT

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>