├── .gitignore ├── AUTHORS ├── CONTRIBUTING ├── LICENSE ├── README.md ├── benchmark ├── .gitignore ├── Makefile ├── assets │ ├── gpyrnn.frag │ ├── std.frag │ └── std.vert ├── include │ ├── processor.h │ ├── renderer.h │ ├── timer.h │ └── utils.h └── src │ ├── main.cc │ ├── processor.cc │ ├── renderer.cc │ └── utils.cc ├── hdrnet ├── .gitignore ├── Makefile ├── __init__.py ├── bin │ ├── freeze_graph.py │ ├── run.py │ ├── scripts │ │ ├── optimize_graph.sh │ │ └── optimize_ms_graph.sh │ ├── train.py │ └── viz_activations.py ├── data_pipeline.py ├── hdrnet_ops.py ├── image.py ├── layers.py ├── metadata.py ├── metrics.py ├── models.py ├── ops │ ├── bilateral_slice.cc │ ├── bilateral_slice.cu.cc │ └── cuda │ │ └── cuda_config.h ├── requirements.txt ├── test │ ├── __init__.py │ └── ops_test.py └── utils.py ├── pretrained_models └── download.py ├── sample_data ├── identity │ ├── filelist.txt │ ├── input │ │ └── input.png │ └── output │ │ └── input.png └── input.png └── scripts ├── benchmark.sh ├── experts ├── a_nn.sh ├── b_nn.sh ├── c_highvar50_nn.sh ├── c_nn.sh ├── c_random250_nn.sh ├── d_nn.sh └── e_nn.sh ├── extra_figures └── compare_to_unet.py ├── faces └── train_nn.sh ├── hdrp └── train_nn.sh ├── ll ├── train_3x3nn_guide.sh ├── train_dilated.sh ├── train_fpyrnn.sh ├── train_gpyr.sh ├── train_gpyrnn.sh ├── train_gpyrnn_cm4.sh ├── train_nn_1024.sh ├── train_nn_guide.sh ├── train_stack.sh ├── train_std.sh └── train_unet.sh ├── ll_strong ├── train_fpyr_simple_guide.sh ├── train_fpyrnn.sh ├── train_fpyrnn2.sh ├── train_fpyrnn3.sh ├── train_fpyrnn3_cm1.sh ├── train_fpyrnn3_cm2.sh ├── train_fullres_features.sh ├── train_fullres_features_ms.sh ├── train_fullres_features_w_guide.sh ├── train_gpyrnn.sh ├── train_gpyrnn_cm4.sh ├── train_nn.sh └── train_std.sh ├── ps ├── early_bird_nn.sh ├── eboye_nn.sh ├── infrared_false_nn.sh ├── infrared_nn.sh ├── instagram1_nn.sh └── lomofi_nn.sh ├── st ├── nst_nn.sh ├── nst_std.sh └── st_nn.sh ├── upgrade.py ├── upgrade.sh └── usm ├── train_fpyrnn.sh ├── train_gpyr.sh ├── train_gpyrnn.sh ├── train_gpyrnn_cm4.sh └── train_std.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/.gitignore -------------------------------------------------------------------------------- /AUTHORS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/AUTHORS -------------------------------------------------------------------------------- /CONTRIBUTING: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/CONTRIBUTING -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/README.md -------------------------------------------------------------------------------- /benchmark/.gitignore: -------------------------------------------------------------------------------- 1 | bin/** 2 | -------------------------------------------------------------------------------- /benchmark/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/Makefile -------------------------------------------------------------------------------- /benchmark/assets/gpyrnn.frag: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/assets/gpyrnn.frag -------------------------------------------------------------------------------- /benchmark/assets/std.frag: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/assets/std.frag -------------------------------------------------------------------------------- /benchmark/assets/std.vert: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/assets/std.vert -------------------------------------------------------------------------------- /benchmark/include/processor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/include/processor.h -------------------------------------------------------------------------------- /benchmark/include/renderer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/include/renderer.h -------------------------------------------------------------------------------- /benchmark/include/timer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/include/timer.h -------------------------------------------------------------------------------- /benchmark/include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/include/utils.h -------------------------------------------------------------------------------- /benchmark/src/main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/src/main.cc -------------------------------------------------------------------------------- /benchmark/src/processor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/src/processor.cc -------------------------------------------------------------------------------- /benchmark/src/renderer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/src/renderer.cc -------------------------------------------------------------------------------- /benchmark/src/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/benchmark/src/utils.cc -------------------------------------------------------------------------------- /hdrnet/.gitignore: -------------------------------------------------------------------------------- 1 | lib 2 | build 3 | -------------------------------------------------------------------------------- /hdrnet/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/Makefile -------------------------------------------------------------------------------- /hdrnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/__init__.py -------------------------------------------------------------------------------- /hdrnet/bin/freeze_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/bin/freeze_graph.py -------------------------------------------------------------------------------- /hdrnet/bin/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/bin/run.py -------------------------------------------------------------------------------- /hdrnet/bin/scripts/optimize_graph.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/bin/scripts/optimize_graph.sh -------------------------------------------------------------------------------- /hdrnet/bin/scripts/optimize_ms_graph.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/bin/scripts/optimize_ms_graph.sh -------------------------------------------------------------------------------- /hdrnet/bin/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/bin/train.py -------------------------------------------------------------------------------- /hdrnet/bin/viz_activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/bin/viz_activations.py -------------------------------------------------------------------------------- /hdrnet/data_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/data_pipeline.py -------------------------------------------------------------------------------- /hdrnet/hdrnet_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/hdrnet_ops.py -------------------------------------------------------------------------------- /hdrnet/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/image.py -------------------------------------------------------------------------------- /hdrnet/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/layers.py -------------------------------------------------------------------------------- /hdrnet/metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/metadata.py -------------------------------------------------------------------------------- /hdrnet/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/metrics.py -------------------------------------------------------------------------------- /hdrnet/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/models.py -------------------------------------------------------------------------------- /hdrnet/ops/bilateral_slice.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/ops/bilateral_slice.cc -------------------------------------------------------------------------------- /hdrnet/ops/bilateral_slice.cu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/ops/bilateral_slice.cu.cc -------------------------------------------------------------------------------- /hdrnet/ops/cuda/cuda_config.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/ops/cuda/cuda_config.h -------------------------------------------------------------------------------- /hdrnet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/requirements.txt -------------------------------------------------------------------------------- /hdrnet/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/test/__init__.py -------------------------------------------------------------------------------- /hdrnet/test/ops_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/test/ops_test.py -------------------------------------------------------------------------------- /hdrnet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/hdrnet/utils.py -------------------------------------------------------------------------------- /pretrained_models/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/pretrained_models/download.py -------------------------------------------------------------------------------- /sample_data/identity/filelist.txt: -------------------------------------------------------------------------------- 1 | input.png 2 | -------------------------------------------------------------------------------- /sample_data/identity/input/input.png: -------------------------------------------------------------------------------- 1 | ../../input.png -------------------------------------------------------------------------------- /sample_data/identity/output/input.png: -------------------------------------------------------------------------------- 1 | ../../input.png -------------------------------------------------------------------------------- /sample_data/input.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/sample_data/input.png -------------------------------------------------------------------------------- /scripts/benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/benchmark.sh -------------------------------------------------------------------------------- /scripts/experts/a_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/experts/a_nn.sh -------------------------------------------------------------------------------- /scripts/experts/b_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/experts/b_nn.sh -------------------------------------------------------------------------------- /scripts/experts/c_highvar50_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/experts/c_highvar50_nn.sh -------------------------------------------------------------------------------- /scripts/experts/c_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/experts/c_nn.sh -------------------------------------------------------------------------------- /scripts/experts/c_random250_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/experts/c_random250_nn.sh -------------------------------------------------------------------------------- /scripts/experts/d_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/experts/d_nn.sh -------------------------------------------------------------------------------- /scripts/experts/e_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/experts/e_nn.sh -------------------------------------------------------------------------------- /scripts/extra_figures/compare_to_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/extra_figures/compare_to_unet.py -------------------------------------------------------------------------------- /scripts/faces/train_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/faces/train_nn.sh -------------------------------------------------------------------------------- /scripts/hdrp/train_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/hdrp/train_nn.sh -------------------------------------------------------------------------------- /scripts/ll/train_3x3nn_guide.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_3x3nn_guide.sh -------------------------------------------------------------------------------- /scripts/ll/train_dilated.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_dilated.sh -------------------------------------------------------------------------------- /scripts/ll/train_fpyrnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_fpyrnn.sh -------------------------------------------------------------------------------- /scripts/ll/train_gpyr.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_gpyr.sh -------------------------------------------------------------------------------- /scripts/ll/train_gpyrnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_gpyrnn.sh -------------------------------------------------------------------------------- /scripts/ll/train_gpyrnn_cm4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_gpyrnn_cm4.sh -------------------------------------------------------------------------------- /scripts/ll/train_nn_1024.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_nn_1024.sh -------------------------------------------------------------------------------- /scripts/ll/train_nn_guide.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_nn_guide.sh -------------------------------------------------------------------------------- /scripts/ll/train_stack.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_stack.sh -------------------------------------------------------------------------------- /scripts/ll/train_std.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_std.sh -------------------------------------------------------------------------------- /scripts/ll/train_unet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll/train_unet.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_fpyr_simple_guide.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_fpyr_simple_guide.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_fpyrnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_fpyrnn.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_fpyrnn2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_fpyrnn2.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_fpyrnn3.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_fpyrnn3.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_fpyrnn3_cm1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_fpyrnn3_cm1.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_fpyrnn3_cm2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_fpyrnn3_cm2.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_fullres_features.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_fullres_features.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_fullres_features_ms.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_fullres_features_ms.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_fullres_features_w_guide.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_fullres_features_w_guide.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_gpyrnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_gpyrnn.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_gpyrnn_cm4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_gpyrnn_cm4.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_nn.sh -------------------------------------------------------------------------------- /scripts/ll_strong/train_std.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ll_strong/train_std.sh -------------------------------------------------------------------------------- /scripts/ps/early_bird_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ps/early_bird_nn.sh -------------------------------------------------------------------------------- /scripts/ps/eboye_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ps/eboye_nn.sh -------------------------------------------------------------------------------- /scripts/ps/infrared_false_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ps/infrared_false_nn.sh -------------------------------------------------------------------------------- /scripts/ps/infrared_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ps/infrared_nn.sh -------------------------------------------------------------------------------- /scripts/ps/instagram1_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ps/instagram1_nn.sh -------------------------------------------------------------------------------- /scripts/ps/lomofi_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/ps/lomofi_nn.sh -------------------------------------------------------------------------------- /scripts/st/nst_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/st/nst_nn.sh -------------------------------------------------------------------------------- /scripts/st/nst_std.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/st/nst_std.sh -------------------------------------------------------------------------------- /scripts/st/st_nn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/st/st_nn.sh -------------------------------------------------------------------------------- /scripts/upgrade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/upgrade.py -------------------------------------------------------------------------------- /scripts/upgrade.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/upgrade.sh -------------------------------------------------------------------------------- /scripts/usm/train_fpyrnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/usm/train_fpyrnn.sh -------------------------------------------------------------------------------- /scripts/usm/train_gpyr.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/usm/train_gpyr.sh -------------------------------------------------------------------------------- /scripts/usm/train_gpyrnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/usm/train_gpyrnn.sh -------------------------------------------------------------------------------- /scripts/usm/train_gpyrnn_cm4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/usm/train_gpyrnn_cm4.sh -------------------------------------------------------------------------------- /scripts/usm/train_std.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mgharbi/hdrnet/HEAD/scripts/usm/train_std.sh --------------------------------------------------------------------------------