在进行u-net医学图像分割保存文件时出现下列警告,查了很多资料后发现下面这个方法可以解决我的问题
Lossy conversion from float32 to uint8. Range [0, 1]. Convert image to uint8 prior to saving to suppress this warning.
解决:在当前文件(一般为data.py)导入后
from skimage import img_as_ubyte
在保存的函数里将这句代码改为下面一种,即将img改为img_as_ubyte(img)
io.imsave(os.path.join(save_path,"%d_predict.png"%i),img)
io.imsave(os.path.join(save_path,"%d_predict.png"%i),img_as_ubyte(img))