百科.dev
全部条目在线工具AI 编程趋势榜开源项目技术资讯提交条目
登录
< 返回工具列表
C

CBAM.PyTorch

> 编程语言
开源

Paper 的非官方实现:CBAM:卷积块注意力模块

1.5K stars0 点赞1 次浏览
访问官网GitHub

工具介绍

Paper 的非官方实现:CBAM:卷积块注意力模块

CBAM.PyTorch

Paper 的非官方实现:CBAM:卷积块注意力模块

简介

代码是 PyTorch 重新实现的论文版本: CBAM: Convolutional Block Attention Module

Woo S, Park J, Lee J Y 等. CBAM: Convolutional Block Attention Module[J]. 2018. ECCV2018

结构

CBAM 概述。该模块包含两个顺序的子模块: 通道和空间。中间特征图通过自适应的细化来提高 我们的模块 (CBAM) 会在深度网络的每个卷积块中运行。

要求

  • Python3
  • PyTorch 0.4.1
  • tensorboardX(可选)
  • torchnet
  • pretrainedmodels(可选)

结果

我们在 ImageNet-1K 上测试了四个模型,训练集和验证集均被缩放到 256(最小边长),只使用 Mirror 和 RandomResizeCrop 作为训练数据增强,在验证过程中,我们使用中心裁剪获得 224x224 的图片片段。

Issues· 10 开放

查看全部 Issues在 GitHub 打开

暂无开放 Issues,或尚未同步最近议题。

> 标签

Python

暂无评论,来聊聊你的看法吧

> 工具信息

发布日期2026年8月1日
最后更新2026年9月17日
分类编程语言
定价开源

> 相关工具

T
TypeScript
JavaScript 的超集,为前端与全栈提供静态类型
P
Python
通用编程语言,广泛用于 Web、数据与 AI
G
Go
Google 推出的简洁高效系统语言