________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ================================================================================ input_image (InputLayer) [(None, 640, 640, 0 ________________________________________________________________________________ tf_op_layer_truediv (Tens [(None, 640, 640, 0 input_image[0][0] ________________________________________________________________________________ tf_op_layer_sub (TensorFl [(None, 640, 640, 0 tf_op_layer_truediv[0][0] ________________________________________________________________________________ MobileNetV2_extrator (Mod ((None, 80, 80, 3 1364864 tf_op_layer_sub[0][0] ________________________________________________________________________________ FPN (FPN) ((None, 80, 80, 6 93440 MobileNetV2_extrator[1][0] MobileNetV2_extrator[1][1] MobileNetV2_extrator[1][2] ________________________________________________________________________________ SSH_0 (SSH) (None, 80, 80, 64 34944 FPN[0][0] ________________________________________________________________________________ SSH_1 (SSH) (None, 40, 40, 64 34944 FPN[0][1] ________________________________________________________________________________ SSH_2 (SSH) (None, 20, 20, 64 34944 FPN[0][2] ________________________________________________________________________________ ClassHead_0 (ClassHead) (None, None, 2) 260 SSH_0[0][0] ________________________________________________________________________________ ClassHead_1 (ClassHead) (None, None, 2) 260 SSH_1[0][0] ________________________________________________________________________________ ClassHead_2 (ClassHead) (None, None, 2) 260 SSH_2[0][0] ________________________________________________________________________________ BboxHead_0 (BboxHead) (None, None, 4) 520 SSH_0[0][0] ________________________________________________________________________________ BboxHead_1 (BboxHead) (None, None, 4) 520 SSH_1[0][0] ________________________________________________________________________________ BboxHead_2 (BboxHead) (None, None, 4) 520 SSH_2[0][0] ________________________________________________________________________________ LandmarkHead_0 (LandmarkH (None, None, 10) 1300 SSH_0[0][0] ________________________________________________________________________________ LandmarkHead_1 (LandmarkH (None, None, 10) 1300 SSH_1[0][0] ________________________________________________________________________________ LandmarkHead_2 (LandmarkH (None, None, 10) 1300 SSH_2[0][0] ________________________________________________________________________________ tf_op_layer_concat_2 (Ten [(None, None, 2)] 0 ClassHead_0[0][0] ClassHead_1[0][0] ClassHead_2[0][0] ________________________________________________________________________________ tf_op_layer_concat (Tenso [(None, None, 4)] 0 BboxHead_0[0][0] BboxHead_1[0][0] BboxHead_2[0][0] ________________________________________________________________________________ tf_op_layer_concat_1 (Ten [(None, None, 10) 0 LandmarkHead_0[0][0] LandmarkHead_1[0][0] LandmarkHead_2[0][0] ________________________________________________________________________________ softmax (Softmax) (None, None, 2) 0 tf_op_layer_concat_2[0][0] ================================================================================ Total params: 1,569,376 Trainable params: 1,541,088 Non-trainable params: 28,288 ________________________________________________________________________________