Awesome
ECCV'18: Deep High Dynamic Range Imaging with Large Foreground Motions
<img src="./img/hdr_teaser.jpg" width="800">This is the implementation for Deep High Dynamic Range Imaging with Large Foreground Motions, Shangzhe Wu, Jiarui Xu, Yu-Wing Tai, Chi-Keung Tang, in ECCV, 2018. More results can be found on our project page.
Get Started
Prerequisites
- Python 3.5
- Tensorflow 1.4.0
- OpenCV 3.4 (from conda-forge)
- Photomatix for tonemapping
Setup
- Clone this repo:
git clone https://github.com/elliottwu/DeepHDR.git
cd DeepHDR
- Download pretrained model: (~60MB)
sh download_pretrained.sh
Demo
sh test.sh
Tonemapping (post-processing)
Generated HDR images are in .hdr
format, which may not be properly displayed in your image viewer directly. You may use Photomatix for tonemapping:
- Download Photomatix free trial, which won't expire.
- Load the generated
.hdr
file in Photomatix. - Adjust the parameter settings. You may refer to pre-defined styles, such as
Detailed
andPainterly2
. - Save your final image in
.tif
or.jpg
.
Train
- Download Kalantari's dataset: (~8GB)
cd dataset
sh download_dataset.sh
cd ..
- Prepare TFRecord: (this takes ~10 minutes)
cd dataset
python convert_to_tfrecord.py
cd ..
- Start training:
sh train.sh
- To monitor training using Tensorboard, copy the following to your terminal and open
localhost:8888
in your browser
tensorboard --logdir=logs --port=8888
Citation
@InProceedings{Wu_2018_ECCV,
author = {Wu, Shangzhe and Xu, Jiarui and Tai, Yu-Wing and Tang, Chi-Keung},
title = {Deep High Dynamic Range Imaging with Large Foreground Motions},
booktitle = {The European Conference on Computer Vision (ECCV)},
month = {September},
year = {2018}
}