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之正规地说一句话
Sep 28 Python
python实现带错误处理功能的远程文件读取方法
Apr 29 Python
Python 查看文件的编码格式方法
Dec 21 Python
如何使用 Pylint 来规范 Python 代码风格(来自IBM)
Apr 06 Python
python3写的简单本地文件上传服务器实例
Jun 04 Python
win7下python3.6安装配置方法图文教程
Jul 31 Python
使用Python横向合并excel文件的实例
Dec 11 Python
Python3爬虫学习之MySQL数据库存储爬取的信息详解
Dec 12 Python
python实现最大优先队列
Aug 29 Python
Pandas聚合运算和分组运算的实现示例
Oct 17 Python
Keras loss函数剖析
Jul 06 Python
Python 详解通过Scrapy框架实现爬取百度新冠疫情数据流程
Nov 11 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删除数组中特定元素的两种方法
2013/07/02 PHP
php获取从html表单传递数组的方法
2015/03/20 PHP
php使用cookie实现记住登录状态
2015/04/27 PHP
PHP区块查询实现方法分析
2018/05/12 PHP
php实现微信原生支付(扫码支付)功能
2018/05/30 PHP
JavaScript中Math对象方法使用概述
2014/01/02 Javascript
node.js中的events.EventEmitter.listenerCount方法使用说明
2014/12/08 Javascript
jquery ui resize 中border-box的bug修正
2015/04/26 Javascript
js控制网页前进和后退的方法
2015/06/08 Javascript
AngularJS ng-blur 指令详解及简单实例
2016/07/30 Javascript
详解JSON1:使用TSQL查询数据和更新JSON数据
2016/11/21 Javascript
微信小程序之ES6与事项助手的功能实现
2016/11/30 Javascript
canvas时钟效果
2017/02/16 Javascript
详解Vue方法与事件
2017/03/09 Javascript
vuex直接赋值的三种方法总结
2018/09/16 Javascript
javascript异常处理实现原理详解
2020/02/17 Javascript
Python RuntimeError: thread.__init__() not called解决方法
2015/04/28 Python
python使用分治法实现求解最大值的方法
2015/05/12 Python
Python中的with...as用法介绍
2015/05/28 Python
Python实现的简单线性回归算法实例分析
2018/12/26 Python
django2.0扩展用户字段示例
2019/02/13 Python
12个Python程序员面试必备问题与答案(小结)
2019/06/24 Python
numpy.transpose()实现数组的转置例子
2019/12/02 Python
Pycharm配置PyQt5环境的教程
2020/04/02 Python
Django models文件模型变更错误解决
2020/05/11 Python
python和js交互调用的方法
2020/06/23 Python
python 将html转换为pdf的几种方法
2020/12/29 Python
中国跨境电子商务网站:NewFrog
2018/03/10 全球购物
银行实习鉴定
2013/12/13 职场文书
邮政员工辞职信
2014/01/16 职场文书
《他得的红圈圈最多》教学反思
2014/04/24 职场文书
金融专业毕业生自荐信
2014/06/26 职场文书
违反交通安全法检讨书
2014/10/24 职场文书
小学新教师个人总结
2015/02/05 职场文书
2016元旦晚会主持词
2015/07/01 职场文书
2016年大学生社区服务活动总结
2016/04/06 职场文书