Skip to content

Latest commit

 

History

History
216 lines (214 loc) · 42 KB

File metadata and controls

216 lines (214 loc) · 42 KB

Model Summary

Model: "ZF_UNET_224" ┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓ ┃ Layer (type) ┃ Output Shape ┃ Param # ┃ Connected to ┃ ┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━┩ │ input_layer (InputLayer) │ (None, 224, 224, 3) │ 0 │ - │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d (Conv2D) │ (None, 224, 224, 32) │ 896 │ input_layer[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization │ (None, 224, 224, 32) │ 128 │ conv2d[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation (Activation) │ (None, 224, 224, 32) │ 0 │ batch_normalization[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_1 (Conv2D) │ (None, 224, 224, 32) │ 9,248 │ activation[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_1 │ (None, 224, 224, 32) │ 128 │ conv2d_1[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_1 (Activation) │ (None, 224, 224, 32) │ 0 │ batch_normalization_1[0][… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ max_pooling2d (MaxPooling2D) │ (None, 112, 112, 32) │ 0 │ activation_1[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_2 (Conv2D) │ (None, 112, 112, 64) │ 18,496 │ max_pooling2d[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_2 │ (None, 112, 112, 64) │ 256 │ conv2d_2[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_2 (Activation) │ (None, 112, 112, 64) │ 0 │ batch_normalization_2[0][… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_3 (Conv2D) │ (None, 112, 112, 64) │ 36,928 │ activation_2[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_3 │ (None, 112, 112, 64) │ 256 │ conv2d_3[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_3 (Activation) │ (None, 112, 112, 64) │ 0 │ batch_normalization_3[0][… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ max_pooling2d_1 │ (None, 56, 56, 64) │ 0 │ activation_3[0][0] │ │ (MaxPooling2D) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_4 (Conv2D) │ (None, 56, 56, 128) │ 73,856 │ max_pooling2d_1[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_4 │ (None, 56, 56, 128) │ 512 │ conv2d_4[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_4 (Activation) │ (None, 56, 56, 128) │ 0 │ batch_normalization_4[0][… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_5 (Conv2D) │ (None, 56, 56, 128) │ 147,584 │ activation_4[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_5 │ (None, 56, 56, 128) │ 512 │ conv2d_5[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_5 (Activation) │ (None, 56, 56, 128) │ 0 │ batch_normalization_5[0][… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ max_pooling2d_2 │ (None, 28, 28, 128) │ 0 │ activation_5[0][0] │ │ (MaxPooling2D) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_6 (Conv2D) │ (None, 28, 28, 256) │ 295,168 │ max_pooling2d_2[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_6 │ (None, 28, 28, 256) │ 1,024 │ conv2d_6[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_6 (Activation) │ (None, 28, 28, 256) │ 0 │ batch_normalization_6[0][… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_7 (Conv2D) │ (None, 28, 28, 256) │ 590,080 │ activation_6[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_7 │ (None, 28, 28, 256) │ 1,024 │ conv2d_7[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_7 (Activation) │ (None, 28, 28, 256) │ 0 │ batch_normalization_7[0][… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ max_pooling2d_3 │ (None, 14, 14, 256) │ 0 │ activation_7[0][0] │ │ (MaxPooling2D) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_8 (Conv2D) │ (None, 14, 14, 512) │ 1,180,160 │ max_pooling2d_3[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_8 │ (None, 14, 14, 512) │ 2,048 │ conv2d_8[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_8 (Activation) │ (None, 14, 14, 512) │ 0 │ batch_normalization_8[0][… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_9 (Conv2D) │ (None, 14, 14, 512) │ 2,359,808 │ activation_8[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_9 │ (None, 14, 14, 512) │ 2,048 │ conv2d_9[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_9 (Activation) │ (None, 14, 14, 512) │ 0 │ batch_normalization_9[0][… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ max_pooling2d_4 │ (None, 7, 7, 512) │ 0 │ activation_9[0][0] │ │ (MaxPooling2D) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_10 (Conv2D) │ (None, 7, 7, 1024) │ 4,719,616 │ max_pooling2d_4[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_10 │ (None, 7, 7, 1024) │ 4,096 │ conv2d_10[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_10 (Activation) │ (None, 7, 7, 1024) │ 0 │ batch_normalization_10[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_11 (Conv2D) │ (None, 7, 7, 1024) │ 9,438,208 │ activation_10[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_11 │ (None, 7, 7, 1024) │ 4,096 │ conv2d_11[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_11 (Activation) │ (None, 7, 7, 1024) │ 0 │ batch_normalization_11[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ up_sampling2d (UpSampling2D) │ (None, 14, 14, 1024) │ 0 │ activation_11[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ concatenate (Concatenate) │ (None, 14, 14, 1536) │ 0 │ up_sampling2d[0][0], │ │ │ │ │ activation_9[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_12 (Conv2D) │ (None, 14, 14, 512) │ 7,078,400 │ concatenate[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_12 │ (None, 14, 14, 512) │ 2,048 │ conv2d_12[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_12 (Activation) │ (None, 14, 14, 512) │ 0 │ batch_normalization_12[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_13 (Conv2D) │ (None, 14, 14, 512) │ 2,359,808 │ activation_12[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_13 │ (None, 14, 14, 512) │ 2,048 │ conv2d_13[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_13 (Activation) │ (None, 14, 14, 512) │ 0 │ batch_normalization_13[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ up_sampling2d_1 │ (None, 28, 28, 512) │ 0 │ activation_13[0][0] │ │ (UpSampling2D) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ concatenate_1 (Concatenate) │ (None, 28, 28, 768) │ 0 │ up_sampling2d_1[0][0], │ │ │ │ │ activation_7[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_14 (Conv2D) │ (None, 28, 28, 256) │ 1,769,728 │ concatenate_1[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_14 │ (None, 28, 28, 256) │ 1,024 │ conv2d_14[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_14 (Activation) │ (None, 28, 28, 256) │ 0 │ batch_normalization_14[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_15 (Conv2D) │ (None, 28, 28, 256) │ 590,080 │ activation_14[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_15 │ (None, 28, 28, 256) │ 1,024 │ conv2d_15[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_15 (Activation) │ (None, 28, 28, 256) │ 0 │ batch_normalization_15[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ up_sampling2d_2 │ (None, 56, 56, 256) │ 0 │ activation_15[0][0] │ │ (UpSampling2D) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ concatenate_2 (Concatenate) │ (None, 56, 56, 384) │ 0 │ up_sampling2d_2[0][0], │ │ │ │ │ activation_5[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_16 (Conv2D) │ (None, 56, 56, 128) │ 442,496 │ concatenate_2[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_16 │ (None, 56, 56, 128) │ 512 │ conv2d_16[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_16 (Activation) │ (None, 56, 56, 128) │ 0 │ batch_normalization_16[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_17 (Conv2D) │ (None, 56, 56, 128) │ 147,584 │ activation_16[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_17 │ (None, 56, 56, 128) │ 512 │ conv2d_17[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_17 (Activation) │ (None, 56, 56, 128) │ 0 │ batch_normalization_17[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ up_sampling2d_3 │ (None, 112, 112, 128) │ 0 │ activation_17[0][0] │ │ (UpSampling2D) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ concatenate_3 (Concatenate) │ (None, 112, 112, 192) │ 0 │ up_sampling2d_3[0][0], │ │ │ │ │ activation_3[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_18 (Conv2D) │ (None, 112, 112, 64) │ 110,656 │ concatenate_3[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_18 │ (None, 112, 112, 64) │ 256 │ conv2d_18[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_18 (Activation) │ (None, 112, 112, 64) │ 0 │ batch_normalization_18[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_19 (Conv2D) │ (None, 112, 112, 64) │ 36,928 │ activation_18[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_19 │ (None, 112, 112, 64) │ 256 │ conv2d_19[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_19 (Activation) │ (None, 112, 112, 64) │ 0 │ batch_normalization_19[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ up_sampling2d_4 │ (None, 224, 224, 64) │ 0 │ activation_19[0][0] │ │ (UpSampling2D) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ concatenate_4 (Concatenate) │ (None, 224, 224, 96) │ 0 │ up_sampling2d_4[0][0], │ │ │ │ │ activation_1[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_20 (Conv2D) │ (None, 224, 224, 32) │ 27,680 │ concatenate_4[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_20 │ (None, 224, 224, 32) │ 128 │ conv2d_20[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_20 (Activation) │ (None, 224, 224, 32) │ 0 │ batch_normalization_20[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_21 (Conv2D) │ (None, 224, 224, 32) │ 9,248 │ activation_20[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ batch_normalization_21 │ (None, 224, 224, 32) │ 128 │ conv2d_21[0][0] │ │ (BatchNormalization) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_21 (Activation) │ (None, 224, 224, 32) │ 0 │ batch_normalization_21[0]… │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ spatial_dropout2d │ (None, 224, 224, 32) │ 0 │ activation_21[0][0] │ │ (SpatialDropout2D) │ │ │ │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ conv2d_22 (Conv2D) │ (None, 224, 224, 1) │ 33 │ spatial_dropout2d[0][0] │ ├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤ │ activation_22 (Activation) │ (None, 224, 224, 1) │ 0 │ conv2d_22[0][0] │ └───────────────────────────────┴───────────────────────────┴─────────────────┴────────────────────────────┘ Total params: 31,466,753 (120.04 MB) Trainable params: 31,454,721 (119.99 MB) Non-trainable params: 12,032 (47.00 KB)