人工智能学习pyTorch的ResNet残差模块示例详解
更新时间：2021年11月11日 14:46:47 作者：Swayzzu
这篇文章主要为大家介绍了人工智能学习pyTorch的ResNet残差模块示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步
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。
以上就是人工智能学习pyTorch的ResNet残差模块示例详解的详细内容，更多关于PyTorch人工智能学习ResNet残差模块的资料请关注脚本之家其它相关文章！
相关文章
Python入门开发教程 windows下搭建开发环境vscode的步骤详解
大家都知道Python是跨平台的，它可以运行在Windows、Mac和各种Linux/Unix系统上。在Windows上写Python程序，放到Linux上也是能够运行的，今天给大家分享Python开发环境搭建vscode的步骤，一起看看吧2021-07-07
这篇文章主要介绍了Python中的字典与成员运算符初步探究,是Python入门学习中的基础知识,需要的朋友可以参考下2015-10-10
大家好，本篇文章主要讲的是用Python爬取2022春节档电影信息，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下2022-01-01
这篇文章主要介绍了Python实现子类调用父类的初始化实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-03-03
这篇文章主要介绍了Python实现给普通PDF添加水印的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-04-04
这篇文章主要介绍了Jmeter调用Python脚本实现参数互相传递的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-01-01
这篇文章主要介绍了Python中is和==的区别详解，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-11-11
这篇文章主要介绍了python中循环语句while用法,实例分析了while语句的使用方法,需要的朋友可以参考下2015-05-05
Python中playwright启动浏览器与常见运行方式详解
Playwright是一个功能强大的工具,可以帮助开发人员自动化测试、网页截图、信息提取等任务,本文主要介绍了如何使用Playwright来启动浏览器,感兴趣的可以了解下2024-05-05
sqlite3模块是Python中的内置模块,用于与SQLite数据库交互,本文就来介绍一下Python的sqlite3模块中常用函数,感兴趣的可以了解一下2023-10-10
最新评论