facebookresearch / silk

SiLK (Simple Learned Keypoint) is a self-supervised deep learning keypoint model.
GNU General Public License v3.0
644 stars 58 forks source link

Have you try pretrained model in C++ Libtorch?? #56

Closed leechangyo closed 11 months ago

leechangyo commented 11 months ago

Hello, im currently try to use your pretrained model in c++ libtroch.

I save your model in pth by jit trace and test it in c++

However, it output of keypoints and descriptor not detected

keypoints : [0, 3] descriptors : [0,128]

and i try it same model in python it works fine keypoints :[10000, 3] descriptors : [10000, 128]

I also checked tensor on image in both python and c++ , are same dimension and same value

i wonder if you have tried this and meet same problem.

leechangyo commented 11 months ago

Vgg tensor input used and it works fine