问题解决:writer.add_image添加的图片模糊,很多点点;原来是要uint8

保存图片:uint8类型转换 训练过程中用writer.add_image向tensorboardX添加图片,发现图片总是下面这种鬼样子: 但是在前一行用import matplotlib.pyplot as plt保存的图片就没有问题。 经过查阅writer.add_image的源码,发现其要求输入的图片数据类型如下: The elements in img_tensor can either
相关文章
相关标签/搜索