Lilac-Lee / PointNetLK_Revisited

Implementation for our CVPR 2021 oral paper "PointNetLK Revisited".
MIT License
47 stars 5 forks source link

How do I make my own data set for training? #14

Open 1404561326521 opened 9 months ago

1404561326521 commented 9 months ago

I want to perform point cloud registration on my own task, how can I only do my own data set, please advise, thank you very much!

Lilac-Lee commented 9 months ago

Hi @1404561326521, you could find a function get_datasets in the train.py file that loads training data from your dataset folder.

In the data_utils.py file, you could find a class named PointRegistration, where you write your own data loader function.

Cheers.