Coder Social home page Coder Social logo

c-yn / adair Goto Github PK

View Code? Open in Web Editor NEW
88.0 2.0 5.0 675 KB

AdaIR: Adaptive All-in-One Image Restoration via Frequency Mining and Modulation

License: MIT License

Python 100.00%
image-deblurring image-dehazing image-denoising image-deraining low-light-image-enhancement pytorch all-in-one-image-restoration

adair's People

Contributors

c-yn 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

Watchers

 avatar  avatar

adair's Issues

BSD400 Denoising Dataset.

Hello,

Thank you for sharing this interesting study.

I have one question about the denoise dataset. In the "denoise.txt" file, the BSD400 data file names appear to differ from the acutal image names (e.g. 'a117025.jpg' in denoise.txt / '117025'jpg' in the actual image name). Could you please clarify if the BSD400 image are used for training in the All-in-one IR??

Thank you for your help.

May I ask for a demo.py file?

Sorry to bother you in your busy schedule! I wanna ask do you have a "demo.py" file similar to the one provided by "PromptIR" that I can use to test other images? Thanks a lot!

Untrained parameter problem

AdaIR/net/model.py

Lines 339 to 347 in 69e13fb

x = self.conv1(x)
mask = torch.zeros(x.shape).to(x.device)
h, w = x.shape[-2:]
threshold = F.adaptive_avg_pool2d(x, 1)
threshold = self.rate_conv(threshold).sigmoid()
for i in range(mask.shape[0]):
h_ = (h//n * threshold[i,0,:,:]).int()
w_ = (w//n * threshold[i,1,:,:]).int()

I found a problem, since I was training with DDP, that would indicate the presence of parameters that were not involved in the training. Through my investigation, self.score_gen and self.conv are unnecessary, and these problems are not serious. But the most important thing is that self.rate_conv will not participate in the gradient calculation, because the operation of generating mask with threshold is not differentiable.

Code for inference

Hello,

terrific work, is available the code for running inference? are the model weights available? Could be possible to run it without the need for training?

Thank you for your help

About the five-degradation table in paper

In the top super-row of the five-task table mentioned in the paper, was the accuracy of the single-task models trained with your training settings? I did not find the accuracies provided in your paper in these papers.

Training cost and GPU requirements

Hi @c-yn ,

Thanks for your interesting explorations in IR from the frequency perspective.

May I know how many GPUs (And which type you used) are needed for training the 3-task setting and the 5-task setting ?

Especially, how long it costs to train a full model with your GPU setting.

Thanks in advance, this would help me a lot as a reference to follow this nice work.

Best and have a nice day,

FMiM

where is FMiM

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.