将tensor转化为图像

原创已于 2022-04-01 22:51:49 修改 · 8.8k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#python

于 2022-04-01 22:51:06 首次发布

人工智能/机器学习/深度学习专栏收录该内容

15 篇文章

订阅专栏

这篇博客介绍了如何使用PyTorch加载并处理图像数据，将Tensor转换为numpy数组，然后利用OpenCV显示图像。通过示例代码展示了从.pt文件中加载数据和标签，将Tensor转换为适合OpenCV显示的格式，并使用cv2.imshow进行展示。

import numpy as np
import cv2
import torch

# opencv支持的图像数据是numpy格式，数据类型为uint8，而且像素值分布在[0,255]之间
data,labels = torch.load("F:/zmz/pycharm/Co-teaching-master/data/processed/test.pt")
print(data[0])
print(labels[0].data)
# 将tensor转化成numpy
img_numpy = data[0].numpy()
img = np.uint8(img_numpy)
cv2.imshow("7",img)
cv2.waitKey(0)

结果：