关于Unet训练时loss为负数的原因

最近用unet训练,但是训练时loss为 -5000左右,一个负数,绝对值还很大,其实这是训练集图片的格式不正确,我从unet的官方训练集下载下来mask看了一下,都是 1bit 的黑白图,但是我训练时是 8bit 的,所以要用到PIL中的Image.convert函数转换图像的位数,如果要转为1bit,需要用img = img.convert('1'),并且保存成png文件就可以了。

你可能感兴趣的:(opencv,算法设计,unet,PIL,图像位数转换,mask,loss函数)