Geodesic distance maps are extracted during the data preprocessing phase for generating the ground truth heatmaps. Compute the shortest geodesic distance from points to the keypoints and save the results as '.txt' files.
Modify the config config/keypoint_saliency.yaml for your path, filename, and categories.
...
python train.py
python test.py
@inproceedings{jin2024keypointdetr,
title={KeypointDETR: an end-to-end 3d keypoint detector},
author={Jin, Hairong and Shen, Yuefan and Lou, Jianwen and Zhou, Kun and Zheng, Youyi},
booktitle={ECCV},
year={2024}
}