fine tuning sam

[theCuriousDev1]

Hello team,

great work.

I had one doubt, during fine tuning, What is used as ground truth to calculate the loss against the masks generated by SAM with any prompt?

From the code I see, you use the masks from the dataset. Is it?

[dyzy41]

I also have this question. The code uses gt_masks to calculate focal_loss and dice_loss. If you use this code to make your own data set, how to obtain bbox and gt_mask. thank you for your reply