最近在做深度学习,涉及到json数据集文件转化为适合UNet模型的mask数据集文件,在将numpy.ndarray转换为图片的过程中,发现转换的图片一直是黑色的,
查阅相关资料发现可能是因为Image.fromarray
的浮点输入定义不正确;它没有很好的文档说明,但是该函数假定输入的布局为无符号的8位整数,要生成您想要获得的输出,请乘以255并转换为uint8
:
为此,只要在之前加入一行代码即可
最近在做深度学习,涉及到json数据集文件转化为适合UNet模型的mask数据集文件,在将numpy.ndarray转换为图片的过程中,发现转换的图片一直是黑色的,
查阅相关资料发现可能是因为Image.fromarray
的浮点输入定义不正确;它没有很好的文档说明,但是该函数假定输入的布局为无符号的8位整数,要生成您想要获得的输出,请乘以255并转换为uint8
:
为此,只要在之前加入一行代码即可