mkartik / Shallow-UWnet

Shallow-UWnet, a neural network which maintains performance and has fewer parameters than the state-of-art underwater image enhancement model. Generalization of the model is demonstrated by benchmarking its performance on combination of synthetic and real-world datasets.
MIT License
51 stars 13 forks source link

where is the model.pth? #4

Closed Owen718 closed 3 years ago

Owen718 commented 3 years ago

Hi! Your work is very impressive! I want to test some underwater images used your implementation. Could you provide the model.pth file, please?

Wubl-cy commented 3 years ago

Hi!Have you achieved the results in the paper? Can you share your experience?THANK YOU!

Owen718 commented 3 years ago

Hi!Have you achieved the results in the paper? Can you share your experience?THANK YOU!

Hi! I am trying to design a SOTA NN model. I haven't tried training shallow-UWnet.But I had tested the FUniE-GAN on the EUVP-dark dataset. I think the water-net indicator on UIEB and the shallow-UWnet indicator on EUVP-dark given in the paper using different indicator test codes. If you have any questions, you can talk with me!

Wubl-cy commented 3 years ago

Thank you for your help. Due to the needs of the project, I would like to test the code related to underwater image enhancement. Do you have any other recommendations? If so, you are welcome to email me at @.***

At 2021-04-28 09:52:05, "Mr.Ye" @.***> wrote:

Hi!Have you achieved the results in the paper? Can you share your experience?THANK YOU!

Hi! I am trying to design a SOTA NN model. I haven't tried training shallow-UWnet.But I had tested the FUniE-GAN on the EUVP-dark dataset. I think the water-net indicator on UIEB and the shallow-UWnet indicator on EUVP-dark given in the paper using different indicator test codes. If you have any questions, you can talk with me!

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

mkartik commented 3 years ago

Hi @Owen718 and @wubilang, thank you, and sorry for my delay in response. I have now uploaded the model.ckpt (link in the README.md) which can be used to test the underwater images I have used in my implementation.