人工智能学习pyTorch的ResNet残差模块示例详解
作者:Swayzzu 时间:2022-05-04 21:45:22
1.定义ResNet残差模块
一个block中,有两个卷积层,之后的输出还要和输入进行相加。因此一个block的前向流程如下:
输入x→卷积层→数据标准化→ReLU→卷积层→数据标准化→数据和x相加→ReLU→输出out
中间加上了数据的标准化(通过nn.BatchNorm2d实现),可以使得效果更好一些。
①各层的定义
②前向传播
在前向传播中输入x,过程中根据前向流程,调用上面定义的层即可。
如此,便定义好了一个残差的模块。
2.ResNet18的实现
此处的ResNet18并没有涉及到太多细节,只是一个大致的内容。
在初始化的时候,定义好所需要使用的模块,根据上面定义好的残差模块,调用即可。在这里使用了4次残差模块,将通道数从输入的3(也就是RGB),变成了512通道。也就是最终提取的高级特征。提取完特征,直接输入给Linear方法,得到图片在10种图片类型上的结果,用于预测以及损失值的求解。
①各层的定义
②前向传播
同样的,调用上面定义好的层,根据流程传播即可。
3.测试ResNet18
如下图,输入的是2张图,RGB,长宽均是32的数据。
通过ResNet18之后,输出的是2张图,每张图对应10种类型的不同取值logits。
过程中的通道数目的转换如下面的结果所示:3→64→128→256→512→512
过程中的尺寸转换,根据设置的卷积核,以及步长,会有不同的结果。但总体都是为了得到更高级的特征,最终输入全连层,得到Logits。
来源:https://blog.csdn.net/Swayzzu/article/details/121156605
标签:pyTorch,ResNet,人工智能,残差模块
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
在SQL Server中编写通用数据访问方法
2009-01-20 11:35:00
在SQL server2005数据库下创建计划任务
2008-12-26 09:19:00
![](https://img.aspxhome.com/file/UploadPic/200812/20081231122259529.jpg)
python实现登录密码重置简易操作代码
2022-12-08 13:34:43
Pandas 对Dataframe结构排序的实现方法
2021-11-26 09:06:48
python之文件读取一行一行的方法
2022-04-09 20:21:19
使用pycharm将自己项目代码上传github(小白教程)
2022-05-02 20:37:06
![](https://img.aspxhome.com/file/2023/6/82526_0s.jpg)
使用Django启动命令行及执行脚本的方法
2022-03-08 22:10:16
PHP7新特性之抽象语法树(AST)带来的变化详解
2023-07-18 14:42:53
XML正在接管Web服务 成为SOA的基础
2008-09-05 17:21:00
一劳永逸彻底解决pip install慢的办法
2023-08-10 20:30:52
python实现图片文件批量重命名
2023-08-10 03:39:57
![](https://img.aspxhome.com/file/2023/5/62265_0s.jpg)
python中super()函数的理解与基本使用
2023-07-02 08:00:33
![](https://img.aspxhome.com/file/2023/0/65520_0s.png)
Python实现图形用户界面计算器
2022-09-02 06:07:22
![](https://img.aspxhome.com/file/2023/7/70257_0s.jpg)
python 计算t分布的双侧置信区间
2023-08-01 03:06:05
![](https://img.aspxhome.com/file/2023/9/61489_0s.jpg)
闲聊html和body标签
2009-02-21 10:50:00
Python translator使用实例
2021-02-09 22:40:24
Django 使用easy_thumbnails压缩上传的图片方法
2023-11-10 07:47:38
Microsoft SQL Server 2008 正式发布
2008-08-07 12:32:00
![](https://img.aspxhome.com/file/UploadPic/20088/7/200887123422184s.gif)
asp如何从数据库中删除废旧的电子信箱地址?
2009-11-15 20:04:00
用javascript来实现仿gogle动画导航
2007-11-30 14:15:00
![](https://img.aspxhome.com/file/UploadPic/200711/30/20071130141752445s.gif)