search

BAAI-DCAI / SegVol

The official code for "SegVol: Universal and Interactive Volumetric Medical Image Segmentation".
MIT License
248 stars 22 forks source link

FH mask #27

Open baiti01 opened 2 weeks ago

baiti01 commented 2 weeks ago

Thank you for the great work. Was wondering whether why you use the FH mask as extra supervision? Will the performance be affected significantly if remove it? Do you have some ablation studies? Thank you.

Yuxin-Du-Lab commented 2 weeks ago

Thanks for your support! Our purpose is to create a universal segmentation method like SAM. If we remove the FH masks, the model will perform badly for the unlabeled area in CTs due to the serious partial label problem in volumetric medical image segmentation datasets. But if you just want it to perform well in a fixed category set, getting rid of the FH mask will improve the performance to some degree. Just depends on your needs.

We conducted some ablation studies during the prototype design stage but they are not systematized so we do not have plan to publish them.