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 过滤字符串的技巧,map与itertools.imap
Sep 06 Python
剖析Python的Tornado框架中session支持的实现代码
Aug 21 Python
各种Python库安装包下载地址与安装过程详细介绍(Windows版)
Nov 02 Python
python爬虫框架talonspider简单介绍
Jun 09 Python
Selenium的使用详解
Oct 19 Python
python实现字符串加密成纯数字
Mar 19 Python
Python应用领域和就业形势分析总结
May 14 Python
python实现比较类的两个instance(对象)是否相等的方法分析
Jun 26 Python
python实现通过flask和前端进行数据收发
Aug 22 Python
简单了解Python多态与属性运行原理
Jun 15 Python
Python下划线5种含义代码实例解析
Jul 10 Python
python神经网络ResNet50模型
May 06 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
150kHz到30Mhz完全冲浪手册
2020/03/20 无线电
关于时间计算的结总
2006/12/06 PHP
window+nginx+php环境配置 附配置搭配说明
2010/12/29 PHP
最常用的8款PHP调试工具
2014/07/06 PHP
php中Socket创建与监听实现方法
2015/01/05 PHP
CI框架(CodeIgniter)操作redis的方法详解
2018/01/25 PHP
PHP判断一个变量是否为整数、正整数的方法示例
2019/09/11 PHP
Laravel服务容器绑定的几种方法总结
2020/06/14 PHP
jQuery的写法不同导致的兼容性问题的解决方法
2010/07/29 Javascript
能说明你的Javascript技术很烂的五个原因分析
2011/10/28 Javascript
谷歌地图打不开的解决办法
2014/08/07 Javascript
JavaScript实现检查页面上的广告是否被AdBlock屏蔽了的方法
2014/11/03 Javascript
浅谈javascript中this在事件中的应用
2015/02/15 Javascript
javascript删除数组重复元素的方法汇总
2015/06/24 Javascript
JS实现文档加载完成后执行代码
2015/07/09 Javascript
基于jquery实现的树形菜单效果代码
2015/09/06 Javascript
微信小程序 MINA文件结构
2016/10/17 Javascript
CSS+jQuery实现简单的折叠菜单
2016/12/20 Javascript
JS实现的简单分页功能示例
2018/08/23 Javascript
vue elementUI table 自定义表头和行合并的实例代码
2019/05/22 Javascript
python实现爬虫下载美女图片
2015/07/14 Python
Python引用传值概念与用法实例小结
2017/10/07 Python
Python中list查询及所需时间计算操作示例
2018/06/21 Python
详解Django中六个常用的自定义装饰器
2018/07/04 Python
TensorFlow打印tensor值的实现方法
2018/07/27 Python
numpy 对矩阵中Nan的处理:采用平均值的方法
2018/10/30 Python
Python图像处理二值化方法实例汇总
2020/07/24 Python
html5实现canvas阴影效果示例
2014/05/07 HTML / CSS
英国优质家居用品网上品牌:URBANARA
2018/06/01 全球购物
主题实践活动总结
2014/05/08 职场文书
2014财务人员自我评价范文
2014/09/21 职场文书
党员个人剖析材料
2014/09/30 职场文书
行为规范主题班会
2015/08/13 职场文书
聊聊pytorch测试的时候为何要加上model.eval()
2021/05/23 Python
python opencv检测直线 cv2.HoughLinesP的实现
2021/06/18 Python
Python实现文字pdf转换图片pdf效果
2022/04/03 Python