Is there any '.ckpt' format pretrained model on SALICON?

alexanderkroner / saliency

Contextual Encoder-Decoder Network for Visual Saliency Prediction [Neural Networks 2020]

MIT License

172 stars 46 forks source link

Is there any '.ckpt' format pretrained model on SALICON? #12

Closed LJOVO closed 3 years ago

LJOVO commented 3 years ago

Dear authors,

I am trying to fine-tune the provided '.pd' format pretrained model but failed. May I request a re-trainable '.ckpt' format pretrained model on SALICON?

Thanks!

alexanderkroner commented 3 years ago

Sure, I uploaded the .ckpt files for the SALICON version here:

CPU: https://drive.google.com/file/d/1ZWRCKQyzjtT_D1HcxT-IXSlVfogB1uJw/view
GPU: https://drive.google.com/file/d/1m9ZvZJXfBqOocAE81dgGMX7L049-oi2q/view

LJOVO commented 3 years ago

Sure, I uploaded the .ckpt files for the SALICON version here:

CPU: https://drive.google.com/file/d/1ZWRCKQyzjtT_D1HcxT-IXSlVfogB1uJw/view

GPU: https://drive.google.com/file/d/1m9ZvZJXfBqOocAE81dgGMX7L049-oi2q/view

Thank you vey much!