Segmentation also uses Threshold

milesial / Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

GNU General Public License v3.0

9.3k stars 2.51k forks source link

Segmentation also uses Threshold #503

Open BenjaminSchaar opened 4 months ago

BenjaminSchaar commented 4 months ago

Hi, for a segmentation task i wanted the unet to ignore some parts of the real object, but it doesnt seem to understand it nd always create the mask also threshold based binarisation would get.