search

XuyangBai / D3Feat

[TensorFlow] Official implementation of CVPR'20 oral paper - D3Feat: Joint Learning of Dense Detection and Description of 3D Local Features https://arxiv.org/abs/2003.03164
MIT License
259 stars 38 forks source link

error when run test_kitti #8

Closed tracykim closed 4 years ago

tracykim commented 4 years ago

hi,when i run test_kitti.py use pretrained model your provided, there is an error in
/utils/tester.py, line 326, in test_kitti T_gth = inputs['trans'] KeyError: 'trans'

i have checked inputs do not have a key 'trans', so how to solve it, thank you.

XuyangBai commented 4 years ago

Hi, @tracykim I just tried the code, and the code went well on my server. So could you check it again? The trans will be saved at this line so there should be a key trans.

tracykim commented 4 years ago

hi,thanks for your reply in time, i can save npz,what i means is this line

XuyangBai commented 4 years ago

Sorry I misunderstood your question, the ground truth transformation was saved in model.anchor_inputs in here so there should be a key named trans for KITTI, did you changed the name of the dataset in test_kitti.py? It should be working if you use the default test_kitti.py

Hope that helps.

tracykim commented 4 years ago

hi,you got it,thanks a lot!