Using pretrained models

liufeng2915 / Implicit_Dense_Correspondence

60 stars 13 forks source link

Using pretrained models #3

Open vikasTmz opened 3 years ago

vikasTmz commented 3 years ago

Hi,

The pretrained models you've provided in the Google Drive contain the following files:

AlbedoDecoder.pth
ShapeDecoder.pth
pretained_latents.mat

How do we use these pretrained weights? The evaluateBHCP.py script requires the checkpoint file to contain the following three keys: InverseImplicitFun_state_dict, Encoder_state_dict and ImplicitFun_state_dict.

Also, which category and dataset are the pretrained weights for?

Thanks.

liufeng2915 commented 2 years ago

Please download the pre-trained models from: https://drive.google.com/file/d/1ENjNVoVQfzmlUGpjbHpGC9xkBsDnwuhx/view?usp=sharing