pytorch初学笔记（九）：神经网络基本结构之卷积层

好喜欢吃红柚子

已于 2022-11-21 18:16:06 修改

阅读量3.1k

点赞数 1

分类专栏： Python和Pytorch学习文章标签： pytorch 神经网络深度学习

于 2022-11-21 18:09:34 首次发布

本文链接：https://blog.csdn.net/weixin_45662399/article/details/127951509

版权

Python和Pytorch学习专栏收录该内容

39 篇文章 226 订阅

订阅专栏

一、torch.nn.CONV2D

1.1 参数介绍

1.2 stride 和 padding 的可视化

一、torch.nn.CONV2D

官方文档：

torch.nn — PyTorch 1.13 documentation

1.1 参数介绍

常用的参数主要是前五个。

(31条消息) Pytorch中dilation（Conv2d）参数详解_MaZhe丶的博客-CSDN博客_conv2d参数解释

in_channels (int) – Number of channels in the input image，输入图片的通道数

out_channels (int) – Number of channels produced by the convolution，输出图片的通道数, 代表卷积核的个数，使用n个卷积核输出的特征矩阵深度即channel就是n

kernel_size (int or tuple) – Size of the convolving kernel，卷积核的大小

e.g. if kernel size = 3, 则卷积核的大小是3*3

stride (int or tuple, optional) – Stride of the convolution. Default: 1，步径大小

padding (int, tuple or str, optional) – Padding added to all four sides of the input. Default: 0

padding_mode (str, optional) – 'zeros', 'reflect', 'replicate' or 'circular'. Default: 'zeros'

dilation (int or tuple, optional) – Spacing between kernel elements. Default: 1

groups (int, optional) – Number of blocked connections from input channels to output channels. Default: 1

bias (bool, optional) – If True, adds a learnable bias to the output. Default: True

如何进行输出图片长和宽的计算？

1.2 stride 和 padding 的可视化

conv_arithmetic/README.md at master · vdumoulin/conv_arithmetic · GitHub

1.3 输入、输出通道数

(31条消息) 【卷积神经网络】多输入通道和多输出通道（channels）_ZSYL的博客-CSDN博客_多输出通道

(31条消息) （pytorch-深度学习系列）CNN的多输入通道和多输出通道_我是一颗棒棒糖的博客-CSDN博客

彩色图像在高和宽2个维度外还有RGB（红、绿、蓝）3个颜色通道。假设彩色图像的高和宽分别是h hh和w ww（像素），那么它可以表示为一个3 × h × w 的多维数组。我们将大小为3的这一维称为通道（channel）维。

1.3.1 多通道输入

1.3.2 多通道输出

多通道输出的通道数 = 卷积核的个数

二、卷积操作练习

2.1 数据集准备

使用cifar10数据集的验证集进行操作。

import torch
import torchvision.datasets
from torch.nn import Conv2d
from torch.utils.data import DataLoader
#数据集准备
dataset = torchvision.datasets.CIFAR10(root=".\CIFAR10",train=False,transform=torchvision.transforms.ToTensor(),download=True)
#使用dataloader加载数据集,批次数为64
dataloader = DataLoader(dataset,batch_size=64)

2.2 自定义神经网络

自定义名为Maweiyi的神经网络。
自定义的神经网络要继承nn.Module框架。
重写init和forward方法。
该神经网络调用conv2d进行一层卷积，输入通道为3层（彩色图像为3通道），卷积核大小为3*3，输出通道为6。
设置步长为1，padding为0，不进行填充。

import torch
import torchvision.datasets
from torch.nn import Conv2d
from torch.utils.data import DataLoader

class Maweiyi(torch.nn.Module):
    def __init__(self):
        super(Maweiyi, self).__init__()
        self.conv1 = Conv2d(in_channels=3,out_channels=6,kernel_size=3,stride=1,padding=0)

    def forward(self,x):
        x = self.conv1(x)
        return x

maweiyi = Maweiyi()
print(maweiyi)

输出：

2.3 卷积操作控制台输出结果

使用for循环输出结果。

# 输出卷积前的图片大小和卷积后的图片大小
for data in dataloader:
    imgs, labels = data
    print(imgs.shape)
    # 卷积操作
    outputs = maweiyi(imgs)
    print(outputs.shape)

输出：

2.4 tensorboard可视化

注意：使用tensorboard输出时需要重新定义图片大小

对于输入的图片集imgs来说，tensor.size([64,3,32,32])，即一批次为64张，一张图片为三个通道，大小为32*32
对于经过卷积后输出的图片集output来说，tensor.size([64,6,30,30])，通道数变成了6，tensorboard不知道怎么显示通道数为6的图片，所以如果直接输出会报错

解决方案：

使用reshape方法对outputs进行重定义，把通道数改成3，如果不知道批次数大小，可以使用-1代替，程序会自动匹配批次大小。

outputs = torch.reshape(outputs,(-1,3,30,30))

# 生成日志
writer = SummaryWriter("logs")
step = 0

# 输出卷积前的图片大小和卷积后的图片大小
for data in dataloader:
    imgs, labels = data
    # 显示输入的图片
    writer.add_images("inputs",imgs,step)

    #卷积操作
    outputs = maweiyi(imgs)

    #重定义输出图片的大小
    outputs = torch.reshape(outputs,(-1,3,30,30))
    # 显示输出的图片
    writer.add_images("outputs",outputs,step)
    step+=1

writer.close()

输出：

input：一个批次64张图片

output：一个批次128张（64*2=128）

三、完整代码

import torch
import torchvision
from torch.nn import Conv2d
from torch.utils.data import DataLoader
# 数据集下载
from torch.utils.tensorboard import SummaryWriter

dataset = torchvision.datasets.CIFAR10(root=".\CIFAR10", train=False, transform=torchvision.transforms.ToTensor(),
                                       download=True)
# 数据加载器
dataloader = DataLoader(dataset, batch_size=64)


class Maweiyi(torch.nn.Module):
    def __init__(self):
        super(Maweiyi, self).__init__()
        # 卷积层
        self.conv1 = Conv2d(in_channels=3, out_channels=6, kernel_size=3, stride=1, padding=0)

    def forward(self, x):
        x = self.conv1(x)
        return x


maweiyi = Maweiyi()
print(maweiyi)

writer = SummaryWriter("logs")
step = 0

for data in dataloader:
    imgs, labels = data
    # 卷积操作
    output = maweiyi(imgs)
    print(imgs.shape)
    print(output.shape)
    writer.add_images("input", imgs, step)

    output = torch.reshape(output, (-1, 3, 30, 30))
    writer.add_images("output", output, step)
    step = step + 1

writer.close()