search

infrontofme / UWGAN_UIE

Source code for UWGAN: Underwater GAN for Real-world Underwater Color Restoration and Dehazing
https://arxiv.org/abs/1912.10269
MIT License
190 stars 35 forks source link

Trained model? #1

Closed branimir29 closed 4 years ago

branimir29 commented 4 years ago

Hello, I am interested in trying your method on my own underwater images. However, downloading full NYU image sets requires for V1 90GB (Raw images) and for v2 it's 428GB for V2. I simply don't have enough space or time to wait to download all this. So, I was wondering if you could upload pre-trained UWGAN/U-Net models instead that one could simply download and try on real life underwater images? It would be a great help.

02250119 commented 4 years ago

I met the same problem

infrontofme commented 4 years ago

@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 commented 4 years ago

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 commented 4 years ago

@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 commented 4 years ago

Thank you for your reply.It will help me.

branimir29 commented 4 years ago

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 commented 4 years ago

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

infrontofme commented 4 years ago

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

priyank200 commented 1 year ago

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