Skip to content

feat(PCSegSAM2): auto pointcloud segmentation using SAM2 and GroundingDino #44

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 10 commits into
base: main
Choose a base branch
from

Conversation

knzo25
Copy link
Contributor

@knzo25 knzo25 commented May 18, 2025

Summary

Since we lack large segmentation databases I leveraged the use of SAM2, GroundingDino, and the vehicle calibration to segment pointclouds using open source foundation models.

This project allows to:

  • Segment t4dataset images using a custom vocab
  • Segment t4dataset pointclouds using the previous image segmentations
  • Refine pointcloud segmentation using 3d object detection cuboids
  • Generate pointcloud segmentation videos

This project essentially copies
https://github.com/IDEA-Research/Grounded-SAM-2/tree/main
and implement the necessary features.
I asked @scepter914 and there was no reticence to just paste all the required code as long as there was no issues with the licences.

Change point

Add a new project for automatic pointcloud segmentation

Note

Since it is a new project with an independent docker, it does not affect any other project in the codebase

Test performed

Build docker, run all tasks presented in the README, and segmented t4data (during April)

knzo25 and others added 9 commits April 11, 2025 00:16
 - non-neded files are present, but since a squash merge is expected, it should not be an issue
 - sam2 for t4datasets
 - projective approach to segment pointclouds

missing:
 - temporal consistency criteria
 - hardcoded filters for pointcloud segmentation

Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
@knzo25 knzo25 requested review from scepter914 and amadeuszsz May 18, 2025 14:16
@knzo25 knzo25 self-assigned this May 18, 2025
@scepter914
Copy link
Collaborator

Memo
As whole design of AWML, changes of this PR looks great to me.
I asked to review code-level for @amadeuszsz 🙏

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants