Skip to content

Commit 7cbeee8

Browse files
authored
Update README.md
1 parent aad5eb8 commit 7cbeee8

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -207,6 +207,8 @@ x = squeeze_excitation(x, channels=64, ratio=16, use_bias=True, sn=True, scope='
207207
x = convolution_block_attention(x, channels=64, ratio=16, use_bias=True, sn=True, scope='convolution_block_attention')
208208

209209
x = global_context_block(x, channels=64, use_bias=True, sn=True, scope='gc_block')
210+
211+
x = srm_block(x, channels=64, use_bias=False, is_training=is_training, scope='srm_block')
210212
```
211213

212214
<div align="center">
@@ -237,6 +239,12 @@ x = global_context_block(x, channels=64, use_bias=True, sn=True, scope='gc_block
237239

238240
---
239241

242+
<div align="center">
243+
<img src=./assets/srm.png height='450' width='600>
244+
</div>
245+
246+
---
247+
240248
## Normalization
241249
```python
242250
x = batch_norm(x, is_training=is_training, scope='batch_norm')

0 commit comments

Comments
 (0)