For anyone who is looking for answers!
Follow the below link!

  
    
    
    Normalisation necessary for image reconstruction? vision
  
  
    Thank you so so much!! This worked like a charm!!
  


Thank you! @Nikronic