--- a +++ b/model-description/encoder-efficientnet-b5.txt @@ -0,0 +1,55 @@ +=============================================================================================== +Layer (type:depth-idx) Input Shape Output Shape +=============================================================================================== +EfficientNetEncoder -- -- +├─Conv2d: 1-1 [1, 3, 512, 512] [1, 48, 256, 256] +├─BatchNorm2d: 1-2 [1, 48, 256, 256] [1, 48, 256, 256] +├─Swish: 1-3 [1, 48, 256, 256] [1, 48, 256, 256] +├─Sequential: 1 -- -- +│ └─Sequential: 2-1 [1, 48, 256, 256] [1, 24, 256, 256] +│ │ └─DepthwiseSeparableConv: 3-1 [1, 48, 256, 256] [1, 24, 256, 256] +│ │ └─DepthwiseSeparableConv: 3-2 [1, 24, 256, 256] [1, 24, 256, 256] +│ │ └─DepthwiseSeparableConv: 3-3 [1, 24, 256, 256] [1, 24, 256, 256] +│ └─Sequential: 2-2 [1, 24, 256, 256] [1, 40, 128, 128] +│ │ └─InvertedResidual: 3-4 [1, 24, 256, 256] [1, 40, 128, 128] +│ │ └─InvertedResidual: 3-5 [1, 40, 128, 128] [1, 40, 128, 128] +│ │ └─InvertedResidual: 3-6 [1, 40, 128, 128] [1, 40, 128, 128] +│ │ └─InvertedResidual: 3-7 [1, 40, 128, 128] [1, 40, 128, 128] +│ │ └─InvertedResidual: 3-8 [1, 40, 128, 128] [1, 40, 128, 128] +│ └─Sequential: 2-3 [1, 40, 128, 128] [1, 64, 64, 64] +│ │ └─InvertedResidual: 3-9 [1, 40, 128, 128] [1, 64, 64, 64] +│ │ └─InvertedResidual: 3-10 [1, 64, 64, 64] [1, 64, 64, 64] +│ │ └─InvertedResidual: 3-11 [1, 64, 64, 64] [1, 64, 64, 64] +│ │ └─InvertedResidual: 3-12 [1, 64, 64, 64] [1, 64, 64, 64] +│ │ └─InvertedResidual: 3-13 [1, 64, 64, 64] [1, 64, 64, 64] +│ └─Sequential: 2-4 [1, 64, 64, 64] [1, 128, 32, 32] +│ │ └─InvertedResidual: 3-14 [1, 64, 64, 64] [1, 128, 32, 32] +│ │ └─InvertedResidual: 3-15 [1, 128, 32, 32] [1, 128, 32, 32] +│ │ └─InvertedResidual: 3-16 [1, 128, 32, 32] [1, 128, 32, 32] +│ │ └─InvertedResidual: 3-17 [1, 128, 32, 32] [1, 128, 32, 32] +│ │ └─InvertedResidual: 3-18 [1, 128, 32, 32] [1, 128, 32, 32] +│ │ └─InvertedResidual: 3-19 [1, 128, 32, 32] [1, 128, 32, 32] +│ │ └─InvertedResidual: 3-20 [1, 128, 32, 32] [1, 128, 32, 32] +│ └─Sequential: 2-5 [1, 128, 32, 32] [1, 176, 32, 32] +│ │ └─InvertedResidual: 3-21 [1, 128, 32, 32] [1, 176, 32, 32] +│ │ └─InvertedResidual: 3-22 [1, 176, 32, 32] [1, 176, 32, 32] +│ │ └─InvertedResidual: 3-23 [1, 176, 32, 32] [1, 176, 32, 32] +│ │ └─InvertedResidual: 3-24 [1, 176, 32, 32] [1, 176, 32, 32] +│ │ └─InvertedResidual: 3-25 [1, 176, 32, 32] [1, 176, 32, 32] +│ │ └─InvertedResidual: 3-26 [1, 176, 32, 32] [1, 176, 32, 32] +│ │ └─InvertedResidual: 3-27 [1, 176, 32, 32] [1, 176, 32, 32] +│ └─Sequential: 2-6 [1, 176, 32, 32] [1, 304, 16, 16] +│ │ └─InvertedResidual: 3-28 [1, 176, 32, 32] [1, 304, 16, 16] +│ │ └─InvertedResidual: 3-29 [1, 304, 16, 16] [1, 304, 16, 16] +│ │ └─InvertedResidual: 3-30 [1, 304, 16, 16] [1, 304, 16, 16] +│ │ └─InvertedResidual: 3-31 [1, 304, 16, 16] [1, 304, 16, 16] +│ │ └─InvertedResidual: 3-32 [1, 304, 16, 16] [1, 304, 16, 16] +│ │ └─InvertedResidual: 3-33 [1, 304, 16, 16] [1, 304, 16, 16] +│ │ └─InvertedResidual: 3-34 [1, 304, 16, 16] [1, 304, 16, 16] +│ │ └─InvertedResidual: 3-35 [1, 304, 16, 16] [1, 304, 16, 16] +│ │ └─InvertedResidual: 3-36 [1, 304, 16, 16] [1, 304, 16, 16] +│ └─Sequential: 2-7 [1, 304, 16, 16] [1, 512, 16, 16] +│ │ └─InvertedResidual: 3-37 [1, 304, 16, 16] [1, 512, 16, 16] +│ │ └─InvertedResidual: 3-38 [1, 512, 16, 16] [1, 512, 16, 16] +│ │ └─InvertedResidual: 3-39 [1, 512, 16, 16] [1, 512, 16, 16] +===============================================================================================