ljlijie / WaterGAN-color-correction-net

Customized verison of Caffe that support underwater color correction net.
https://github.com/ljlijie/WaterGAN-color-correction-net
Other
16 stars 11 forks source link

Watergan maybe needs depth information to producing underwater composite images #2

Open IPNUISTlegal opened 6 years ago

IPNUISTlegal commented 6 years ago

https://github.com/kskin/WaterGAN/issues/3 The code seems to fail without depth data(*.mat).

djjandXMU commented 5 years ago

You should download NYU dataset.Then you can load the nyu_depth_data_labeled by MATLAB.After that,you can check out the variables about the .mat, you will find 2200+images. Next, you could read this images and depth.mat and save it by yourself. I have trained the model sucessfully. But as for the correction model, I have no idea.