Pytorch:CNN理论基础

目录

一、图象基础知识

1、图像的构成

2、图像的分类

3、拓展知识

4、图像的加载方法

二、卷积神经网络概述

1、什么是卷积神经网络?

2、卷积神经网络的组成

3、CNN概述

三、卷积层

1、卷积计算

2、padding(填充)

3、stride(步长)

4、多通道卷积计算

5、多卷积核计算

6、特征图大小

7、卷积层API

四、池化层

1、池化层的作用

2、池化分类

3、池化层API


CNN = 卷积神经网络(Convolutional Neural Network)

一、图象基础知识

1、图像的构成

由像素点构成

2、图像的分类
  1. 灰度图像(单通道)
  • 只有 1 个通道
  • 每个像素:0~255
    • 0 = 纯黑
    • 255 = 纯白
  • 形状:[H,W] 高 × 宽
  1. 彩色图像(RGB 三通道)
  • R 红、G 绿、B 蓝 3 个通道
  • 每个通道像素值:0~255
  • 形状:[H,W,C] 高 × 宽 × 通道数
3、拓展知识
  • PyTorch 格式:[B,C,H,W] 批次、通道、高、宽
  • 黑白图不是只有黑色白色,是深浅灰色组成
  • 彩色图 = 三张灰度图叠加(R/G/B 各一张)

总结速记版:

图片由像素组成,灰度单通道,RGB 三通道,数值 0 到 255,矩阵存储。

4、图像的加载方法

plt.imread()

plt.imshow()

二、卷积神经网络概述

1、什么是卷积神经网络?

**卷积神经网络(Convolutional Neural Network)**是含有卷积层的神经网络. 卷积层的作用就是用来自动学习、提取图像的特征。

2、卷积神经网络的组成

CNN网络主要由三部分构成:卷积层、池化层和全连接层构成:

卷积层负责提取图像中的局部特征;

池化层用来大幅降低参数量级(降维);

全连接层用来输出想要的结果。

3、CNN概述

CNN要做的事情是:

给定一张图片,是车还是马未知,是什么车也未知,现在需要模型判断这张图片里具体是一个什么东西,总之输出一个结果:如果是车 那是什么车

最左边是 数据输入层:对数据做一些处理,比如去均值(各维度都减对应维度的均值,使得输入数据各个维度都中心化为0,避免数据过多偏差,影响训练效果)、归一化(把所有的数据都归一到同样的范围)、PCA等等。CNN只对训练集做"去均值"这一步。

中间是 卷积层(CONV):线性乘积求和,提取图像中的局部特征

激励层(RELU):ReLU激活函数,输入数据转换成输出数据

池化层(POOL):取区域平均值或最大值,大幅降低参数量级(降维) 最右边是

全连接层(FC):输出CNN模型预测结果

功能总结:

卷积层:特征提取

池化层:降维

全连接层:得到输出结果

三、卷积层

1、卷积计算

输入通过滤波器得到输出:

计算一下参数用二维表示:

1\]\[1\]=4 = 1\*1 +1\*0 +1\*1 + 0\*0+1\*1 +1+0 +0\*1 +0\*0 +1+1=4 ![](https://i-blog.csdnimg.cn/direct/298f9d94cee44de7a133a63e6594e5d9.png) ![](https://i-blog.csdnimg.cn/direct/eec997bb8940485f94d89a2361951848.png) \[2\]\[2\]=4= 1\*1 + 1\*0 + 1+1 +0\*1 +1\*1 +1\*0 + 0\*1 + 1\*0 +1\*1 =4 可以根据示意图计算一下其他位置的数据验证。 ###### 2、padding(填充) padding(填充)操作用于处理卷积时图像边缘的像素。 其目的是在输入图像的边界周围添加额外的像素(通常是零),从而解决卷积操作时边缘信息丢失的问题。 ###### 3、stride(步长) stride(步长)指的是卷积核在图像上滑动时的步伐大小,即每次卷积时卷积核在图像中向右(或向下)移动的像素数。步长直接影响卷积操作后输出特征图的尺寸,以及计算量和模型的特征提取能力。 ###### 4、多通道卷积计算 ![](https://i-blog.csdnimg.cn/direct/b022273385f34913842048f3fdfa0a45.png) ###### 5、多卷积核计算 ![](https://i-blog.csdnimg.cn/direct/8d089903b0634ae0b06b2b2546b73eb1.png) ![](https://i-blog.csdnimg.cn/direct/b2528c5ce37e4cf7b8ccf7d098562155.png) ###### 6、特征图大小 输出特征图的大小与以下参数息息相关: size: 卷积核/过滤器大小,一般会选择为奇数,比如有 1\*1 、3\*3、5\*5 Padding: 零填充的方式 Stride: 步长 那计算方法如下图所示: 输入图像大小: W x W 卷积核大小: F x F Stride: S Padding: P 输出图像大小: N x N 计算公式: ![](https://i-blog.csdnimg.cn/direct/5a8d876d77594c4ba29aa16a61f0c218.png) 计算示例: > 以下图为例: > > 图像大小: 5 x 5 > > 卷积核大小: 3 x 3 > > Stride: 1 > > Padding: 1 > > (5 - 3 + 2) / 1 + 1 = 5, 即得到的特征图大小为: 5 x 5 ![](https://i-blog.csdnimg.cn/direct/462a2e66bf7e48d9a889948f8d4a8a54.png) ###### 7、卷积层API ```python conv = nn.Conv2d(in_channels, out_channels, kernel_size, stride, padding) """ 参数说明: in_channels: 输入通道数, out_channels: 输出通道,也可以理解为卷积核kernel的数量 kernel_size:卷积核的高和宽设置,一般为3,5,7... stride:卷积核移动的步长 padding:在四周加入padding的数量,默认补0 """ ``` #### 四、池化层 ###### 1、池化层的作用 池化层 (Pooling) 降低维度, 从而减少计算量、减少内存消耗,并提高模型的鲁棒性。 ###### 2、池化分类 (1) 最大池化 (2)平均池化 ![](https://i-blog.csdnimg.cn/direct/7fc8d8075527408f91d4c3cabe74c9dd.png) ###### 3、池化层API ```python # 最大池化 nn.MaxPool2d(kernel_size=2, stride=2, padding=1) # 平均池化 nn.AvgPool2d(kernel_size=2, stride=1, padding=0) ```

相关推荐
一点一木2 小时前
深度体验TRAE SOLO移动端7天:作为独立开发者,我把工作流揣进了兜里
前端·人工智能·trae
Lee川3 小时前
mini-cursor 揭秘:从 Tool 定义到 Agent 循环的完整实现
前端·人工智能·后端
weelinking3 小时前
【产品】00_产品经理用Claude实现产品系列介绍
数据库·人工智能·sql·数据挖掘·github·产品经理
Agent产品评测局3 小时前
制造业模具管理AI系统,主流产品能力对比详解:2026年智能制造选型深度洞察
人工智能·ai·chatgpt·制造
研华科技Advantech4 小时前
如何用一套实训设备,打通工业AI预测性维护技术全流程?
人工智能
Lab_AI4 小时前
AI for Science: MaXFlow AI Agent+ 报告体验双升级,让AI智能体更高效易用!
人工智能·ai for science·ai agent·ai智能体
李坤4 小时前
让 Codex 和 Claude 互相 Review:告别手动复制
人工智能·openai·claude
南屹川4 小时前
【API设计】GraphQL实战:从REST到GraphQL的演进
人工智能
KJ_BioMed4 小时前
当计算生物学遇上生成式AI:从头设计生物分子的“新范式”初探
人工智能·从头设计·生命科学·生物医药·科研干货·科晶生物
明月醉窗台4 小时前
深度学习(17)YOLO训练中的超参数详解
人工智能·深度学习·yolo