├── .gitignore ├── LICENSE ├── README.md ├── ckpts └── .gitkeep ├── data ├── labels.csv ├── test_hdr.csv ├── train.csv └── val.csv ├── environment.yml ├── images └── teaser.png ├── infer_bracketed.py ├── logs └── .gitkeep ├── results └── .gitkeep ├── src ├── architectures │ ├── __init__.py │ ├── base_unet.py │ └── modules │ │ ├── __init__.py │ │ ├── act_norm.py │ │ ├── arch_utils.py │ │ ├── dec_blk.py │ │ ├── enc_blk.py │ │ ├── spade.py │ │ └── upsampler.py ├── datamodules │ ├── __init__.py │ ├── datamodule.py │ └── dataset.py ├── models │ ├── __init__.py │ ├── ednet.py │ └── losses.py └── utils │ ├── __init__.py │ ├── photomatix-presets │ ├── balance.xmp │ ├── enhance.xmp │ ├── photomatix-preset.xmp │ ├── smooth.xmp │ └── soft.xmp │ └── tonemap_img.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/README.md -------------------------------------------------------------------------------- /ckpts/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/labels.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/data/labels.csv -------------------------------------------------------------------------------- /data/test_hdr.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/data/test_hdr.csv -------------------------------------------------------------------------------- /data/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/data/train.csv -------------------------------------------------------------------------------- /data/val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/data/val.csv -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/environment.yml -------------------------------------------------------------------------------- /images/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/images/teaser.png -------------------------------------------------------------------------------- /infer_bracketed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/infer_bracketed.py -------------------------------------------------------------------------------- /logs/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /results/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/architectures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/architectures/base_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/architectures/base_unet.py -------------------------------------------------------------------------------- /src/architectures/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/architectures/modules/__init__.py -------------------------------------------------------------------------------- /src/architectures/modules/act_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/architectures/modules/act_norm.py -------------------------------------------------------------------------------- /src/architectures/modules/arch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/architectures/modules/arch_utils.py -------------------------------------------------------------------------------- /src/architectures/modules/dec_blk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/architectures/modules/dec_blk.py -------------------------------------------------------------------------------- /src/architectures/modules/enc_blk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/architectures/modules/enc_blk.py -------------------------------------------------------------------------------- /src/architectures/modules/spade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/architectures/modules/spade.py -------------------------------------------------------------------------------- /src/architectures/modules/upsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/architectures/modules/upsampler.py -------------------------------------------------------------------------------- /src/datamodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datamodules/datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/datamodules/datamodule.py -------------------------------------------------------------------------------- /src/datamodules/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/datamodules/dataset.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/ednet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/models/ednet.py -------------------------------------------------------------------------------- /src/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/models/losses.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/photomatix-presets/balance.xmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/utils/photomatix-presets/balance.xmp -------------------------------------------------------------------------------- /src/utils/photomatix-presets/enhance.xmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/utils/photomatix-presets/enhance.xmp -------------------------------------------------------------------------------- /src/utils/photomatix-presets/photomatix-preset.xmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/utils/photomatix-presets/photomatix-preset.xmp -------------------------------------------------------------------------------- /src/utils/photomatix-presets/smooth.xmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/utils/photomatix-presets/smooth.xmp -------------------------------------------------------------------------------- /src/utils/photomatix-presets/soft.xmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/utils/photomatix-presets/soft.xmp -------------------------------------------------------------------------------- /src/utils/tonemap_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/src/utils/tonemap_img.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/single_image_hdr/HEAD/train.py --------------------------------------------------------------------------------