KeypointDETR: an end-to-end 3d keypoint detector

Run

1. Extracting Geodesic Distance Maps

scripts/geodesic_distance.py

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.

2. Config

Modify the config config/keypoint_saliency.yaml for your path, filename, and categories.

3. Requirement

...

4. Train

python train.py

5. Test

python test.py

Citation

@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}
}

Acknowledgement

DGCNN

Point Transformer

KeypointNet Dataset