在项目中遇到一个tensor格式的数据,要保存为图像,此文对转换过程通过示例分享,以记录学习过程和帮助大家遇到同类问题时使用。
import torch
import cv2
import numpy as np# 创建一个示例张量(tensor)
input_tensor = torch.randn((1, 480, 640))# 将张量转换为 NumPy 数组
np_array = input_tensor.detach().cpu().numpy()[0]# 使用 OpenCV 保存为图像
image = cv2.imwrite('output333.jpg', np_array)
print("成功保存图像!")
示例过程:
(1)先创建一个张量,此处可以换成自己的数据。
(2)然后.detach()来释放与该结果相关的缓存。
(3)再转为CPU格式,避免某些张量处于GPU状态。
(4)然后使用.numpy()转为numpy数据格式。
(5)[0]是数据提取,将(1,480,640)维度提取后为(480,640)。
(6)*255,是将像素值进行处理。
(7)使用opencv函数保存图像。
示例中,选择使用.detach()来释放与该结果相关的缓存。这样可以避免不必要的内存占用,提高内存使用效率。