Trained model? about uwgan_uie

[infrontofme]

the trained model has been recovered, the pre-trained model and datasets will be made publicly available soon.

[02250119]

I met the same problem

[infrontofme]

@branimir29 @02250119
Thank you very much for your interest in our project.

Our computing server crashed two weeks ago due to some unexpected reasons, resulting in data loss on the server, including pre-trained models for our project. We are trying to recover the data or retrain our model, but this may take several weeks. Please be patient.

We used only part of the data in the NYU Depth Datasets to train UWGAN, and this part of the data was uploaded to Baidu Cloud Link. With these RGB-D data and underwater RGB images, you should be able to use UWGAN to synthesize underwater-style images, and then you can train U-net with image pairs.

You are welcome to follow our project at any time, if there is any update, we will push it on Github.

[branimir29]

Thanks for quick reply @infrontofme . There is no way for me to access Baidu Cloud, any chance you could forward me that package from Baidu via WeTransfer, GDrive, Dropbox?
It would be a great help.

[infrontofme]

@branimir29
We have uploaded datasets to Google Drive, you can find underwater images here WaterImages, and you can find RGB-D images here NYU.
If you have any questions about our project, please contact us.

[02250119]

Thank you for your reply.It will help me.

[branimir29]

That's really nice of you, thx. I downloaded Water images, but NYU require your permission to access. I sent access request from my gmail address you'll see.

[Jinksi]

@infrontofme did you recover the trained model and make it publicly available?

[priyank200]

Hello, @infrontofme, can you please provide me the requirement.txt file? I want to run the code with a small dataset.