Pytorch 保存模型生成图片方式
更新时间：2020年01月10日 15:50:26 作者：wzy_zju
今天小编就为大家分享一篇Pytorch 保存模型生成图片方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
三通道数组转成彩色图片
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的安装下载配置,Python是一门非常强大好用的语言,也有着易上手的特性,本文为入门教程,需要的朋友可以参考下2023-04-04
这篇文章主要介绍了关于Python中的海象运算符“:=”使用方法详解,海象运算符（walrus operator）是 Python 3.8 中引入的一种新的语法,需要的朋友可以参考下2023-04-04
这篇文章主要介绍了关于DataFrame中某列值的替换map(dict),具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-02-02
这篇文章主要介绍了Python面向对象的程序设计详情，面向对象的程序设计在Python中具有非常重要的地位，熟练的使用面向对象编程能够为我们的Python编程提供很多的便利之处，希望您阅读完本文后能够有所收获2022-01-01
这篇文章主要为大家详细介绍了如何利用Python实现24点游戏，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-12-12
这篇文章主要介绍了pytorch配置双显卡方式,使用双显卡跑代码,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-06-06
这篇文章主要讲解了Python爬虫的基础知识，文中相关知识讲解的非常详细，感兴趣的小伙伴可以参考参考2021-08-08
今天小编就为大家分享一篇python读取文件名并改名字的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
这篇文章主要介绍了Python colormap库的安装和使用详情，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-10-10
这篇文章主要介绍了如何基于python实现画不同品种的樱花树,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-01-01
最新评论