[模型基础] 卷积神经网络
二维卷积神经网络在宽度和高度两个方向上移动,要求:卷积核的通道数与输入通道数一致;卷积核的个数与输出通道数一致。效果如下:torch.nn.Conv2d(in_channel, out_channel, kernel_size,stride,bias,padding)其中,in_channel, out_channel, kernel_size必须得写。kernel_size1=3, 表示卷积核大
·
二维卷积神经网络在宽度和高度两个方向上移动,要求:
- 卷积核的通道数与输入通道数一致;
- 卷积核的个数与输出通道数一致。
效果如下:
torch.nn.Conv2d(in_channel, out_channel, kernel_size,stride,bias,padding)
其中,in_channel, out_channel, kernel_size
必须得写。kernel_size1=3
, 表示卷积核大小为3×3,kernel_size=(3×5)
,表示卷积核大小为3×5。
卷积之后得到的width核height大小与stride和padding的值有关,bias为True或False。
卷积之后通常输入全连接层,得到多个分类。
全连接fc=torch.nn.Linear(in, out)
其中,in
是卷积之后所有维度拉长之后的维度总数,out
是希望得到的分类数。
参考:
- 《PyTorch深度学习实践》完结合集——P10 卷积神经网络(基础篇):https://www.bilibili.com/video/BV1Y7411d7Ys?p=10
- PyTorch 深度学习实践 第10讲:https://blog.csdn.net/bit452/article/details/109690712 (31-38分)

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。
更多推荐
所有评论(0)