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 的 ultralytics 库功能及安装方法

    Python 的 ultralytics 库功能及安装方法

    ultralytics库由Ultralytics团队开发,旨在为YOLO系列模型提供高效、灵活且易于使用的工具,本文将详细介绍ultralytics库的功能、安装方法、核心模块以及使用示例,感兴趣的朋友一起看看吧
    2025-03-03
  • python自动化测试无法启动谷歌浏览器问题

    python自动化测试无法启动谷歌浏览器问题

    这篇文章主要介绍了python自动化测试无法启动谷歌浏览器问题,本文给大家介绍的非常详细,具有一定的参考借鉴价值,需要的朋友可以参考下
    2019-10-10
  • pyqt5 删除layout中的所有widget方法

    pyqt5 删除layout中的所有widget方法

    今天小编就为大家分享一篇pyqt5 删除layout中的所有widget方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-06-06
  • Django和websocket实现简单的多人聊天的示例代码

    Django和websocket实现简单的多人聊天的示例代码

    本文主要介绍了使用Django和WebSocket实现一个简单的多人聊天应用,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2025-01-01
  • Python操作MongoDB的教程详解(插,查,改,排,删)

    Python操作MongoDB的教程详解(插,查,改,排,删)

    MongoDB是一个基于分布式文件存储的数据库。是一个介于关系数据库和非关系数据库之间的产品,是非关系数据库当中功能最丰富,最像关系数据库的。本文将详细和大家聊聊Python操作MongoDB的方法,需要的可以参考一下
    2022-09-09
  • Python中变量交换的例子

    Python中变量交换的例子

    这篇文章主要介绍了Python中变量交换的例子,Python中的变量交换不需要临时变量,一个等号即可实现变量的交换,非常方便,需要的朋友可以参考下
    2014-08-08
  • python环境下安装opencv库的方法

    python环境下安装opencv库的方法

    这篇文章主要介绍了python环境下安装opencv库的方法 ,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-03-03
  • Django rstful登陆认证并检查session是否过期代码实例

    Django rstful登陆认证并检查session是否过期代码实例

    这篇文章主要介绍了Django rstful登陆认证并检查session是否过期代码实例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-08-08
  • Python 读取 Word 文档操作

    Python 读取 Word 文档操作

    这篇文章主要介绍了Python读取Word文档操作,文章围绕主题展开详细的内容介绍,具有一定的参考价值,需要的小伙伴可以参考一下
    2022-08-08
  • python创建堆的方法实例讲解

    python创建堆的方法实例讲解

    在本篇文章里小编给大家整理的是一篇关于python创建堆的方法实例讲解内容,有兴趣的朋友们可以学习下。
    2021-03-03

最新评论