From e832adfb032d6bae20ba53552cd3f0352148f698 Mon Sep 17 00:00:00 2001 From: Chan Lee <150145948+ChanLumerico@users.noreply.github.com> Date: Sun, 8 Sep 2024 18:59:29 +0900 Subject: [PATCH] SE-ResNet -> neural/README.md --- luma/neural/README.md | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/luma/neural/README.md b/luma/neural/README.md index e8a0240..f82b9d9 100644 --- a/luma/neural/README.md +++ b/luma/neural/README.md @@ -117,6 +117,7 @@ Deep learning models and neural network utilities of Luma | `ResNetBlock.Basic` | 7~ | $(N,C_{in},H_{in},W_{in})$ | $(N,C_{out},H_{out},W_{out})$ | | `ResNetBlock.Bottleneck` | 10~ | $(N,C_{in},H_{in},W_{in})$ | $(N,C_{out},H_{out},W_{out})$ | | `ResNetBlock.PreActBottleneck` | 10~ | $(N,C_{in},H_{in},W_{in})$ | $(N,C_{out},H_{out},W_{out})$ | +| `Bottleneck_SE` | 16~ | $(N,C_{in},H_{in},W_{in})$ | $(N,C_{out},H_{out},W_{out})$ | ### Xception Blocks @@ -126,6 +127,14 @@ Deep learning models and neural network utilities of Luma | `XceptionBlock.Middle` | 14 | $(N,728,19,19)$ | $(N,728,19,19)$ | | `XceptionBlock.Exit` | 11 | $(N,728,19,19)$ | $(N,1024,9,9)$ | +### SE Blocks + +| Class | # of Layers | Input Shape | Output Shape | +| --- | --- | --- | --- | +| `SEBlock1D` | 6 | $(N,C,W)$ | $(N,C,W)~\text{or}~(N,C)$ | +| `SEBlock2D` | 6 | $(N,C,H,W)$ | $(N,C,H,W)~\text{or}~(N,C)$ | +| `SEBlock3D` | 6 | $(N,C,D,H,W)$ | $(N,C,D,H,W)~\text{or}~(N,C)$ | + ### MobileNet Blocks | Class | # of Layers | Input Shape | Output Shape | @@ -258,6 +267,14 @@ Information Processing Systems, 2012. ### SENet Series +> Hu, Jie, et al. “Squeeze-and-Excitation Networks.” Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2018, pp. 7132-7141. +> + +| Class | # of Layers | Input Shape | Weights | Biases | Total Param. | Implemented | +| --- | --- | --- | --- | --- | --- | --- | +| `SE_ResNet` | 263 | $(N,3,224,224)$ | 35,615,808 | 46,440 | 35,662,248 | ✅ | +| `SE_InceptionNet` | | | | | | 🚧 | + *Waiting for future updates…🔮* ---