淘先锋技术网

首页 1 2 3 4 5 6 7

重点: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通道,高,宽

下面是源代码可以测试: