Giter VIP home page Giter VIP logo

image-dehazing-using-gman-net's Introduction

Image-Dehazing-using-GMAN-net

License PRs Welcome PWC

Introduction

Generic Model-Agnostic Convolutional Neural Network(GMAN) is a convolutional neural network proposed for haze removal and clear image restoration. It is an end-to-end deep learning system that employs the encoder-decoder network for denoising image. I've used Kaggle notebook for the purpose of implementation and training. Dataset used for training and validation is SOTS outdoor available here.

Detailed explanation and documentation here. Modified model along with web app and deployment code will be there soon.

Note: Incase notebook is not loading on GitHub, you can check notebook with validation output upto 10 epochs here.

Requirements

  • Python(3.6+)
  • Tensorflow(2+)
  • GPU: Nvidia Tesla P100(provided by Kaggle)

How to use on your images

  1. Download the saved model.
  2. pip install -r requirements.txt
  3. Give model path and image path to test.py and run.
    (Note: Saved model folder, test.py and images should be in same folder.)

Evaluation Results

I've used naturally hazed images downloaded randomly from google and some images are from dataset. You can see the dehazed test images against hazy images here, some of them are below. Dehazed test images with good resolution are available here.

test_104
test_104
test_104
test_104

Citation

@article{liu2019single,
  title={Single Image Dehazing with a Generic Model-Agnostic Convolutional Neural Network},
  author={Liu, Zheng and Xiao, Botao and Alrabeiah, Muhammad and Wang, Keyan and Chen, Jun},
  journal={IEEE Signal Processing Letters},
  volume={26},
  number={6},
  pages={833--837},
  year={2019},
  publisher={IEEE}
}

https://github.com/Seanforfun/GMAN_Net_Haze_Removal

image-dehazing-using-gman-net's People

Contributors

sanchitvj avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar

image-dehazing-using-gman-net's Issues

Only 412*548 dimension image expected during evaluation

Hey!
I hope you are well.
There's one issue that i face.
When using the model for evaluation, it doesn't accept any image other then an image of dimensions 412548, cause that's the input shape defined in the model.
So i have to change the evaluation function to have it resize to 412
548 (which distorts the images), while you yourself are using an 1920*1080 image resize, which did worked for you.
Can you guide me anyhow on how to resolve this.
Thankyou :)

Screenshot from 2021-05-18 07-56-05

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.