google-research / maxim

[CVPR 2022 Oral] Official repository for "MAXIM: Multi-Axis MLP for Image Processing". SOTA for denoising, deblurring, deraining, dehazing, and enhancement.

Home Page:https://arxiv.org/abs/2201.02973

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training code

EricYu97 opened this issue · comments

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!

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