tensor转图像

最新推荐文章于 2024-01-16 04:45:00 发布

Echo_ac

最新推荐文章于 2024-01-16 04:45:00 发布

阅读量177

点赞数

文章标签： python opencv numpy

本文链接：https://blog.csdn.net/qq_45769627/article/details/129571851

版权

这段代码是一个将[-1,1]范围内的张量转换为OpenCV兼容图像的函数。它涉及到张量的规范化、类型转换以及利用PIL库创建图像。函数接受一个形状为(b,3,128,128)的张量，并返回一个图像列表。

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

代码来自我的师兄：

# [-1,1] b,3,128,128 to opencv image
def tensor2img(tensor):
    tensor = tensor.detach().cpu().numpy()
# clamp
    tensor = np.clip(tensor, -1, 1)
    tensor = (tensor + 1) / 2.0 * 255.0
    tensor = tensor.transpose(0, 2, 3, 1)
    tensor = tensor.astype(np.uint8)
    tensor = tensor[...,0]
    images = []
    for i in range(tensor.shape[0]):
        images.append(Image.fromarray(tensor[i]))
    return images