This question is about how to eliminate the white box in the image

bodokaiser / piwise

Pixel-wise segmentation on VOC2012 dataset using pytorch.

BSD 3-Clause "New" or "Revised" License

383 stars 86 forks source link

This question is about how to eliminate the white box in the image #21

Open mshmoon opened 6 years ago

mshmoon commented 6 years ago

Every label of image have a white margin , I want to ignore the margin when I train net ,But I do not know how to do,So i ask you this question Could you help me?

bodokaiser commented 6 years ago

You should be able to provide custom weights to the loss function so that you accumulate no gradients from the white margins.