保存图片:uint8类型转换
训练过程中用writer.add_image向tensorboardX添加图片,发现图片总是下面这种鬼样子:但是在前一行用import matplotlib.pyplot as plt保存的图片就没有问题。
经过查阅writer.add_image的源码,发现其要求输入的图片数据类型如下:The elements in img_tensor can either have values in [0, 1] (float32) or [0, 255] (uint8)
于是将图片利用np.array(img, dtype=‘uint8’)转换为uint8类型。之后发现tensorboard上变得正常了。