Pytorch 保存模型生成图片方式

yipeiwu_com5年前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 pandas库中DataFrame对行和列的操作实例讲解

用pandas中的DataFrame时选取行或列: import numpy as np import pandas as pd from pandas import Sereis,...

Python 模拟动态产生字母验证码图片功能

Python 模拟动态产生字母验证码图片功能

模拟动态产生字母验证码图片 模拟生成验证码,首先要做的是生成随机的字母,然后对字母进行模糊处理。这里介绍一下 Python 提供的 Pillow 模块。 Pillow PIL:Pytho...

python如何让类支持比较运算

本文实例为大家分享了python类支持比较运算的具体代码,供大家参考,具体内容如下 案例:   有时我们希望自定义的类,实例间可以使用比较运算符进行比较,我们自定义比较的行为。   需求...

利用Python绘制有趣的万圣节南瓜怪效果

利用Python绘制有趣的万圣节南瓜怪效果

关于万圣节 万圣节又叫诸圣节,在每年的11月1日,是西方的传统节日;而万圣节前夜的10月31日是这个节日最热闹的时刻。在中文里,常常把万圣节前夜(Halloween)讹译为万圣节(All...

python多环境切换及pyenv使用过程详解

python多环境切换及pyenv使用过程详解

1.安装pyenv    https://github.com/pyenv/pyenv-installer curl -L https://g...