openmedlab/MedLSAM

Value Error at inference

Opened this issue · 3 comments

Screenshot from 2023-11-20 11-24-14

How to resolve this error please?

Also I want to know if this model can be used to segment images outside the 22 classes mentioned in the paper.

I'm planning to use it for lung nodules segmentation.

Screenshot from 2023-11-20 11-24-14

How to resolve this error please?

Also I want to know if this model can be used to segment images outside the 22 classes mentioned in the paper.

I'm planning to use it for lung nodules segmentation.

Hi,
It seems like there is no fg pixel in your label. I suggest you check your support label data to make sure it contains your interested fg classes.
MedLSAM could be used to segment any anatomical structure and organs, but not the lesions and diseases. So it could not segment lung nodules.

Thanks for the reply.

Can you please provide some training scripts and methodologies so one can train this model for nodule detection?

Author

Hi,
Sorry for the late reply. We have added the training code here: https://github.com/openmedlab/MedLSAM/tree/main/train