pytorch 彩色图像转灰度图像实例
更新时间：2020年01月13日 08:36:44 作者：zzw小凡
今天小编就为大家分享一篇pytorch 彩色图像转灰度图像实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
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 opencv医学处理的实现过程,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-05-05
这篇文章主要介绍了如何利用Python实现读取Excel中的图片的功能，文中的实现步骤讲解详细，对我们学习Python有一定帮助，需要的可以参考一下2022-01-01
本文详细介绍了在已安装Python的情况下如何正确安装和配置Anaconda环境,包括Python环境变量的配置、Anaconda的下载安装以及环境变量配置,文中通过代码就介绍的非常详细,需要的朋友可以参考下2024-11-11
下面小编就为大家分享一篇Python实现邮件的批量发送的示例代码，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-01-01
这篇文章主要介绍了python输出结果刷新及进度条的实现操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-07-07
这篇文章主要为大家详细介绍了如何在 Python 中获取和处理用户的连续输入,包括基本的输入方法、输入验证、异常处理以及一些常见的应用场景,感兴趣的可以了解下2024-03-03
这篇文章主要为大家分享如何利用Python中的Tkinter库制作一个简易的在线个性签名生成工具，文中的示例代码讲解详细，感兴趣的小伙伴可以了解一下2022-03-03
这篇文章主要介绍了python 引用传递和值传递详解(实参,形参)。具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
这篇文章主要介绍了pytorch查看模型weight与grad方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
这篇文章主要介绍了Python结合Window计划任务监测邮件的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-08-08
最新评论