theodorakontogianni / InterObject3D

MIT License
28 stars 1 forks source link

Runing Speed #5

Open congwei45 opened 2 months ago

congwei45 commented 2 months ago

When I try to segment a point cloud about 30M, the code becomes very slow. How can I solve this problem?

theodorakontogianni commented 2 months ago

Is the visualization slow or the inference? What GPU do you use? Some options since 30M is a relatively large point cloud:

  1. Adapt the code after the first positive click to operate (and visualize) only on a 3D bounding box roughly placed around the object to segment. In most cases, you will not need to click all over the scene for an object relatively small. If the object spans the whole scene (eg. a construction beam) it can always be merged together on post-processing based on the same identifier/label.
  2. Crop the scene to smaller pieces in pre-processing
  3. The inference runs on CPU also if needed, it might help with memory.
congwei45 commented 2 months ago

The visualization and inference speed are both slow. I am using a RTX 4090 GPU. Thank you very much for your suggestion. I am taking the approach of cropping to speed it up.