Pytorch 保存模型生成图片方式

yipeiwu_com6年前Python基础

三通道数组转成彩色图片

    img=np.array(img1)

    img=img.reshape(3,img1.shape[2],img1.shape[3])

    img=(img+0.5)*255##img做过归一化处理,【-0.5,0.5】

    img_path='/home/isee/wei/image/imageset/result.jpg'

    img=cv2.merge(img)

    cv2.imwrite(img_path,img)

单通道数组转化成灰度图

    Img_mask=np.array(img_mask)*255 ##img_mask为二值图

    Img_mask=cv2.merge(img_mask)

    cv2.imwrite(img_path,img_mask)

以上这篇Pytorch 保存模型生成图片方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

用Python将动态GIF图片倒放播放的方法

用Python将动态GIF图片倒放播放的方法

这次让我们一个用 Python 做一个小工具:将动态 GIF 图片倒序播放! GIF(Graphics Interchange Format) 是一种可以用来呈现动画效果的图片格式,原...

Python实现读取目录所有文件的文件名并保存到txt文件代码

代码: (使用os.listdir) 复制代码 代码如下: import os def ListFilesToTxt(dir,file,wildcard,recursion): &nb...

Sanic框架配置操作分析

本文实例讲述了Sanic框架配置操作。分享给大家供大家参考,具体如下: 简介 Sanic是一个类似Flask的Python 3.5+ Web服务器,它的写入速度非常快。除了Flask之外...

Python3通过Luhn算法快速验证信用卡卡号的方法

本文实例讲述了Python3通过Luhn算法快速验证信用卡卡号的方法。分享给大家供大家参考。具体分析如下: Python3通过Luhn算法快速验证信用卡卡号,python用起来就是爽,很...

详解使用 pyenv 管理多个版本 python 环境

 随着同时开发的项目越来越多,需要不停的在各个不同版本的 python 环境之间切换,所以想到了pyenv。以前一直使用的 virtualenv只能管理同一个 python 版...