| | Layer (type) | Output shape | Param # | Connected to |
| | input (InputLayer) | (None, 240, 240, 3) | 0 | | | stem_conv (Conv2D) | (None, 120, 120, 32) | 864 | input_2[0][0] | | stem_bn (BatchNormalization) | (None, 120, 120, 32) | 128 | stem_conv[0][0] | | stem_activation (Activation) | (None, 120, 120, 32) | 0 | stem_bn[0][0] | | block1a_dwconv (DepthwiseConv2D) | (None, 120, 120, 32) | 288 | stem_activation[0][0] | | block1a_bn (BatchNormalization) | (None, 120, 120, 32) | 128 | block1a_dwconv[0][0] | | block1a_activation (Activation) | (None, 120, 120, 32) | 0 | block1a_bn[0][0] | | block1a_se_squeeze (GlobalAvera) | (None, 32) | 0 | block1a_activation[0][0] | | block1a_se_reshape (Reshape) | (None, 1, 1, 32) | 0 | block1a_se_squeeze[0][0] | | block1a_se_reduce (Conv2D) | (None, 1, 1, 8) | 264 | block1a_se_reshape[0][0] | | block1a_se_expand (Conv2D) | (None, 1, 1, 32) | 288 | block1a_se_reduce[0][0] | | block1a_se_excite (Multiply) | (None, 120, 120, 32) | 0 | block1a_activation[0][0] | | | | | block1a_se_expand[0][0] | | block1a_project_conv (Conv2D) | (None, 120, 120, 16) | 512 | block1a_se_excite[0][0] | | block1a_project_bn (BatchNormal) | (None, 120, 120, 16) | 64 | block1a_project_conv[0][0] | | Continue to block 1b until block 7a (script not shown) | | block7b_expand_conv (Conv2D) | (None, 8, 8, 1920) | 614400 | block7a_project_bn[0][0] | | block7b_expand_bn (BatchNormali) | (None, 8, 8, 1920) | 7680 | block7b_expand_conv[0][0] | | block7b_expand_activation (Acti) | (None, 8, 8, 1920) | 0 | block7b_expand_bn[0][0] | | block7b_dwconv (DepthwiseConv2D) | (None, 8, 8, 1920) | 17280 | block7b_expand_activation[0][0] | | block7b_bn (BatchNormalization) | (None, 8, 8, 1920) | 7680 | block7b_dwconv[0][0] | | block7b_activation (Activation) | (None, 8, 8, 1920) | 0 | block7b_bn[0][0] | | block7b_se_squeeze (GlobalAvera) | (None, 1920) | 0 | block7b_activation[0][0] | | block7b_se_reshape (Reshape) | (None, 1, 1, 1920) | 0 | block7b_se_squeeze[0][0] | | block7b_se_reduce (Conv2D) | (None, 1, 1, 80) | 153680 | block7b_se_reshape[0][0] | | block7b_se_expand (Conv2D) | (None, 1, 1, 1920) | 155520 | block7b_se_reduce[0][0] | | block7b_se_excite (Multiply) | (None, 8, 8, 1920) | 0 | block7b_activation[0][0] | | | | | block7b_se_expand[0][0] | | block7b_project_conv (Conv2D) | (None, 8, 8, 320) | 614400 | block7b_se_excite[0][0] | | block7b_project_bn (BatchNormal) | (None, 8, 8, 320) | 1280 | block7b_project_conv[0][0] | | block7b_drop (FixedDropout) | (None, 8, 8, 320) | 0 | block7b_project_bn[0][0] | | block7b_add (Add) | (None, 8, 8, 320) | 0 | block7b_drop[0][0] | | | | | block7a_project_bn[0][0] | | top_conv (Conv2D) | (None, 8, 8, 1280) | 409600 | block7b_add[0][0] | | top_bn (BatchNormalization) | (None, 8, 8, 1280) | 5120 | top_conv[0][0] | | top_activation (Activation) | (None, 8, 8, 1280) | 0 | top_bn[0][0] | | global_average_pooling2d_1 (Glo) | (None, 1280) | 0 | top_activation[0][0] | | dropout_4 (Dropout) | (None, 1280) | 0 | global_average_pooling2d_1[0][0] | | dense_4 (Dense) | (None, 1024) | 1311744 | dropout_4[0][0] | | dense_5 (Dense) | (None, 3) | 3075 | dense_4[0][0] | | Layer (type) | Output shape | Param # | Connected to | | Input (InputLayer) | (None, 240, 240, 3) | 0 | |
|
|