honeygupta / UW-Net

Code for paper [ICIP 2019] Unsupervised Single Image Underwater Depth Estimation (UW-Net)
MIT License
27 stars 14 forks source link

Do you have the evaluation code for Berman dataset? #10

Closed AttackonMuggle closed 2 years ago

AttackonMuggle commented 2 years ago

Thank you for the pre-trained model and test code. I'm a little confused about evaluating the Berman dateset. Could you please share the evaluation code for Berman you showed in Tab 1?

honeygupta commented 2 years ago

Hi, you can find the evaluation codes in this script - https://github.com/honeygupta/UW-Net/blob/master/evaluate_metrics.py

AttackonMuggle commented 2 years ago

Thanks for your reply, but how did you prepare the ground-truth images. Since in evaluate_metrics.py line 41, the gt between 0-255 will all be converted to 0.