atomeyang / DPR

Code for Deep Single-image Portrait Image Relighting

Deep Single-Image Portrait Relighting [Project Page]

Hao Zhou, Sunil Hadap, Kalyan Sunkavalli, David W. Jacobs. In ICCV, 2019

Overview

• Test script for 512x512 images: testNetwork_demo_512.py
• Test script for 1024x1024 images: testNetwork_demo_1024.py

Dependencies

pytorch >= 1.0.0

opencv >= 4.0.0

shtools: https://shtools.oca.eu/shtools/ (optional)

Notes

We include an example image and seven example lightings in data. Note that different methods may have different coordinate system for Spherical Harmonics (SH), you may need to change the coordiante system if you use SH lighting from other sources. The coordinate system of our method is in accordance with shtools, we provide a function utils_normal.py in utils to help you tansfer the coordinate system from bip2017 and sfsNet to our coordinate system. To use utils_normal.py you need to install shtools. The code is for research purpose only.

Data Preparation

We publish the code for data preparation, please find it in (https://github.com/zhhoper/RI_render_DPR).

Citation

If you use this code for your research, please consider citing:

@InProceedings{DPR,
  title={Deep Single Portrait Image Relighting},
  author = {Hao Zhou and Sunil Hadap and Kalyan Sunkavalli and David W. Jacobs},
  booktitle={International Conference on Computer Vision (ICCV)},
  year={2019}
}