meidachen / STPLS3D

🔥 Synthetic and real-world 2d/3d dataset for semantic and instance segmentation (BMVC 2022 Oral)
236 stars 20 forks source link

Pre-trained models and dataset generation #26

Open ageorgiev97 opened 1 year ago

ageorgiev97 commented 1 year ago

I have two questions:

  1. Are there any pre-trained models for the baseline?
  2. Are you planing to provide the tooling/code for generating the dataset?
meidachen commented 1 year ago

Hi @ageorgiev97 ,

Yes, and yes. We are still preparing the models, are you interested in the semantic segmentation or the instance segmentation?

meidachen commented 1 year ago

Hi @ageorgiev97 ,

Yes, and yes. We are still preparing the models, are you interested in the semantic segmentation or the instance segmentation?

ageorgiev97 commented 1 year ago

Hi! Thanks for the answer I'm more interested in the instance segmentation case. Also when we can expect the scripts for generating the dataset?

meidachen commented 1 year ago

We are still preparing the data generation scripts and instructions since the scripts were not currently all connected smoothly and a lot more work needed to be done to make them useable for users that are not familiar with different tools in the pipeline.

For instance segmentation, I can share with you the model for sure, but I would recommend you to check SoftGroup at https://github.com/thangvubk/SoftGroup where they have already released the pre-trained model for our data (the reason is that SoftGroup has already outperformed our baselines). Let me know if for any reason you need the HAIS model, and I will find it and upload it.

LujZhang commented 1 year ago

Thanks a lot for your work. You provide a perfect semantic segmentation dataset, the codes and experiments of several exemplary methods. Do you plan to make the semantic segmentation pre-trained model public?

Rambo8248 commented 11 months ago

i'd like to use the HAIS pre-trained model to test the instance segmentation result on OUR OWN dataset. But the training is time-consuming, can you provide the HAIS pre-trained model for SPTLS3D? Thank you very much!!