
CBAM工具(PyTorch实现),为论文提供非官方支持。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
CBAM.PyTorch 论文:这些代码旨在为“CBAM:卷积块注意模块”这篇论文提供一个非官方的实现版本,由 Woo S,Park J,Lee JY 等人撰写。 CBAM 是一种卷积块注意模块,其结构包含两个紧密相连的子模块:通道和空间维度。 该模块通过对深度网络中每个卷积块的中间特征图进行自适应调整,从而提升性能。 为了运行此代码,需要 Python 3 以及 PyTorch 0.4.1 和 tensorboardX(作为可选依赖项)。此外,火炬网提供的预训练模型也可以使用(同样是可选)。 在实验中,我们仅在 ImageNet-1K 数据集上评估了四个不同的模型。 训练集和验证集都经过缩放到 256 像素(以最小边的尺寸为准),并利用 Mirror 和 RandomResizeCrop 技术进行数据扩充。 在验证阶段,我们采用中心裁剪方法生成 224x224 的图像补丁。 ImageNet-1K 数据集的楷模验证结果显示,Top-1 准确率达到了 74.26%,而前五名的准确率则为 91。
全部评论 (0)
还没有任何评论哟~


