451 lines
43 KiB
Text
451 lines
43 KiB
Text
Model: "DeepLabV3+"
|
|
__________________________________________________________________________________________________
|
|
Layer (type) Output Shape Param # Connected to
|
|
==================================================================================================
|
|
input_1 (InputLayer) [(None, 128, 128, 8 0 []
|
|
)]
|
|
|
|
conv1_pad (ZeroPadding2D) (None, 134, 134, 8) 0 ['input_1[0][0]']
|
|
|
|
conv1_conv (Conv2D) (None, 64, 64, 64) 25152 ['conv1_pad[0][0]']
|
|
|
|
conv1_bn (BatchNormalization) (None, 64, 64, 64) 256 ['conv1_conv[0][0]']
|
|
|
|
conv1_relu (Activation) (None, 64, 64, 64) 0 ['conv1_bn[0][0]']
|
|
|
|
pool1_pad (ZeroPadding2D) (None, 66, 66, 64) 0 ['conv1_relu[0][0]']
|
|
|
|
pool1_pool (MaxPooling2D) (None, 32, 32, 64) 0 ['pool1_pad[0][0]']
|
|
|
|
conv2_block1_1_conv (Conv2D) (None, 32, 32, 64) 4160 ['pool1_pool[0][0]']
|
|
|
|
conv2_block1_1_bn (BatchNormal (None, 32, 32, 64) 256 ['conv2_block1_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv2_block1_1_relu (Activatio (None, 32, 32, 64) 0 ['conv2_block1_1_bn[0][0]']
|
|
n)
|
|
|
|
conv2_block1_2_conv (Conv2D) (None, 32, 32, 64) 36928 ['conv2_block1_1_relu[0][0]']
|
|
|
|
conv2_block1_2_bn (BatchNormal (None, 32, 32, 64) 256 ['conv2_block1_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv2_block1_2_relu (Activatio (None, 32, 32, 64) 0 ['conv2_block1_2_bn[0][0]']
|
|
n)
|
|
|
|
conv2_block1_0_conv (Conv2D) (None, 32, 32, 256) 16640 ['pool1_pool[0][0]']
|
|
|
|
conv2_block1_3_conv (Conv2D) (None, 32, 32, 256) 16640 ['conv2_block1_2_relu[0][0]']
|
|
|
|
conv2_block1_0_bn (BatchNormal (None, 32, 32, 256) 1024 ['conv2_block1_0_conv[0][0]']
|
|
ization)
|
|
|
|
conv2_block1_3_bn (BatchNormal (None, 32, 32, 256) 1024 ['conv2_block1_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv2_block1_add (Add) (None, 32, 32, 256) 0 ['conv2_block1_0_bn[0][0]',
|
|
'conv2_block1_3_bn[0][0]']
|
|
|
|
conv2_block1_out (Activation) (None, 32, 32, 256) 0 ['conv2_block1_add[0][0]']
|
|
|
|
conv2_block2_1_conv (Conv2D) (None, 32, 32, 64) 16448 ['conv2_block1_out[0][0]']
|
|
|
|
conv2_block2_1_bn (BatchNormal (None, 32, 32, 64) 256 ['conv2_block2_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv2_block2_1_relu (Activatio (None, 32, 32, 64) 0 ['conv2_block2_1_bn[0][0]']
|
|
n)
|
|
|
|
conv2_block2_2_conv (Conv2D) (None, 32, 32, 64) 36928 ['conv2_block2_1_relu[0][0]']
|
|
|
|
conv2_block2_2_bn (BatchNormal (None, 32, 32, 64) 256 ['conv2_block2_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv2_block2_2_relu (Activatio (None, 32, 32, 64) 0 ['conv2_block2_2_bn[0][0]']
|
|
n)
|
|
|
|
conv2_block2_3_conv (Conv2D) (None, 32, 32, 256) 16640 ['conv2_block2_2_relu[0][0]']
|
|
|
|
conv2_block2_3_bn (BatchNormal (None, 32, 32, 256) 1024 ['conv2_block2_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv2_block2_add (Add) (None, 32, 32, 256) 0 ['conv2_block1_out[0][0]',
|
|
'conv2_block2_3_bn[0][0]']
|
|
|
|
conv2_block2_out (Activation) (None, 32, 32, 256) 0 ['conv2_block2_add[0][0]']
|
|
|
|
conv2_block3_1_conv (Conv2D) (None, 32, 32, 64) 16448 ['conv2_block2_out[0][0]']
|
|
|
|
conv2_block3_1_bn (BatchNormal (None, 32, 32, 64) 256 ['conv2_block3_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv2_block3_1_relu (Activatio (None, 32, 32, 64) 0 ['conv2_block3_1_bn[0][0]']
|
|
n)
|
|
|
|
conv2_block3_2_conv (Conv2D) (None, 32, 32, 64) 36928 ['conv2_block3_1_relu[0][0]']
|
|
|
|
conv2_block3_2_bn (BatchNormal (None, 32, 32, 64) 256 ['conv2_block3_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv2_block3_2_relu (Activatio (None, 32, 32, 64) 0 ['conv2_block3_2_bn[0][0]']
|
|
n)
|
|
|
|
conv2_block3_3_conv (Conv2D) (None, 32, 32, 256) 16640 ['conv2_block3_2_relu[0][0]']
|
|
|
|
conv2_block3_3_bn (BatchNormal (None, 32, 32, 256) 1024 ['conv2_block3_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv2_block3_add (Add) (None, 32, 32, 256) 0 ['conv2_block2_out[0][0]',
|
|
'conv2_block3_3_bn[0][0]']
|
|
|
|
conv2_block3_out (Activation) (None, 32, 32, 256) 0 ['conv2_block3_add[0][0]']
|
|
|
|
conv3_block1_1_conv (Conv2D) (None, 16, 16, 128) 32896 ['conv2_block3_out[0][0]']
|
|
|
|
conv3_block1_1_bn (BatchNormal (None, 16, 16, 128) 512 ['conv3_block1_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block1_1_relu (Activatio (None, 16, 16, 128) 0 ['conv3_block1_1_bn[0][0]']
|
|
n)
|
|
|
|
conv3_block1_2_conv (Conv2D) (None, 16, 16, 128) 147584 ['conv3_block1_1_relu[0][0]']
|
|
|
|
conv3_block1_2_bn (BatchNormal (None, 16, 16, 128) 512 ['conv3_block1_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block1_2_relu (Activatio (None, 16, 16, 128) 0 ['conv3_block1_2_bn[0][0]']
|
|
n)
|
|
|
|
conv3_block1_0_conv (Conv2D) (None, 16, 16, 512) 131584 ['conv2_block3_out[0][0]']
|
|
|
|
conv3_block1_3_conv (Conv2D) (None, 16, 16, 512) 66048 ['conv3_block1_2_relu[0][0]']
|
|
|
|
conv3_block1_0_bn (BatchNormal (None, 16, 16, 512) 2048 ['conv3_block1_0_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block1_3_bn (BatchNormal (None, 16, 16, 512) 2048 ['conv3_block1_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block1_add (Add) (None, 16, 16, 512) 0 ['conv3_block1_0_bn[0][0]',
|
|
'conv3_block1_3_bn[0][0]']
|
|
|
|
conv3_block1_out (Activation) (None, 16, 16, 512) 0 ['conv3_block1_add[0][0]']
|
|
|
|
conv3_block2_1_conv (Conv2D) (None, 16, 16, 128) 65664 ['conv3_block1_out[0][0]']
|
|
|
|
conv3_block2_1_bn (BatchNormal (None, 16, 16, 128) 512 ['conv3_block2_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block2_1_relu (Activatio (None, 16, 16, 128) 0 ['conv3_block2_1_bn[0][0]']
|
|
n)
|
|
|
|
conv3_block2_2_conv (Conv2D) (None, 16, 16, 128) 147584 ['conv3_block2_1_relu[0][0]']
|
|
|
|
conv3_block2_2_bn (BatchNormal (None, 16, 16, 128) 512 ['conv3_block2_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block2_2_relu (Activatio (None, 16, 16, 128) 0 ['conv3_block2_2_bn[0][0]']
|
|
n)
|
|
|
|
conv3_block2_3_conv (Conv2D) (None, 16, 16, 512) 66048 ['conv3_block2_2_relu[0][0]']
|
|
|
|
conv3_block2_3_bn (BatchNormal (None, 16, 16, 512) 2048 ['conv3_block2_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block2_add (Add) (None, 16, 16, 512) 0 ['conv3_block1_out[0][0]',
|
|
'conv3_block2_3_bn[0][0]']
|
|
|
|
conv3_block2_out (Activation) (None, 16, 16, 512) 0 ['conv3_block2_add[0][0]']
|
|
|
|
conv3_block3_1_conv (Conv2D) (None, 16, 16, 128) 65664 ['conv3_block2_out[0][0]']
|
|
|
|
conv3_block3_1_bn (BatchNormal (None, 16, 16, 128) 512 ['conv3_block3_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block3_1_relu (Activatio (None, 16, 16, 128) 0 ['conv3_block3_1_bn[0][0]']
|
|
n)
|
|
|
|
conv3_block3_2_conv (Conv2D) (None, 16, 16, 128) 147584 ['conv3_block3_1_relu[0][0]']
|
|
|
|
conv3_block3_2_bn (BatchNormal (None, 16, 16, 128) 512 ['conv3_block3_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block3_2_relu (Activatio (None, 16, 16, 128) 0 ['conv3_block3_2_bn[0][0]']
|
|
n)
|
|
|
|
conv3_block3_3_conv (Conv2D) (None, 16, 16, 512) 66048 ['conv3_block3_2_relu[0][0]']
|
|
|
|
conv3_block3_3_bn (BatchNormal (None, 16, 16, 512) 2048 ['conv3_block3_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block3_add (Add) (None, 16, 16, 512) 0 ['conv3_block2_out[0][0]',
|
|
'conv3_block3_3_bn[0][0]']
|
|
|
|
conv3_block3_out (Activation) (None, 16, 16, 512) 0 ['conv3_block3_add[0][0]']
|
|
|
|
conv3_block4_1_conv (Conv2D) (None, 16, 16, 128) 65664 ['conv3_block3_out[0][0]']
|
|
|
|
conv3_block4_1_bn (BatchNormal (None, 16, 16, 128) 512 ['conv3_block4_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block4_1_relu (Activatio (None, 16, 16, 128) 0 ['conv3_block4_1_bn[0][0]']
|
|
n)
|
|
|
|
conv3_block4_2_conv (Conv2D) (None, 16, 16, 128) 147584 ['conv3_block4_1_relu[0][0]']
|
|
|
|
conv3_block4_2_bn (BatchNormal (None, 16, 16, 128) 512 ['conv3_block4_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block4_2_relu (Activatio (None, 16, 16, 128) 0 ['conv3_block4_2_bn[0][0]']
|
|
n)
|
|
|
|
conv3_block4_3_conv (Conv2D) (None, 16, 16, 512) 66048 ['conv3_block4_2_relu[0][0]']
|
|
|
|
conv3_block4_3_bn (BatchNormal (None, 16, 16, 512) 2048 ['conv3_block4_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv3_block4_add (Add) (None, 16, 16, 512) 0 ['conv3_block3_out[0][0]',
|
|
'conv3_block4_3_bn[0][0]']
|
|
|
|
conv3_block4_out (Activation) (None, 16, 16, 512) 0 ['conv3_block4_add[0][0]']
|
|
|
|
conv4_block1_1_conv (Conv2D) (None, 8, 8, 256) 131328 ['conv3_block4_out[0][0]']
|
|
|
|
conv4_block1_1_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block1_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block1_1_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block1_1_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block1_2_conv (Conv2D) (None, 8, 8, 256) 590080 ['conv4_block1_1_relu[0][0]']
|
|
|
|
conv4_block1_2_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block1_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block1_2_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block1_2_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block1_0_conv (Conv2D) (None, 8, 8, 1024) 525312 ['conv3_block4_out[0][0]']
|
|
|
|
conv4_block1_3_conv (Conv2D) (None, 8, 8, 1024) 263168 ['conv4_block1_2_relu[0][0]']
|
|
|
|
conv4_block1_0_bn (BatchNormal (None, 8, 8, 1024) 4096 ['conv4_block1_0_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block1_3_bn (BatchNormal (None, 8, 8, 1024) 4096 ['conv4_block1_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block1_add (Add) (None, 8, 8, 1024) 0 ['conv4_block1_0_bn[0][0]',
|
|
'conv4_block1_3_bn[0][0]']
|
|
|
|
conv4_block1_out (Activation) (None, 8, 8, 1024) 0 ['conv4_block1_add[0][0]']
|
|
|
|
conv4_block2_1_conv (Conv2D) (None, 8, 8, 256) 262400 ['conv4_block1_out[0][0]']
|
|
|
|
conv4_block2_1_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block2_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block2_1_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block2_1_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block2_2_conv (Conv2D) (None, 8, 8, 256) 590080 ['conv4_block2_1_relu[0][0]']
|
|
|
|
conv4_block2_2_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block2_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block2_2_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block2_2_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block2_3_conv (Conv2D) (None, 8, 8, 1024) 263168 ['conv4_block2_2_relu[0][0]']
|
|
|
|
conv4_block2_3_bn (BatchNormal (None, 8, 8, 1024) 4096 ['conv4_block2_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block2_add (Add) (None, 8, 8, 1024) 0 ['conv4_block1_out[0][0]',
|
|
'conv4_block2_3_bn[0][0]']
|
|
|
|
conv4_block2_out (Activation) (None, 8, 8, 1024) 0 ['conv4_block2_add[0][0]']
|
|
|
|
conv4_block3_1_conv (Conv2D) (None, 8, 8, 256) 262400 ['conv4_block2_out[0][0]']
|
|
|
|
conv4_block3_1_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block3_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block3_1_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block3_1_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block3_2_conv (Conv2D) (None, 8, 8, 256) 590080 ['conv4_block3_1_relu[0][0]']
|
|
|
|
conv4_block3_2_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block3_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block3_2_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block3_2_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block3_3_conv (Conv2D) (None, 8, 8, 1024) 263168 ['conv4_block3_2_relu[0][0]']
|
|
|
|
conv4_block3_3_bn (BatchNormal (None, 8, 8, 1024) 4096 ['conv4_block3_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block3_add (Add) (None, 8, 8, 1024) 0 ['conv4_block2_out[0][0]',
|
|
'conv4_block3_3_bn[0][0]']
|
|
|
|
conv4_block3_out (Activation) (None, 8, 8, 1024) 0 ['conv4_block3_add[0][0]']
|
|
|
|
conv4_block4_1_conv (Conv2D) (None, 8, 8, 256) 262400 ['conv4_block3_out[0][0]']
|
|
|
|
conv4_block4_1_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block4_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block4_1_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block4_1_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block4_2_conv (Conv2D) (None, 8, 8, 256) 590080 ['conv4_block4_1_relu[0][0]']
|
|
|
|
conv4_block4_2_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block4_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block4_2_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block4_2_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block4_3_conv (Conv2D) (None, 8, 8, 1024) 263168 ['conv4_block4_2_relu[0][0]']
|
|
|
|
conv4_block4_3_bn (BatchNormal (None, 8, 8, 1024) 4096 ['conv4_block4_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block4_add (Add) (None, 8, 8, 1024) 0 ['conv4_block3_out[0][0]',
|
|
'conv4_block4_3_bn[0][0]']
|
|
|
|
conv4_block4_out (Activation) (None, 8, 8, 1024) 0 ['conv4_block4_add[0][0]']
|
|
|
|
conv4_block5_1_conv (Conv2D) (None, 8, 8, 256) 262400 ['conv4_block4_out[0][0]']
|
|
|
|
conv4_block5_1_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block5_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block5_1_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block5_1_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block5_2_conv (Conv2D) (None, 8, 8, 256) 590080 ['conv4_block5_1_relu[0][0]']
|
|
|
|
conv4_block5_2_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block5_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block5_2_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block5_2_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block5_3_conv (Conv2D) (None, 8, 8, 1024) 263168 ['conv4_block5_2_relu[0][0]']
|
|
|
|
conv4_block5_3_bn (BatchNormal (None, 8, 8, 1024) 4096 ['conv4_block5_3_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block5_add (Add) (None, 8, 8, 1024) 0 ['conv4_block4_out[0][0]',
|
|
'conv4_block5_3_bn[0][0]']
|
|
|
|
conv4_block5_out (Activation) (None, 8, 8, 1024) 0 ['conv4_block5_add[0][0]']
|
|
|
|
conv4_block6_1_conv (Conv2D) (None, 8, 8, 256) 262400 ['conv4_block5_out[0][0]']
|
|
|
|
conv4_block6_1_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block6_1_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block6_1_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block6_1_bn[0][0]']
|
|
n)
|
|
|
|
conv4_block6_2_conv (Conv2D) (None, 8, 8, 256) 590080 ['conv4_block6_1_relu[0][0]']
|
|
|
|
conv4_block6_2_bn (BatchNormal (None, 8, 8, 256) 1024 ['conv4_block6_2_conv[0][0]']
|
|
ization)
|
|
|
|
conv4_block6_2_relu (Activatio (None, 8, 8, 256) 0 ['conv4_block6_2_bn[0][0]']
|
|
n)
|
|
|
|
average_pooling2d (AveragePool (None, 1, 1, 256) 0 ['conv4_block6_2_relu[0][0]']
|
|
ing2D)
|
|
|
|
conv2d (Conv2D) (None, 1, 1, 256) 65792 ['average_pooling2d[0][0]']
|
|
|
|
batch_normalization (BatchNorm (None, 1, 1, 256) 1024 ['conv2d[0][0]']
|
|
alization)
|
|
|
|
conv2d_1 (Conv2D) (None, 8, 8, 256) 65536 ['conv4_block6_2_relu[0][0]']
|
|
|
|
conv2d_2 (Conv2D) (None, 8, 8, 256) 589824 ['conv4_block6_2_relu[0][0]']
|
|
|
|
conv2d_3 (Conv2D) (None, 8, 8, 256) 589824 ['conv4_block6_2_relu[0][0]']
|
|
|
|
conv2d_4 (Conv2D) (None, 8, 8, 256) 589824 ['conv4_block6_2_relu[0][0]']
|
|
|
|
tf.nn.relu (TFOpLambda) (None, 1, 1, 256) 0 ['batch_normalization[0][0]']
|
|
|
|
batch_normalization_1 (BatchNo (None, 8, 8, 256) 1024 ['conv2d_1[0][0]']
|
|
rmalization)
|
|
|
|
batch_normalization_2 (BatchNo (None, 8, 8, 256) 1024 ['conv2d_2[0][0]']
|
|
rmalization)
|
|
|
|
batch_normalization_3 (BatchNo (None, 8, 8, 256) 1024 ['conv2d_3[0][0]']
|
|
rmalization)
|
|
|
|
batch_normalization_4 (BatchNo (None, 8, 8, 256) 1024 ['conv2d_4[0][0]']
|
|
rmalization)
|
|
|
|
up_sampling2d (UpSampling2D) (None, 8, 8, 256) 0 ['tf.nn.relu[0][0]']
|
|
|
|
tf.nn.relu_1 (TFOpLambda) (None, 8, 8, 256) 0 ['batch_normalization_1[0][0]']
|
|
|
|
tf.nn.relu_2 (TFOpLambda) (None, 8, 8, 256) 0 ['batch_normalization_2[0][0]']
|
|
|
|
tf.nn.relu_3 (TFOpLambda) (None, 8, 8, 256) 0 ['batch_normalization_3[0][0]']
|
|
|
|
tf.nn.relu_4 (TFOpLambda) (None, 8, 8, 256) 0 ['batch_normalization_4[0][0]']
|
|
|
|
concatenate (Concatenate) (None, 8, 8, 1280) 0 ['up_sampling2d[0][0]',
|
|
'tf.nn.relu_1[0][0]',
|
|
'tf.nn.relu_2[0][0]',
|
|
'tf.nn.relu_3[0][0]',
|
|
'tf.nn.relu_4[0][0]']
|
|
|
|
conv2d_5 (Conv2D) (None, 8, 8, 256) 327680 ['concatenate[0][0]']
|
|
|
|
batch_normalization_5 (BatchNo (None, 8, 8, 256) 1024 ['conv2d_5[0][0]']
|
|
rmalization)
|
|
|
|
conv2d_6 (Conv2D) (None, 32, 32, 48) 3072 ['conv2_block3_2_relu[0][0]']
|
|
|
|
tf.nn.relu_5 (TFOpLambda) (None, 8, 8, 256) 0 ['batch_normalization_5[0][0]']
|
|
|
|
batch_normalization_6 (BatchNo (None, 32, 32, 48) 192 ['conv2d_6[0][0]']
|
|
rmalization)
|
|
|
|
up_sampling2d_1 (UpSampling2D) (None, 32, 32, 256) 0 ['tf.nn.relu_5[0][0]']
|
|
|
|
tf.nn.relu_6 (TFOpLambda) (None, 32, 32, 48) 0 ['batch_normalization_6[0][0]']
|
|
|
|
concatenate_1 (Concatenate) (None, 32, 32, 304) 0 ['up_sampling2d_1[0][0]',
|
|
'tf.nn.relu_6[0][0]']
|
|
|
|
conv2d_7 (Conv2D) (None, 32, 32, 256) 700416 ['concatenate_1[0][0]']
|
|
|
|
batch_normalization_7 (BatchNo (None, 32, 32, 256) 1024 ['conv2d_7[0][0]']
|
|
rmalization)
|
|
|
|
tf.nn.relu_7 (TFOpLambda) (None, 32, 32, 256) 0 ['batch_normalization_7[0][0]']
|
|
|
|
conv2d_8 (Conv2D) (None, 32, 32, 256) 589824 ['tf.nn.relu_7[0][0]']
|
|
|
|
batch_normalization_8 (BatchNo (None, 32, 32, 256) 1024 ['conv2d_8[0][0]']
|
|
rmalization)
|
|
|
|
tf.nn.relu_8 (TFOpLambda) (None, 32, 32, 256) 0 ['batch_normalization_8[0][0]']
|
|
|
|
up_sampling2d_2 (UpSampling2D) (None, 128, 128, 25 0 ['tf.nn.relu_8[0][0]']
|
|
6)
|
|
|
|
conv2d_9 (Conv2D) (None, 128, 128, 2) 514 ['up_sampling2d_2[0][0]']
|
|
|
|
==================================================================================================
|
|
Total params: 11,868,290
|
|
Trainable params: 11,835,554
|
|
Non-trainable params: 32,736
|
|
__________________________________________________________________________________________________
|