栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

卷积操作Conv2d详细解释以及案例

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

卷积操作Conv2d详细解释以及案例

重点:1.图片的通道数,卷积核的通道要一样。

           2.每个卷积核处理图片后,所有通道要相加,得到一个通道

一、单通道图片,单核处理:

卷积核也为单通道:【【1,0,1】

                                    【0,1,1】

                                    【1,0,1】】

二、多通道图片,单核处理:

 

三.3通道图片,2核处理,输出为2通道:

 

下面是用卷积层conv处理一个图片x的案例,卷积层可以理解为函数,本质是类

import torch

x = torch.randn(2,1 ,3,2)  (2张图片,通道数1单色,高3,宽2)

conv = torch.nn.Conv2d(1 ,8 ,(2,3) ) #此处创建一个卷积层。

输入通道数1,和上面x数据保持一致

 输出的通道  8,也就是核的个数,决定了输出数据的通道数,也可以理解为有8个核处理了图片输出8个矩阵。

y=conv(x)  #用卷积处理x返回y

y的shape为(2,8,?,?)  #还是2张,8通道,高,宽

下面是源代码可以测试:


#2d中所有数据一般都是4维度的

import torch

cov = torch.nn.Conv2d(3, 2, (3, 2), stride=1, padding=1, bias=False)

#(输入通道,核数,(高,宽),滑动步长,周围补0的个数)

x=torch.ones((2,3,2,2))  #(图片张数,通道,高,宽)  #输入数据

y=cov(x)

print(y)

print("输出形状:",y.shape)   #(张数,通道,高,宽) 这个通道与cov的核数一致

print("权重形状",cov.weight.shape)

#总结:x 有2张图片,处理后 y还是2张,y的通道是卷积处理的结果,取决于卷积层核的数量

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/724924.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号