Giter Club home page Giter Club logo

c3net's Introduction

C3Net

This is a PyTorch implementation of the New Trends in Image Restoration and Enhancement workshop and challenges on image and video restoration and enhancement (NTIRE 2020 with CVPR 2020) paper, C3Net: Demoireing Network Attentive in Channel, Color and Concatenation.

If you find our project useful in your research, please consider citing:

@InProceedings{Kim_2020_CVPR_Workshops,
author = {Kim, Sangmin and Nam, Hyungjoon and Kim, Jisu and Jeong, Jechang},
title = {C3Net: Demoireing Network Attentive in Channel, Color and Concatenation},
booktitle = {The IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops},
month = {June},
year = {2020}
}

Dependencies

Python 3.6.9
PyTorch 1.4.0

Data

Reference

You have to sign in Codalab and apply to NTIRE 2020 Demoireing Challenge before getting the data.

Proposed algorithm

C3Net (Track 1: Single Image)
AVC_Block
AttBlock
ResBlock
C3Net-Burst (Track 2: Burst)
AVC_Block-Burst

Training

Use the following command to use our training codes

python train.py

There are other options you can choose. Please refer to train.py.

Test

Use the following command to use our test codes

python test.py

There are other options you can choose. Please refer to test.py.

Performance (PSNR/SSIM)

To use heavier model, we also used numpy to read input data, not hdf5. Hyung-Joon and jisukim helped it.
Our best records can be derived in the code by changing h5 into numpy and reducing GPU memory.

Validation Server PSNR SSIM Rank
Track 1: Single Image 41.30 0.99 9th
Track 2: Burst 40.55 0.99 5th

Burst_Results_List

Testing Server Reference

Testing Server PSNR SSIM Rank
Track 1: Single Image 41.11 0.99 4th
Track 2: Burst 40.33 0.99 5th

Final_Results

Honorable_Mention_Award

Contact

If you have any question about Demoireing model and the CVPR2020 challenge paper, feel free to ask me to [email protected].
If you have any question about Deblurring model, visit here and feel free to ask Hyung-Joon to [email protected].
If you have any question about using more heavier C3Net, visit here and feel free to ask jisukim to [email protected].

Acknowledgement

Thanks for SaoYan who gave the implementaion of DnCNN.
Thanks for yun_yang who gave the implementation of DRRN.
Thanks for BumjunPark who gave the implementation of DHDN.

Hint of color loss from Jorge Pessoa.
Hint of concatenation and residual learning from RDN (informal implementation).
Hint of U-net block from DIDN (formal implementation).

C3Net started from RUN.

More Details

Also, we won 3rd Place in NTIRE 2020 Challenge on Image and Video Deblurring thanks to Hyung-Joon and jisukim.
The code is available at here.

3rd_Place

c3net's People

Contributors

bmycheez 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

Watchers

 avatar  avatar

c3net's Issues

about the dataset

hello, thanks for your sharing in this nice project with relative trained models. But i can't find the downloading link of the CFAMoire dataset, mayby because the competition is over. Could you send me this dataset? my email [email protected].

Thanks ~

from models_v2 import Net

hi,thanks for your best contributions,
when I try to run test.py based on your pre-trained models, i see no module named models_v2,and i try to exchange models_v2 to models, i see the program always fails,can you tell me how to solve this problem,thanks

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.