pytorch 获取神经网络中间图像Variable to img

最新推荐文章于 2025-06-19 16:21:51 发布

原创最新推荐文章于 2025-06-19 16:21:51 发布 · 1.1k 阅读

1 ·

CC 4.0 BY-SA版权

机器学习专栏收录该内容

3 篇文章

订阅专栏

本文介绍如何在PyTorch中将Variable类型的数据转化为图片输出，通过定义一个名为tensor_to_PIL的函数实现这一过程。该函数首先将Variable类型的张量转化为普通的tensor，然后逐个保存为图片。

部署运行你感兴趣的模型镜像

pytorch中使用Variable 转化为图片输出，

import os
import torch.nn as nn
import torch.optim as optim
from base_networks import *
from torchvision.transforms import *

def tensor_to_PIL(variable):                           #新加函数将Variable转换为图片
    image = variable.cpu().clone()
    image=image.data      #Variable张量转化为tensor
    for i in range(len(image)):
        save_file="./record/process_img/up4/"+str(i)+".jpg"
        temp_image=image[i].clone()
        temp_image = temp_image.unsqueeze(0)                  #压缩数组维数，去掉axis=''的维，0代表None 去掉一维的
        temp_image = transforms.ToPILImage()(temp_image.float()).convert("L")  #ToPILImage 需要一个三维的图像
        temp_image.save(save_file,quality=95)

您可能感兴趣的与本文相关的镜像