神经网络输出像素值溢出的问题

用pytorch在测试训练完成或未完成的网络时，得到输出y以后将其保存为图片的形式。首先得到数据类型为tensor类型，将其转化为np.array后，可能会存在像素值溢出的问题。即y.max() > 256。

 

这种溢出问题会造成江输出保存成图片时，下列情形

                                                       

如上图，正常情形应为左图所示，如若像素值溢出，则会导致右图情形。

问题解决：

应将输出的y的值限制在0到256这个区间内，进行下列操作

y = ( y - y.min() ) / ( y.max() - y.min() ) # 将输出y进行归一化
y = y * 256 # 将y值限制在0到256的范围内

便可以解决像素值溢出的问题