SENet.mxnet: 挤压与激励网络（包括SE-ResNext、SE-Resnet、SE-Inception-v4等）
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
SENet.mxnet是实现挤压与激励机制的库，包含SE-ResNext、SE-Resnet和SE-Inception-v4等多种模型，用于提升神经网络性能。
SENet.mxnet 实现了挤压与激励网络（包括 SE-ResNext 18,50,101,152、SE-Resnet 和 SE-Inception-v4 及 SE-Inception-Resnet-v2）。这些架构如论文中所述，作者在 SENet 中应用此“挤压和激发”块，并赢得了 Imagenet 2017 分类任务。作者的 Caffe 实现可以在 GitHub 上找到。
以下是“挤压与激励”模块的图示：
SE-ResNet 模块的具体实现如下：
对于 SE-ResNext 50，其实现细节如表所示。
这是 MXNet 版本的实现方式。
我还借鉴了一些其他资源来完善这个项目。值得一提的是，在最后一个全连接层之前我添加了一个 dropout 层。
全部评论 (0)
还没有任何评论哟~
