Code Monkey home page Code Monkey logo

image_enhancement_techniques's Introduction

Image Enhancement Techniques


Overview

This work is about removing noise/haze from noisy/hazy images to obtain clearer images using stacked autoencoders.

Highlights

The work is a result of participation in (New Trends in Image Restoration and Enhancement workshop) NTIRE 2020 Real Image Denoising Challenge - Track2 โ€“ sRGB, in conjunction with (Conference on Computer Vision and Pattern Recognition) CVPR 2020, Seattle, US.

Participation in NTIRE 2020 Denoising Challenge led my team secure 16th position among 250+ registrants. Publication Link

This work also discusses implementation on Image Dehazing as an extension to the original work on Image Denoising. Our results from the Pix2Pix model position among the Top-16 out of 270+ registrants for the NTIRE 2019 Image Dehazing Challenge. Report

Results

Further work was done post the competition. The summary of results is as follows:

Technique Model Epochs Mean PSNR Mean SSIM
Denoising Stacked Autoencoder 1000 26.95 0.78
Dehazing Stacked Autoencoder 500 10.24 0.54
Dehazing CycleGAN 1000 13.44 0.54
Dehazing Pix2Pix 1000 15.96 0.60

From the results, we can conclude that our basic PyTorch implementation has been able to perform quite well in general, and compared to two state-of-the-art approaches for Image Translation tasks - CycleGAN and Pix2Pix.

noisy hazy

Dependencies

  • PyTorch
  • OpenCV

Installation

  • To install PyTorch, see installation instructions on the PyTorch website.
  • To install OpenCV: pip install opencv-python

Dataset

Image Denoising

  • This work makes use of the Smartphone Image Denoising Dataset SIDD. The data was made available on the Codalab website where the competition was hosted.

Image Dehazing

  • This work makes use of Dense-Haze dataset for the purpose of Image Dehazing.

Data Preprocessing

The data in original forms are extremely high resolution images. Due to lack of computing resources, for training purposes, the images have been down-sized to 256 pixels.

For Image Denoising

  • Use the denoising_make_data.py script to process images and convert them to a numPy array (.npy) file.

For Image Dehazing

  • Use the dehazing_make_data.py script to process images and convert them to a numPy array (.npy) file.

References

image_enhancement_techniques's People

Contributors

rajatguptakgp avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

image_enhancement_techniques's Issues

help

Hello, I had some problems. In dehazing_autoencoder.ipynb,
mask = np.load('mask.npy',allow_pickle=True)
where can I find this file?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.