Skip to content

source code for paper "Learning Better Keypoints for Multi-Object 6DoF Pose Estimation".

License

Notifications You must be signed in to change notification settings

aaronWool/keygnet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

KeyGNet: Learning Better Keypoints for Multi-Object 6DoF Pose Estimation

teaser

Learning Better Keypoints for Multi-Object 6DoF Pose Estimation Yangzheng Wu, Michael Greenspan WACV 2024

Preliminary

Dependencies

Install PyTorch prior to other python packages: Install the rest dependencies by using conda:

conda env create -f environment.yml
conda activate keygnet

Datasets

Download BOP core datasets from BOP website.

Pre-trained keypoints for testing

Download the pre-trained keypoints from 'keypoints' folder and unzip it into the root of each dataset.

Training

Train KeyGNet

python train.py --dataset dname --batch_size 32 --keypointsNo 3 --lr 1e-3 --data_root "PathToData" --ckpt_root "PathToLogs"

Testing

Test with the keyGNet keypoints by using RCVPose, PVNet, and PVN3D github repositories.

Citation

If you find our work useful in your research, please consider citing:

@inproceedings{wu2024keygnet,
  title={Learning Better Keypoints for Multi-Object 6DoF Pose Estimation},
  author={Wu, Yangzheng and Greenspan, Michael},
  booktitle={WACV},
  year={2024}
}

About

source code for paper "Learning Better Keypoints for Multi-Object 6DoF Pose Estimation".

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages