pytorch 彩色图像转灰度图像实例
作者:zzw小凡 时间:2023-08-02 17:28:37
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
来源:https://blog.csdn.net/zzw000000/article/details/77942406
标签:pytorch,彩色,图像,灰度
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
python 遍历可迭代对象的实现方法
2021-03-02 07:13:07
Laravel框架之blade模板新手入门教程及小技巧
2024-05-11 09:54:14
![](https://img.aspxhome.com/file/2023/3/125823_0s.png)
Python实例之wxpython中Frame使用方法
2022-10-16 00:09:07
Python3的介绍、安装和命令行的认识(推荐)
2022-10-09 22:45:36
![](https://img.aspxhome.com/file/2023/9/117939_0s.png)
在matplotlib中改变figure的布局和大小实例
2022-07-09 00:45:11
![](https://img.aspxhome.com/file/2023/8/121398_0s.jpg)
python numpy实现rolling滚动案例
2023-08-24 17:12:45
MySql完整卸载的四个步骤详解
2024-01-18 00:30:52
![](https://img.aspxhome.com/file/2023/0/80020_0s.jpg)
商业价值与用户价值的平衡
2008-12-10 18:42:00
![](https://img.aspxhome.com/file/UploadPic/200812/10/002_small-39s.jpg)
python爬虫智能翻页批量下载文件的实例详解
2023-06-20 14:17:12
Firefox 3.5 新增加的支持(整理)
2009-08-01 12:51:00
只需7行Python代码玩转微信自动聊天
2021-03-09 09:03:22
![](https://img.aspxhome.com/file/2023/9/93779_0s.png)
mysql双向加密解密方式用法详解
2024-01-15 05:55:02
MySQL中对于NULL值的理解和使用教程
2024-01-27 22:51:02
mysql免安装版配置步骤详解分享
2024-01-22 08:18:07
SQL_Server全文索引的用法解析
2024-01-17 08:37:14
asp.net(c#)实现从sqlserver存取二进制图片的代码
2023-06-26 21:48:03
MySQL数据库的触发器和事务
2024-01-15 21:35:08
Python实现绘制双柱状图并显示数值功能示例
2023-12-06 06:31:03
![](https://img.aspxhome.com/file/2023/4/101474_0s.png)
用Python抢火车票的简单小程序实现解析
2023-09-06 06:31:28
Python通过zookeeper实现分布式服务代码解析
2021-09-11 11:09:48