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

Add Web Demo & Docker environment #11

Closed chenxwh closed 2 years ago

chenxwh commented 2 years ago

This pull request makes it possible to run your model inside a Docker environment, which makes it easier for other people to run it. We're using an open source tool called Cog to make this process easier.

This also means we can make a web page where other people can try out your model! We have implemented Image Denoising, Deblurring, Dehazing, Deraining and Enhancement in one demo, view it here: https://replicate.com/google-research/maxim. You can find the docker file under the tab ‘run model with docker’.

We have added some examples to the page, but do claim the page so you can own the page, customise the Example gallery as you like, push any future update to the web demo, and we'll feature it on our website and tweet about it too. You can find the 'Claim this model' button on the top of the page. Any member of the google-research organisation on GitHub can claim the model ~

In case you're wondering who I am, I'm from Replicate, where we're trying to make machine learning reproducible. We got frustrated that we couldn't run all the really interesting ML work being done. So, we're going round implementing models we like. 😊

vztu commented 2 years ago

Hi @chenxwh, thanks a lot for helping us create these useful tools for easier usage! I have checked the web demo and find it looks super awesome!

I have just one minor request -- I simply tweaked with the deblurring model and found that it sometimes doesn't work very well on real-world pictures (since trained on synthetic datasets). So I am wondering could you also help create the other three deblurring models we provided here&prefix=&forceOnObjectsSortingFiltering=false)? I believe it will allow users to have multiple choices to try if one deblurring model doesn't work as expected. Thanks again for your hard work in creating this cool demo!!

chenxwh commented 2 years ago

Hi @vztu ,

Glad to hear you like the Replicate demo :)

I have now added the other three deblurring models to the demo. Let me know if any other changes you'd like. Do claim the page with the button at the top if you are happy with the demo, you can then add additional examples (and change the current ones) ~

We'd also like to encourage you to try out Cog as well, with cog.yaml for setting up environment and predict.py for inference, then cog push r8.im/google-research/maxim for pushing to the web page (after you are logged in and claimed the ownership for a page), it is easy to update the demo anyway you like :)

Yinxiaoli commented 2 years ago

Thanks, Chenxi !

zeke commented 2 years ago

Hi @vztu and @Yinxiaoli 👋🏼

I'm Zeke from the @replicate team.

This model is so cool! I find the "de-raining" feature especially impressive. ☔

I just wanted to let you know that the arXiv.org website now includes a link to your model on Replicate!

See the "Demos" tab at the bottom of the page here: https://arxiv.org/abs/2201.02973

Screen Shot 2022-05-02 at 2 31 21 PM