search

ethnhe / PVN3D

Code for "PVN3D: A Deep Point-wise 3D Keypoints Hough Voting Network for 6DoF Pose Estimation", CVPR 2020
MIT License
488 stars 105 forks source link

Training other datasets with pvn3d #48

Closed BenjaminBo closed 3 years ago

BenjaminBo commented 3 years ago

Hi, I would like to train other datasets using pvn3d. The datasets are given here https://bop.felk.cvut.cz/datasets/. As a first step I would like to train the "tless" dataset. I'm not quite sure yet, how I would go about doing that.

Could somebody help me understand how I would have to change the format of the tless dataset or how and where I would have to adjust the code so that training on the tless dataset would be possible?

BenjaminBo commented 3 years ago

@ethnhe

ethnhe commented 3 years ago

You can revise the data preprocess script dataets/ycb/ycb_dataset.py or datasets/linemod/linemod_dataset.py to fit tless dataset. Also, you may need to revise some variables in common.py

ethnhe commented 3 years ago

The guide for adaptation to new datasets has been added in README.

engqing commented 1 year ago

@BenjaminBo Have you solved the problem yet?