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

最新推荐文章于 2022-03-24 16:17:33 发布

菜鸟起飞闯天涯1233445

最新推荐文章于 2022-03-24 16:17:33 发布

阅读量1k

点赞数 2

分类专栏：机器学习

本文链接：https://blog.csdn.net/wangxiaai/article/details/85075223

版权

机器学习专栏收录该内容

3 篇文章 0 订阅

订阅专栏

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)