Giter Site home page Giter Site logo

zero-dce_extension's Introduction

Zero-DCE++

You can find more details here: https://li-chongyi.github.io/Proj_Zero-DCE++.html.

You can find the details of our CVPR version: https://li-chongyi.github.io/Proj_Zero-DCE.html.

✌If you use this code, please cite our paper. Please hit the star at the top-right corner. Thanks!

We also provided a MindSpore version of our code: https://pan.baidu.com/s/1kEjKtYYSwvzHCzDh_4niew (passwords: 37ne).

🌈We released a survey on deep learning-based low-light image enhancement------- Low-Light Image and Video Enhancement Using Deep Learning: A Survey, an online platform, a new dataset. Have fun! https://github.com/Li-Chongyi/Lighting-the-Darkness-in-the-Deep-Learning-Era-Open.

Pytorch

Pytorch implementation of Zero-DCE++

Requirements

  1. Python 3.7
  2. Pytorch 1.0.0
  3. opencv
  4. torchvision 0.2.1
  5. cuda 10.0

Zero-DCE++ does not need special configurations. Just basic environment.

Or you can create a conda environment to run our code like this: conda create --name zerodce++_env opencv pytorch==1.0.0 torchvision==0.2.1 cuda100 python=3.7 -c pytorch

Folder structure

Download the Zero-DCE++ first. The following shows the basic folder structure.


├── data
│   ├── test_data 
│   └── train_data 
├── lowlight_test.py # testing code
├── lowlight_train.py # training code
├── model.py # Zero-DEC++ network
├── dataloader.py
├── snapshots_Zero_DCE++
│   ├── Epoch99.pth #  A pre-trained snapshot (Epoch99.pth)

Test:

cd Zero-DCE++

python lowlight_test.py 

The script will process the images in the sub-folders of "test_data" folder and make a new folder "result" in the "data". You can find the enhanced images in the "result" folder.

Train:

cd Zero-DCE++

python lowlight_train.py 

License

The code is made available for academic research purpose only. Under Attribution-NonCommercial 4.0 International License.

Bibtex

@inproceedings{Zero-DCE++,
 author = {Li, Chongyi and Guo, Chunle Guo and Loy, Chen Change},
 title = {Learning to Enhance Low-Light Image via Zero-Reference Deep Curve Estimation},
 booktitle = {IEEE Transactions on Pattern Analysis and Machine Intelligence},
 pages    = {},
 month = {},
 year = {2021}
 doi={10.1109/TPAMI.2021.3063604}
}

(Full paper: https://ieeexplore.ieee.org/document/9369102 or arXiv version: https://arxiv.org/abs/2103.00860)

Contact

If you have any questions, please contact Chongyi Li at [email protected] or Chunle Guo at [email protected].

zero-dce_extension's People

Contributors

li-chongyi avatar

Stargazers

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

Watchers

 avatar  avatar  avatar

zero-dce_extension's Issues

train problems

When training, if it doesn't work, it's just that the terminal inputs Python lowerlight_ After train.py, it ended directly without any errors, but there was no training information. If the answer is provided, we would greatly appreciate it

about learning rate decay strategy

Hi,thank you for your excellent work, but I have a question that why there is no learning rate decay strategy in trianing process? Do you think it will help to improve enhancement?

[nan loss occur]

Hi, nan loss occured when training on some images. How to solve this?

Where to download DICM dataset?

Hello, when I evaluate the code, there is no file containing the dataset DICM、LIME、MEF infered in paper "Learning to Enhance Low-Light Image via Zero-Reference Deep Curve Estimation", can I ask for a help?

Why the Wtv is so big?

In the paper, it says that Wcol =0.5 and WtvA=20, while other loss's W=1. But in the code, WtvA is set = 1600, it is really strange...
image

[Question] abnormal results

Thx a lot for the great work!
Is there any way to solve the abnormal artifacts?
original picture:
t.png
results:
t.png
(sorry for my bad english)

License

Hey,

could you elaborate on the license? It says research only, but under MIT (which permits commercial use). Thanks for your great work!

Error while loading the model in macos cpu

After installing all the module in requirements, I get the following error when loading the model
_pickle.UnpicklingError: A load persistent id instruction was encountered
but no persistent_load function was specified.

I checked the forums for this error
https://blog.csdn.net/wavehaha/article/details/114900600
which suggests inconsistency pytorch version
But i am using pytorch=1.0.0 only

Has anybody encountered it? How to load the model on cpu ?

请教

非常感谢开源,对这方面很感兴趣,请问下你这边有做过高曝光的研究吗,想请教下是不是通用的

Images are saturated for adobe5k

Hi,

I trained ZDCE++ in adobe5k for 100 epochs. I had kept 4000 images for training, 500 for validation and 500 for testing. I used the original folder for training. I found the results to be saturated.
One more question is on why we are not checking for validation loss. If we monitor validation loss, can this be improved.
Any help will be appreciated.

Thank you

a0014-WP_CRW_6320

a0001-jmac_DSC1459

a0002-dgw_005

测试文件运行失败

作者您好!运行测试文件出现_pickle.UnpicklingError: A load persistent id instruction was encountered,but no persistent_load function was specified.不过运行环境按照您推荐的搭建的,请问这个是什么问题?

Why scale factor is 12 in testing but 1 in training?

Hello, authors. Thanks for your excellent work.
I notice that line 20 of lowlight_test.py defines the scale factor as 12, whereas line 105 of lowlight_train.py assumes that the default scale factor is 1.
I would like to know (1) the reason for this inconsistency (2) which one I should adopt for training the model.
Thanks!

exposure question

I have a question. Is the training data used by the author the same as the one released? I trained my own model, but it has overexposure issues, for example: areas that were originally bright in the image become very bright. However, the official model doesn't have this problem. I'd like to ask how to solve it?

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.