Home

Awesome

ParametricNet

ParametricNet is a new 6DoF pose estimation network for parametric shapes with keypoint learning and Hough voting scheme.

This is the code of tensorflow version for our ICRA2021 paper: ParametricNet: 6DoF Pose Estimation Network for Parametric Shapes in Stacked Scenarios.

Environment

Ubuntu 18.04/20.04

python3.6, tensorflow1.5, opencv-python, sklearn, h5py, et al.

Our backbone PointSIFT is borrowed from pointSIFT.

Dataset

Siléane dataset is available at here.

Parametric dataset is available at here.

Evaluation metric

The python code of evaluation metric is available at here.

Citation

If you use this codebase in your research, please cite:

@inproceedings{zeng2021parametricnet,
  title={ParametricNet: 6DoF Pose Estimation Network for Parametric Shapes in Stacked Scenarios},
  author={Zeng, Long and Lv, Wei Jie and Zhang, Xin Yu and Liu, Yong Jin},
  booktitle={2021 IEEE International Conference on Robotics and Automation (ICRA)},
  pages={772--778},
  year={2021},
  organization={IEEE}
}