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画环形图的方法
这篇文章主要为大家详细介绍了python画环形图的相关代码，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2020-03-03
Dlib+OpenCV深度学习人脸识别的方法示例
这篇文章主要介绍了Dlib+OpenCV深度学习人脸识别的方法示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-05-05
Django读取Mysql数据并显示在前端的实例
今天小编就为大家分享一篇Django读取Mysql数据并显示在前端的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-05-05
python UNIX_TIMESTAMP时间处理方法分析
这篇文章主要介绍了python UNIX_TIMESTAMP时间处理方法,结合实例形式分析了Python针对UNIX_TIMESTAMP时间的常见运算技巧,需要的朋友可以参考下
2016-04-04
Pandas分组函数groupby的用法详解
在数据分析时,经常需要将数据分成不同的群组,pandas中的groupby()函数可以完美地完成各种分组操作,本文就来介绍一下Pandas分组函数groupby的用法,感兴趣的可以了解一下
2024-01-01
一文详解Python灰色预测模型实现示例
这篇文章主要为大家介绍了Python灰色预测模型实现示例解析，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2023-02-02
Python爬虫的两套解析方法和四种爬虫实现过程
本文想针对某一网页对 python 基础爬虫的两大解析库（ BeautifulSoup 和 lxml ）和几种信息提取实现方法进行分析，及同一网页爬虫的四种实现方式，需要的朋友参考下吧
2018-07-07
Python排序算法快速排序VS归并排序深入对比分析
快速排序和归并排序是两种常见的排序算法,在Python中有着重要的应用,本文将深入探讨这两种算法的原理和实现,并提供丰富的示例代码来说明它们的工作方式
2024-01-01
Python pandas删除指定行/列数据的方法实例
这篇文章主要给大家介绍了关于Python pandas删除指定行/列数据的相关资料,文中通过实例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2022-01-01
详解OpenCV中简单的鼠标事件处理
谈及鼠标事件，就是在触发鼠标按钮后程序所做出相应的反应，但是不影响程序的整个线程。本文将主要介绍OpenCV中的简单鼠标事件处理，感兴趣的可以学习一下
2022-01-01