sjmoran / CURL

Code for the ICPR 2020 paper: "CURL: Neural Curve Layers for Image Enhancement"
205 stars 34 forks source link

Pretrained models directory is missing. #13

Closed sanchit88 closed 3 years ago

sanchit88 commented 3 years ago

Hi, I just wanted to test out the pretrained CURL model on some test images. However, you have mentioned in the inference section that there must exist a pretrained_models directory and which should contain the following model:

Can you please make it available? Or, update a link from where it can be downloaded.

There exists a zip file CURL_for_RGB_images.zip which contains two pre-trained models. However, they are not the original model, i.e., they are trained until epoch 17 and epoch 30. They have higher PSNR values than mentioned in the above model. Thanks.

sjmoran commented 3 years ago

Hi, I would not recommend using those pre-trained models as they assume ProPhoto RGB colour space. A replacement pre-trained model will be upload for sRGB colour space soon.

sanchit88 commented 3 years ago

Hi SJ, so do you know when will it be released? Thanks.