search

openmedlab / MedLSAM

MedLSAM: Localize and Segment Anything Model for 3D Medical Images
Apache License 2.0
495 stars 19 forks source link

Value Error at inference #9

Open ManuBN786 opened 11 months ago

ManuBN786 commented 11 months ago

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.

LWHYC commented 11 months ago

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.

ManuBN786 commented 11 months ago

Thanks for the reply.

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

LWHYC commented 10 months ago

Author

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