保存图片:uint8类型转换

训练过程中用writer.add_image向tensorboardX添加图片,发现图片总是下面这种鬼样子:
问题解决:writer.add_image添加的图片模糊,很多点点;原来是要uint8但是在前一行用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上变得正常了。
问题解决:writer.add_image添加的图片模糊,很多点点;原来是要uint8

相关文章:

  • 2021-05-24
  • 2021-07-03
  • 2022-12-23
  • 2021-12-05
  • 2021-08-05
  • 2022-12-23
  • 2021-11-09
  • 2022-12-23
猜你喜欢
  • 2021-10-10
  • 2022-12-23
  • 2022-02-06
  • 2021-10-20
  • 2022-01-15
  • 2022-01-01
  • 2022-12-23
相关资源
相似解决方案