search

Atif-Anwer / SHMGAN

Joint network for specular highlight detection and adversarial generation of specular-free images trained with polarimetric data
https://authors.elsevier.com/a/1hqRr3INukOUwT
3 stars 0 forks source link

Trained weights #1

Closed JovinLeong closed 3 months ago

JovinLeong commented 3 months ago

Hi, are the trained weights from the paper available?

Atif-Anwer commented 3 months ago

Hi Leong!

Thankyou for taking interest in the paper. Unfortunately, at the moment I cannot provide the trained weights as I dont have access to them at the moment as the work is still in progress.

Thanks again!

JovinLeong commented 3 months ago

Hi Anwer, thanks for your reply! Understood - so in that case, to reproduce your results, the idea would be for us to retrain the model on the PSD and SHIQ datasets?

Atif-Anwer commented 3 months ago

Yes, that would be the ideal way. It should train without any errors hopefully if you run it on a single GPU, since it wasnt coded for multi-gpu configurations..