AHDRNet-PyTorch

This is the PyTorch implementation of Attention-guided Network for Ghost-free High Dynamic Range Imaging (CVPR 2019).
The official code is at AHDRNet. However, there exists some problems in training and testing within the official implementation which are not solved.

1. Environment

Python >= 3.7
PyTorch >= 1.4.0
opencv-python = 4.5.1
imageio = 2.13.3
matplotlib

2. Dataset

The training data and testing data is from Kalantari (ACM TOG 2017), the dataset can be downloade from Kalantari Dataset.

3. Quick Demo (Only for tiff format 48-depth images (same with the Kalantari Dataset) now, others in progress)

Clone this repository:

git clone https://github.com/ytZhang99/AHDRNet-PyTorch.git

Place the test image folders in ./data/Test/:

Test
└── test_data
    ├── Name_A
    |   ├── 1.tif
    |   ├── 2.tif
    |   ├── 3.tif
    |   ├── exposure.txt
    |   └── HDRImg.hdr (optional)
    └── Name_B

Run the following command to test:
```
python main.py --test_only
```
The output images are placed in ./results/0_epoch/

4. Training

Place the training image folders in ./data/Train/:

Train
└── train_data
    ├── Name_A
    |   ├── 1.tif
    |   ├── 2.tif
    |   ├── 3.tif
    |   ├── exposure.txt
    |   └── HDRImg.hdr
    └── Name_B

Modify the main.sh file and run the following command to train:
```
sh main.sh
```
Notice that the default setting of this program is implementing validation on the test dataset after training, you can modify main.sh to close the validation progress.

The trained model is saved in ./ckp/, then you can test your own model:

python main.py --test_only --model latest.pth
python main.py --test_only --model best_checkpoint.pth (This model is accessible with validation)