JonasSchult / Mask3D

Mask3D predicts accurate 3D semantic instances achieving state-of-the-art on ScanNet, ScanNet200, S3DIS and STPLS3D.
MIT License
528 stars 105 forks source link

How to train Mask3D with a custom dataset. #50

Closed IvanGarcia7 closed 1 year ago

IvanGarcia7 commented 1 year ago

Hello.

I'm really interested to know how to train the model using my own pointcloud dataset, for my thesis. Can you provide some instructions about how to do that?

Thanks in advance!!

JonasSchult commented 1 year ago

Hi Ivan,

thanks for your interest in our work. I unfortunately do not have detailed instructions to integrate new datasets in the Mask3D codebase.

I recommend to take a look into the preprocessing folder (here) to see how we preprocessed different datasets to a common format which we used in the rest of the codebase. If you write such a preprocessing script for your custom dataset, you can use it with Mask3D.

Let me know if you have any further questions! :)

Best, Jonas