search

google-research / maxim

[CVPR 2022 Oral] Official repository for "MAXIM: Multi-Axis MLP for Image Processing". SOTA for denoising, deblurring, deraining, dehazing, and enhancement.
https://arxiv.org/abs/2201.02973
Apache License 2.0
1.02k stars 110 forks source link

Training code #8

Closed EricYu97 closed 2 years ago

EricYu97 commented 2 years ago

Hello, I'm very impressed to see the results of your maxim model and want to train with my dataset. But I do not see any training code in this repository. Would you like to commit the training code? Thanks!

vztu commented 2 years ago

Hi we don't have any plans to release the training code on Jax, unfortunately.

But we plan to reimplement it in PyTorch and provide training code. Please keep an eye on maxim-pytorch