Thanks for your codes and I have finished the single-class segmentation. Now I want the multi-class segmentation tasks can be supported, where should the modifications be made? I tried to modify the "pred_mask" part in evf-sam2.py, but I'm not sure how to change the output of "p" to support multiple classes. Currently, it's all set to 1. My gt_masks like (batchsize,num_class,w,h).
Thank you to the author for patiently reading through this. If you happen to see this, I would greatly appreciate any guidance or advice you could offer.
Thanks for your codes and I have finished the single-class segmentation. Now I want the multi-class segmentation tasks can be supported, where should the modifications be made? I tried to modify the "pred_mask" part in evf-sam2.py, but I'm not sure how to change the output of "p" to support multiple classes. Currently, it's all set to 1. My gt_masks like (batchsize,num_class,w,h).
Thank you to the author for patiently reading through this. If you happen to see this, I would greatly appreciate any guidance or advice you could offer.