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
977 stars 105 forks source link

TensorFlow/Keras port #30

Closed sayakpaul closed 1 year ago

sayakpaul commented 1 year ago

@vztu @Yinxiaoli

I worked on porting all the public MAXIM models to TensorFlow: https://github.com/sayakpaul/maxim-tf.

There are some WIPs: inference Colab, polished README, etc. Actively working on those.

Here, one can find all the ported SavedModels: gs://maxim-tf.

Additionally, I have created several Space applications on Hugging Face for better interactivity with the models:

Will be adding the rest in the coming days. Would you be willing you to add a line about the TF repository from the README? I will be publishing the TF ported models on TensorFlow Hub too. As of today, they're available on Hugging Face: https://huggingface.co/sayakpaul.

@innat I also saw #29. Seems like we both started working on porting the models. If you're interested, you can contribute the rest of the Spaces applications.

Cc: @gustheman

innat commented 1 year ago

@sayakpaul This is great, will look into it. Along with tf-hub, I think it would be nice to have maxim from keras-cv, wdyt? I made a ticket.

cc. @LukeWood

sayakpaul commented 1 year ago

Sure. If the KerasCV team includes it in the ROADMAP, I don't see any problems. But we can always revisit it. Thanks for the support :)

innat commented 1 year ago

As no implementation cost from scratch, I think it shouldn't be a problem. @LukeWood @tanzhenyu

vztu commented 1 year ago

Thanks a lot for these great efforts @sayakpaul @innat! We'll be updating the README to acknowledge your codebase as well as huggingface spaces soon. Please let us know if you need any supports on our side.

sayakpaul commented 1 year ago

Thank you :)

sayakpaul commented 1 year ago

Thanks for mentioning the project. Would you also mind mentioning Space application links?

vztu commented 1 year ago

Sure, glad to do that. I found you have multiple spaces for each task. Could you make it more unified so that I only need to add a single link? Or you can summarize them in your README file and I point to that.

sayakpaul commented 1 year ago

I created a PR: https://github.com/google-research/maxim/pull/33

vztu commented 1 year ago

Merged. Thank you!

sayakpaul commented 1 year ago

Thanks!