Skip to content

kangkanbharadwaj/High-Dynamic-Range-imaging-using-CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Learning HDR Imaging from Synthetic data

Objective

  1. Creation of a static dataset using blender
  2. Learn a model on the data
  3. Use the model to test real scenes

Introduction

  • LDR: Images of 0-255 pixel range, can be displayed on a standard device.
  • HDR: Range of luminance that is equivalent to the one experienced by a human eye (upto 105).
  • Tone-mapping: The scene with original irradiance(linear space) is mapped to pixel values of (0-255).

image segmentation vs semantic segmentation

Motivation

  • Image from DSLR camera cannot capture much details in very dark or very bright regions in one exposure setting.
  • Surveillance or medical applications

image segmentation vs semantic segmentation


image segmentation vs semantic segmentation



Approach

  • Data set generation
  • Learn a network model from the data set
  • Run Experiements on the learned model

Dataset generation

  • Created using Blender seggregating them into 331 train and 75 test images.
  • One scene was rendered multiple times in different exposures. The exposures ranged from EV:-5 to EV: +4.
  • Render engines are Cycles render and Blender Render.
  • The image resolution was kept to 640 X 480.

image segmentation vs semantic segmentation


Network Architecture

UNet

  • 19 Convolution layers, 4 Pooling layers and up-convolution
  • 3X3 kernel size, padding of 1, mini batch size 4
  • Initial LR of 0.0001, gamma 0.1 and step size 20000
  • ReLU activation, Adam Solver

image segmentation vs semantic segmentation


Experiments and Results

  • The experiments were performed on 331 training images and 75 test images.
  • Visualization after tone mapping using Gamma Correction

Basic experiments


image segmentation vs semantic segmentation


image segmentation vs semantic segmentation

image segmentation vs semantic segmentation


Experiments to fight disadvantage of smaller dataset


image segmentation vs semantic segmentation


image segmentation vs semantic segmentation

image segmentation vs semantic segmentation


  • These experiments were conducted to overcome the problem of having a smaller dataset
  • For this we incorporated dropout variations and model reduction

image segmentation vs semantic segmentation


image segmentation vs semantic segmentation

image segmentation vs semantic segmentation


Choose the best for real images

  • The HDR Candidates used are obtained by applying an inverse to the camera response function that maps the Low Dynamic Range Images to their corresponding HDR counterparts.

image segmentation vs semantic segmentation


image segmentation vs semantic segmentation

image segmentation vs semantic segmentation


HDR Candidates

  • The HDR Candidates used are obtained by applying an inverse the camera response function that maps the Low Dynamic Range Images to their corresponding HDR counterparts.

image segmentation vs semantic segmentation


image segmentation vs semantic segmentation

image segmentation vs semantic segmentation


Experiments on random set of LDRs

  • Experiments were performed on the same set of train and test data but we shuffled the stack of LDR images.

image segmentation vs semantic segmentation


image segmentation vs semantic segmentation


Network qualification on real time images


image segmentation vs semantic segmentation image segmentation vs semantic segmentation

image segmentation vs semantic segmentation image segmentation vs semantic segmentation

image segmentation vs semantic segmentation image segmentation vs semantic segmentation


Don’t use our network for a motion dataset


image segmentation vs semantic segmentation


Compare UNet to state-of-the-art


image segmentation vs semantic segmentation



image segmentation vs semantic segmentation image segmentation vs semantic segmentation


Conclusions

  • Successfully created a sythetic dataset that can be used for further experimental and research purposes.
  • Our network was able to justify on real time images being trained on synthetic data.
  • Choice of the network should be taken into consideration as for a smaller dataset a simpler network is more preferable.
  • Comparing to architecture from Nima Khademi Kalantari using only an encoder (convolution without de-convolution) can also be an option.

Future Work

  • Real world contains motions
  • A real world dataset will contain some motion due to objects moving, hand shaking etc.
  • Solving HDR after LDR alignment
  • Create a data set that involves motion simulating real time situations.
  • Estimating the flow using a FlowNet
  • Train the best network from previous findings
  • Test generalization on real images with motion

About

Predict HDR images from LDR images using CNN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published