将tensor张量转换成图片格式并保存

最新推荐文章于 2023-07-30 14:20:40 发布

码龄为零的程序员

最新推荐文章于 2023-07-30 14:20:40 发布

阅读量1.2w

点赞数 5

分类专栏： Pytorch 文章标签： numpy 深度学习 pytorch 神经网络机器学习

本文链接：https://blog.csdn.net/manzubaolong/article/details/109241086

版权

该博客介绍了一个工具，用于将经过transforms.Normalize和transforms.ToTensor处理的Tensor图像转换回.jpg或.png格式，适用于深度学习和机器学习中的神经网络模型。

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

这是一个工具包，功能：反向操作transforms.Normalize和transforms.ToTensor函数，将tensor格式的图片转换成.jpg、.png格式的图片

import torch
import numpy as np
from PIL import Image
import torchvision.transforms as transforms

def transform_convert(img_tensor, transform):
    """
    param img_tensor: tensor
    param transforms: torchvision.transforms
    """
    if 'Normalize' in str(transform):
        normal_transform = list(filter(lambda x: isinstance(x, transforms.Normalize), transform.transforms))
        mean = torch