CASIA-IVA-Lab / FastSAM

Fast Segment Anything
GNU Affero General Public License v3.0
7.35k stars 687 forks source link

about the train code #60

Closed yo-WASSUP closed 1 year ago

yo-WASSUP commented 1 year ago

Thanks for your work, does the training code mean that I can use my labeled segmented data containing one class to train a model that segments only that one class?

感谢您的工作,请问训练代码是指可以使用我包含一个类的已标注分割数据训练一个只分割这一个类的模型吗?

zxDeepDiver commented 1 year ago

Yes. 是的

Sijie-L commented 1 year ago

So why in the code it says the training is not supported?

def train(self, **kwargs):
    """Function trains models but raises an error as FastSAM models do not support training."""
    raise NotImplementedError("FastSAM models don't support training")
zxDeepDiver commented 1 year ago

So why in the code it says the training is not supported?

def train(self, **kwargs):
    """Function trains models but raises an error as FastSAM models do not support training."""
    raise NotImplementedError("FastSAM models don't support training")

This version of codes does not support training. We are working to release the training codes recently.

Rwik2000 commented 1 year ago

When can we expect the training code?

zxDeepDiver commented 1 year ago

When can we expect the training code?

If you wish to promptly obtain the code for training or fine-tuning purposes, kindly reach out to us via email and specify your intended use case. We welcome your usage across various scenarios. For emailing, we recommend contacting us at xu.zhao@nlpr.ia.ac.cn.