Skip to content

Commit fa96d6e

Browse files
committed
whenxuan: update the readme
1 parent d6df172 commit fa96d6e

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

README.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@ A plug-and-play channel attention mechanism module implemented in PyTorch.
55
<div align="center">
66

77
[Installation](#Installation) | [Usage](#Usage) | [Modules](#Modules) | [Blog](https://mp.weixin.qq.com/s/D6O5SBl2RYHdkiinV6UM8w)
8-
98
</div>
109

1110
<div align="center">
@@ -35,6 +34,11 @@ We only develop and test with PyTorch. Please make sure to install it from [PyTo
3534

3635

3736

37+
1. [`SEAttention`](https://github.com/wwhenxuan/Channel-Attention/blob/main/channel_attention/squeeze_excitation.py): [[paper]]() The Squeeze-and-Excitation Attention.
38+
<img width="100%" align="middle" src="images/SEAttention.png">
39+
40+
41+
3842
| Attention | Module Implementation Architecture |
3943
| ------------------------------------------------------------ | ------------------------------------------- |
4044
| <div style="text-align: center;"> SEAttention<br/>[[paper]]()</div> | ![SEAttention](images/SEAttention.png) |

images/ID2_OD2.jpg

312 KB
Loading

0 commit comments

Comments
 (0)