hjwdzh / PrimitiveNet

PrimitiveNet: Primitive Instance Segmentation with Local Primitive Embedding under Adversarial Metric (ICCV 2021)
MIT License
62 stars 11 forks source link

Prediction when inputs are point clouds #3

Closed MulinYu closed 2 years ago

MulinYu commented 2 years ago

Hello,

Thanks for providing the source code.

I tried to use your code to segment a point cloud. I noticed that your code directly read the edge information from the '.npz' files. In your paper, you said that 'Otherwise we generate k=16 nearest neighbors from each point to form E'. So I think your code has the ability to segment a point cloud without providing 'F' in the input file, right? Can you explain how to do that?

Thanks in advance. Best. Mulin

hjwdzh commented 2 years ago

The scene data is processed without F. knn utility is in "src/lib/knn/knn.cc".