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实现可以断点续传和并发的ftp程序

前言 下载文件时,最怕中途断线,无法成功下载完整的文件。断点续传就是从文件中断的地方接下去下载,而不必重新下载。这项功能对于下载较大文件时非常有用。那么这篇文章就来给大家分享如何利用py...

使用python开发vim插件及心得分享

使用python开发vim插件及心得分享

vim有各种强大的插件,这不仅归功于其提供的用来编写插件的脚本语言vimL,还得益于它良好的接口实现,从而支持python等语言编写插件。当vim编译时带有+python特性时就能使用p...

浅析Python多线程下的变量问题

在多线程环境下,每个线程都有自己的数据。一个线程使用自己的局部变量比使用全局变量好,因为局部变量只有线程自己能看见,不会影响其他线程,而全局变量的修改必须加锁。 但是局部变量也有问题,就...

Python实现求解一元二次方程的方法示例

本文实例讲述了Python实现求解一元二次方程的方法。分享给大家供大家参考,具体如下: 1. 引入math包 2. 定义返回的对象 3. 判断b*b-4ac的大小 具体计算代码如下:...

Python 网络编程之UDP发送接收数据功能示例【基于socket套接字】

本文实例讲述了Python 网络编程之UDP发送接收数据功能。分享给大家供大家参考,具体如下: demo.py(UDP发送数据): import socket # 导入socket模...