unet
unet copied to clipboard
data.py
first, thank you for your work . and then I want to ask some question about data.py file when I read the code about data.py, I found that the codes like:
elif(np.max(img) > 1):
img = img / 255
mask = mask /255
mask[mask > 0.5] = 1
mask[mask <= 0.5] = 0
Does this mean that you are predicting the background, because the background is white, 255?
it define 2 class white and black, so if the mask less than 122,5 it gonna be black and if the mask more than 122,5 it gonna be white