search

Pointcept / SegmentAnything3D

[ICCV'23 Workshop] SAM3D: Segment Anything in 3D Scenes
https://arxiv.org/abs/2306.03908
MIT License
1.02k stars 70 forks source link

Why does "sam3d.py" run as a ".pth" file? #12

Closed LinLin1031 closed 1 year ago

LinLin1031 commented 1 year ago

Why does "sam3d.py" run as a ".pth" file? Normally, shouldn't the result of semantic segmentation be a tag file or a point cloud file?

".pth" file is the final training result? If so, how should I test it?

LinLin1031 commented 1 year ago

@yhyang-myron @Gofinge Hope to hear from you soon, thanks!

ayaan-mphasis commented 1 year ago

Hi, I have the same doubt. Could you please clarify. And how can I possible visualise its results

yhyang-myron commented 1 year ago

Hi. The group_id generated by sam3d.py is in the .pth file. And you visualize it with here.

ayaan-mphasis commented 1 year ago

@yhyang-myron The .pth file that is being generated as the output for me is a simple array with shape (81369,) Is there supposed to be a key names as group_id in the output? I want to interpret the output .pth file better.

yhyang-myron commented 1 year ago

The array in the final .pth file is the group_id for every point in the original point clouds, and you can use it with original coordinates and color.