pytorch使用记录（五）关于tensor、PIL以及numpy转换的问题

最新推荐文章于 2024-09-04 18:35:59 发布

daydayjump

最新推荐文章于 2024-09-04 18:35:59 发布

阅读量6.8k

点赞数 6

分类专栏： pytorch学习与使用文章标签： tensor array PIL 类型转换

本文链接：https://blog.csdn.net/daydayjump/article/details/88808394

版权

本文记录了在使用PyTorch过程中遇到的Tensor、numpy.array和PIL Image格式转换问题。重点是理解数据形状CHW与HWC的区别，以及在tensorboardX中SummaryWriter.add_image()函数的使用。解决方法包括调整dataformats参数或转换数据形状，同时介绍了Tensor自身类型转换、与numpy.array间的转换，以及与PIL Image的转换操作。

摘要由CSDN通过智能技术生成

最近在运行程序的时候一直出现如下错误：

  File "/home/daydayjump/Glow/glow/trainer.py", line 172, in train
    self.writer.add_image("1_prob/{}".format(bi), plot_prob([y_pred[bi], y_true[bi]], ["pred", "true"]).cuda().float(), self.global_step)
  File "/home/daydayjump/anaconda3/lib/python3.7/site-packages/tensorboardX/writer.py", line 427, in add_image
    image(tag, img_tensor, dataformats=dataformats), global_step, walltime)
  File "/home/daydayjump/anaconda3/lib/python3.7/site-packages/tensorboardX/summary.py", line 216, in image
    image = make_image(tensor, rescale=rescale)
  File "/home/daydayjump/anaconda3/lib/python3.7/site-packages/tensorboardX/summary.py", line 254, in make_image
    image = Image.fromarray(tensor)
  File "/home/daydayjump/anaconda3/lib/python3.7/site-packages/PIL/Image.py", line 2517, in fromarray
    raise TypeError("Cannot handle this data type")
Typ