zyc00 / Point-SAM

Point-SAM: This is the official repository of "Point-SAM: Promptable 3D Segmentation Model for Point Clouds". We provide codes for running our demo and links to download checkpoints.
MIT License
138 stars 8 forks source link

Question: Which datasets was pretrained Point-SAM trained on? #9

Open SpeedyGonzales949 opened 1 month ago

SpeedyGonzales949 commented 1 month ago

Hi! Thanks for the awesome work!

I saw you published the weights file on HugginFace. Which Datasets did you use for this specific training ? Did you use every dataset mentioned in the paper ?

Thanks for the help!

Jiayuan-Gu commented 1 month ago

Yes, we used all the datasets mentioned in the paper: PartNet, PartNet-Mobility, Fusion360, ScanNet, and ShapeNet (pseudo labels).

LinLin1031 commented 2 weeks ago

Yes, we used all the datasets mentioned in the paper: PartNet, PartNet-Mobility, Fusion360, ScanNet, and ShapeNet (pseudo labels).

If I want to migrate Point-SAM's capabilities to other datasets, how should I train them? In other words, how is the pre-trained model ‘model.safetensors’ trained? Can you provide the training code? @Jiayuan-Gu @zyc00