search

ZhaoyangLyu / Point_Diffusion_Refinement

A Conditional Point Diffusion-Refinement Paradigm for 3D Point Cloud Completion
Apache License 2.0
143 stars 20 forks source link

Question about REFINEMENT NETWORK #3

Open yufeng9819 opened 2 years ago

yufeng9819 commented 2 years ago

Hey! Thanks for your wonderful work!

I am interested in your work. I don't know if I understand correctly: it is possible to apply REFINEMENT NETWORK to refine any other corse point clouds by training on other point cloud datasets. But I can not know how to train REFINEMENT NETWORK from your given code. So could you please give me some guidance for training on other point cloud dataset？

Or are there any checkpoints from training on shapenet?

Thanks for your help! Looking forward to your reply.

ZhaoyangLyu commented 1 year ago

It is possible to train the REFINEMENT NETWORK on other datasets. You just need to give the correct inputs to the REFINEMENT NETWORK. It need two inputs: The first is the coarse point clouds and the second is the condition point cloud (incomplete point cloud in our case). And use the ground-truth clean point cloud as supervision. See line 475-491 in train.py for how to use the REFINEMENT NETWORK.