pytorch 彩色图像转灰度图像实例


Posted in Python onJanuary 13, 2020

pytorch 库

pytorch 本身具有载入cifar10等数据集的函数,但是载入的是3*200*200的张量,当碰到要使用灰度图像时,可以使用他本身的函数进行修改,以较快速的完成彩色图像转灰度图像

pytorch函数

dataset = dset.CIFAR10(root='../train/data', download=True,transform=transforms.Compose([
        transforms.Scale(200),
        transforms.ToTensor(),
        transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
       ]))

这里包含了对图像进行的一些处理,尺寸的控制,Normalize等

修改pytorch 库函数

找到transforms 的库函数位置,在后面添加 类 Gray()

class Gray(object)

 def __call__(self, tensor):
  # TODO: make efficient
  R = tensor[0]
  G = tensor[1]
  B = tensor[2]
  tensor[0]=0.299*R+0.587*G+0.114*B
  tensor = tensor[0]
  tensor = tensor.view(1,200,200)
  return tensor

tensor.view 这里使用是为了将tensor 做成 1*200*200的大小,否则出来是200*200

以上这篇pytorch 彩色图像转灰度图像实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python模拟登录百度贴吧(百度贴吧登录)实例
Dec 18 Python
Python实现获取操作系统版本信息方法
Apr 08 Python
初步认识Python中的列表与位运算符
Oct 12 Python
浅谈Python实现贪心算法与活动安排问题
Dec 19 Python
详解python中__name__的意义以及作用
Aug 07 Python
python实现126邮箱发送邮件
May 20 Python
python filecmp.dircmp实现递归比对两个目录的方法
May 22 Python
利用keras使用神经网络预测销量操作
Jul 07 Python
Python selenium如何打包静态网页并下载
Aug 12 Python
提高python代码运行效率的一些建议
Sep 29 Python
python批量生成条形码的示例
Oct 10 Python
Manjaro、pip、conda更换国内源的方法
Nov 17 Python
Ranorex通过Python将报告发送到邮箱的方法
Jan 12 #Python
python opencv实现信用卡的数字识别
Jan 12 #Python
Python 实现递归法解决迷宫问题的示例代码
Jan 12 #Python
Python3.x+迅雷x 自动下载高分电影的实现方法
Jan 12 #Python
tensorflow的计算图总结
Jan 12 #Python
python利用JMeter测试Tornado的多线程
Jan 12 #Python
Django 批量插入数据的实现方法
Jan 12 #Python
You might like
php.ini save_handler 修改不生效的解决办法
2014/07/22 PHP
php简单实现屏蔽指定ip段用户的访问
2015/04/29 PHP
Paypal实现循环扣款(订阅)功能
2017/03/23 PHP
php+iframe 实现上传文件功能示例
2020/03/04 PHP
动态调用CSS文件的JS代码
2010/07/29 Javascript
ExtJS下 Ext.Direct加载和提交过程排错小结
2013/04/02 Javascript
Javascript Object 对象学习笔记
2014/12/17 Javascript
快速掌握Node.js环境的安装与运行方法
2016/02/16 Javascript
javascript 解决浏览器不支持的问题
2016/09/24 Javascript
第一次接触神奇的Bootstrap
2016/10/14 Javascript
教你用十行node.js代码读取docx的文本
2017/03/08 Javascript
详解vue2.0组件通信各种情况总结与实例分析
2017/03/22 Javascript
vue.js在标签属性中插入变量参数的方法
2018/03/06 Javascript
解决LayUI数据表格复选框不居中显示的问题
2019/09/25 Javascript
JS实现简单随机3D骰子
2019/10/24 Javascript
Python基于Tkinter的HelloWorld入门实例
2015/06/17 Python
Python 中的lambda函数介绍
2018/10/10 Python
解决python ogr shp字段写入中文乱码的问题
2018/12/31 Python
python利用selenium进行浏览器爬虫
2019/04/25 Python
Django REST framework 分页的实现代码
2019/06/19 Python
Java文件与类动手动脑实例详解
2019/11/10 Python
在python下实现word2vec词向量训练与加载实例
2020/06/09 Python
Blancsom美国/加拿大:服装和生活用品供应商
2018/07/27 全球购物
日本最大的旅游网站:Rakuten Travel(乐天旅游)
2018/08/02 全球购物
信息专业毕业生五年职业规划参考
2014/02/06 职场文书
教学质量评估实施方案
2014/03/17 职场文书
竞选副班长演讲稿
2014/04/24 职场文书
离婚协议书范本及离婚须知
2014/10/15 职场文书
先进人物事迹材料
2014/12/29 职场文书
医生个人年终总结
2015/02/28 职场文书
会议室管理制度范本
2015/08/06 职场文书
小学校本教研总结
2015/08/13 职场文书
重阳节主题班会
2015/08/17 职场文书
2016党员干部廉政准则学习心得体会
2016/01/20 职场文书
环境保护宣传标语大全!
2019/06/28 职场文书
安装Ruby和 Rails的详细步骤
2022/04/19 Ruby