search

yd-yin / SAI3D

[CVPR 2024] SAI3D: Segment Any Instance in 3D Scenes
85 stars 5 forks source link

How to segment my own dataset #6

Open WPC-WHU opened 5 months ago

WPC-WHU commented 5 months ago

How to segment my own dataset? 2D image is must?

Ly-kc commented 5 months ago

Our current implementation requires 2D images, depth, and poses. If your dataset doesn’t include 2D images, you may need to render the 3D scene to obtain them. Once you have all this 2D data prepared, you can customize the functions get_mask_data, get_seg_data, and the code begining at line 234 in sai3d.py to suit your specific data.

Lee123lyl commented 1 month ago

If I only have 3D data (such as the S3DIS dataset), after I get the 2D images, how do I get the depth and poses?