whenxuan: update the images

Author: wwhenxuan

.gitignore

@@ -205,3 +205,6 @@ cython_debug/
 marimo/_static/
 marimo/_lsp/
 __marimo__/
+
+./tests.ipynb
+.ipynb

README.md

@@ -11,3 +11,19 @@ pip install channel-attention
 
 We only develop and test with PyTorch. Please make sure to install it from [PyTorch official website](https://pytorch.org/get-started/locally/) based on your system configuration.
 
+## Usage
+
+
+
+
+
+## Modules
+
+
+
+| Attention                                                    | Module Implementation Architecture          |
+| ------------------------------------------------------------ | ------------------------------------------- |
+| <div style="text-align: center;">  SEAttention<br/>[[paper]]()</div> | ![SEAttention](images/SEAttention.png)      |
+| <div style="text-align: center;">  ChannelAttention<br/>[[paper]]()</div> | ![SEAttention](images/ChannelAttention.png) |
+| <div style="text-align: center;">  ChannelAttention<br/>[[paper]]()</div> |                                             |
+

channel_attention/__init__.py

@@ -21,5 +21,3 @@
 from .efficient_channel_attention import EfficientChannelAttention
 
 from .local_attention import LocalAttention
-
-from .frequency_enhanced_channel_attention import FreqEnhancedAttention