search

cameronfabbri / Underwater-Color-Correction

Using GANs to correct color distortion in underwater images.
291 stars 86 forks source link

either dataset or pretrained model #1

Closed pribadihcr closed 6 years ago

pribadihcr commented 6 years ago

hi @cameronfabbri , Is there the dataset that can we try it or the pretrained model?

cameronfabbri commented 6 years ago

Hi, yes you can download the dataset here.

You can also find this link on the project page. I'll upload a pretrained model later today.

Xiaodong-Bran commented 6 years ago

@cameronfabbri
Hi , is it possible to upload the test image with the ground truth i.e. the image before distortion?

Thanks !

cameronfabbri commented 6 years ago

@Xiaodong-Bran actually the test images I am using have no ground truth. That's the difficulty in the underwater domain, we only have access to the distorted images, i.e those are "ground truth".

Xiaodong-Bran commented 6 years ago

@cameronfabbri Thanks for your reply. Understood. By the way, do you know why my training result often make images resulting in red hue ( with your provided training pair,around 100 epochs) ?

I found your pretrained model is around 100,000 step with batch-size 32, that will be around 500 epochs right? Do I need to keep on training these models?

Thanks!