pytorch 彩色图像转灰度图像
2017-09-12 10:39
423 查看
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
相关文章推荐
- 图像边缘检测—sobel算子(灰度图像,彩色图像)
- 彩色图像与灰度图像之间的转换
- 二值图像、灰度图像、彩色图像
- 把彩色图像变成灰度图像
- 基于FPGA的彩色图像转灰度算法实现
- 【Matlab】将彩色图像转换为灰度图像
- -01-RGB彩色图像转换为灰度图像【ARM NEON加速】
- 双边滤波器在灰度和彩色图像处理中的应用
- RGB彩色图像与灰度图像转换 opencv实现 代码及分析
- 数字图像处理基础:教你如何区分单色图像、灰度图像、伪彩色图像、真彩色图像
- VTK修炼之道23:图像基本操作_灰度图像映射成伪彩色图像(查表法)
- 让网页图片变灰色将彩色图像变成灰度的三种方法
- 用计算机的随机数做成的图像(灰度和彩色
- matlab中二值图像、灰度图像、彩色图像、索引图像都是几维矩阵,区别是什么?
- 图形图像处理-之-彩色转化到灰度的速度优化
- 彩色模型和彩色图像转换为灰度图像
- 彩色图像转换为灰度图像
- 彩色图像转灰度图像
- 彩色图像与灰度图像的转换
- vc将彩色图像转换为灰度图像