Durgesh93 / AHDR-Net

Re-implementation of CVPR 2019 Paper: Attention-guided Network for Ghost-free High Dynamic Range Imaging by Qingsen Yan et. al.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AHDR-Net-Tensorflow

This repository is the tensorflow implementation for the HDR deghosting, AHDR-Net, proposed in the paper - Attention-guided Network for Ghost-free High Dynamic Range Imaging.

Requirements

system requirements: Python:3.5.2
numpy==1.16.4
tensorboard==1.13.1
tensorflow-estimator==1.13.0
tensorflow-gpu==1.13.1
opencv-python==4.1.0.25

Training

The training script for this model is fusion_train_256_ahdr.py.The model was trained on UCSD dataset for 50 epochs with batch size of 8.

Testing

`python ahdr_test.py

Test Outputs

Test outputs consist of predicted hdr images and tonemapped its version.

Contact

For further queries, please mail at durgesh080793 <at> gmail <dot> com.

About

Re-implementation of CVPR 2019 Paper: Attention-guided Network for Ghost-free High Dynamic Range Imaging by Qingsen Yan et. al.

License:MIT License


Languages

Language:Python 100.0%