WZH0120 / SAM2-UNet

SAM2-UNet: Segment Anything 2 Makes Strong Encoder for Natural and Medical Image Segmentation
Apache License 2.0
163 stars 18 forks source link

Do you support adding normal samples to Sam2-Unet network training? #28

Open AaddX-ai opened 1 week ago

AaddX-ai commented 1 week ago

What I mean is that I added normal samples to the training, and the mask image is all black. Will this reduce false positives and improve segmentation stability ?

xiongxyowo commented 1 week ago

Hi, for most public segmentation datasets, both the training and test sets contain only positive samples. In practice, this can lead to a higher false positive rate. Adding negative samples to the training set can help mitigate this, but it may lower the accuracy on positive samples. To address this, you might consider carefully adjusting the loss function to make the network more compatible with negative samples, as current IoU-based losses tend to be quite sensitive to all-zero black masks.

AaddX-ai commented 1 week ago

many thanks for u@xiongxyowo