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 保存模型生成图片方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Django和websocket实现简单的多人聊天的示例代码
本文主要介绍了使用Django和WebSocket实现一个简单的多人聊天应用,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2025-01-01
Python操作MongoDB的教程详解(插,查,改,排,删)
MongoDB是一个基于分布式文件存储的数据库。是一个介于关系数据库和非关系数据库之间的产品,是非关系数据库当中功能最丰富,最像关系数据库的。本文将详细和大家聊聊Python操作MongoDB的方法,需要的可以参考一下2022-09-09
Django rstful登陆认证并检查session是否过期代码实例
这篇文章主要介绍了Django rstful登陆认证并检查session是否过期代码实例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08


最新评论