How to get the layers of the YOLOv8 model when getting the configuration of the model ?

[MagiPrince]

Hi everyone,

I am working with the YOLOV8Detector from the YOLOv8 model, and I want to get all informations about the layers of the model from the get_config() function. But doing so I only have the following output :

{'name': 'yolov8_detector_quantized', 'trainable': True, 'num_classes': 2, 'bounding_box_format': 'center_xywh', 'fpn_depth': 2, 'backbone': {'module': 'keras_cv.models.object_detection.yolo_v8_quantized.yolo_v8_backbone_quantized', 'class_name': 'YOLOV8BackboneQuantized', 'config': {'name': 'yolov8_backbone_quantized', 'trainable': True, 'include_rescaling': True, 'input_shape': (64, 64, 3), 'stackwise_channels': [32, 64, 128, 256], 'stackwise_depth': [1, 2, 2, 1], 'activation': 'swish'}, 'registered_name': 'keras_cv.models>YOLOV8BackboneQuantized', 'build_config': {'input_shape': (None, 64, 64, 3)}}, 'label_encoder': <keras_cv.models.object_detection.yolo_v8_quantized.yolo_v8_label_encoder_quantized.YOLOV8LabelEncoderQuantized object at 0x7ff66457e160>, 'prediction_decoder': <keras_cv.layers.object_detection.multi_class_non_max_suppression.MultiClassNonMaxSuppression object at 0x7ff66457ecd0>}

And I wanted to get something like when getting the config of the model layer :

{'name': 'model', 'trainable': True, 'layers': [{'module': 'keras.layers', 'class_name': 'InputLayer', 'config': {'batch_input_shape': (None, 64, 64, 3), 'dtype': 'float32', 'sparse': False, 'ragged': False, 'name': 'input_1'}, 'registered_name': None, 'name': 'input_1', 'inbound_nodes': []}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stem_1_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 64, 64, 3)}, 'name': 'stem_1_pad', 'inbound_nodes': [[['input_1', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stem_1_conv', 'trainable': True, 'dtype': 'float32', 'filters': 16, 'kernel_size': (3, 3), 'strides': (2, 2), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 66, 66, 3)}, 'name': 'stem_1_conv', 'inbound_nodes': [[['stem_1_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stem_1_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 32, 32, 16)}, 'name': 'stem_1_bn', 'inbound_nodes': [[['stem_1_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stem_1', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 32, 32, 16)}, 'name': 'stem_1', 'inbound_nodes': [[['stem_1_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stem_2_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 32, 32, 16)}, 'name': 'stem_2_pad', 'inbound_nodes': [[['stem_1', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stem_2_conv', 'trainable': True, 'dtype': 'float32', 'filters': 32, 'kernel_size': (3, 3), 'strides': (2, 2), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 34, 34, 16)}, 'name': 'stem_2_conv', 'inbound_nodes': [[['stem_2_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stem_2_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 32)}, 'name': 'stem_2_bn', 'inbound_nodes': [[['stem_2_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stem_2', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 32)}, 'name': 'stem_2', 'inbound_nodes': [[['stem_2_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack1_c2f_pre_conv', 'trainable': True, 'dtype': 'float32', 'filters': 32, 'kernel_size': (1, 1), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 32)}, 'name': 'stack1_c2f_pre_conv', 'inbound_nodes': [[['stem_2', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack1_c2f_pre_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 32)}, 'name': 'stack1_c2f_pre_bn', 'inbound_nodes': [[['stack1_c2f_pre_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack1_c2f_pre', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 32)}, 'name': 'stack1_c2f_pre', 'inbound_nodes': [[['stack1_c2f_pre_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack1_c2f_pre_0_1_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 32)}, 'name': 'stack1_c2f_pre_0_1_pad', 'inbound_nodes': [[['stack1_c2f_pre', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack1_c2f_pre_0_1_conv', 'trainable': True, 'dtype': 'float32', 'filters': 16, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 18, 18, 32)}, 'name': 'stack1_c2f_pre_0_1_conv', 'inbound_nodes': [[['stack1_c2f_pre_0_1_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack1_c2f_pre_0_1_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 16)}, 'name': 'stack1_c2f_pre_0_1_bn', 'inbound_nodes': [[['stack1_c2f_pre_0_1_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack1_c2f_pre_0_1', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 16)}, 'name': 'stack1_c2f_pre_0_1', 'inbound_nodes': [[['stack1_c2f_pre_0_1_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack1_c2f_pre_0_2_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 16)}, 'name': 'stack1_c2f_pre_0_2_pad', 'inbound_nodes': [[['stack1_c2f_pre_0_1', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack1_c2f_pre_0_2_conv', 'trainable': True, 'dtype': 'float32', 'filters': 16, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 18, 18, 16)}, 'name': 'stack1_c2f_pre_0_2_conv', 'inbound_nodes': [[['stack1_c2f_pre_0_2_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack1_c2f_pre_0_2_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 16)}, 'name': 'stack1_c2f_pre_0_2_bn', 'inbound_nodes': [[['stack1_c2f_pre_0_2_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack1_c2f_pre_0_2', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 16)}, 'name': 'stack1_c2f_pre_0_2', 'inbound_nodes': [[['stack1_c2f_pre_0_2_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack1_c2f_output_conv', 'trainable': True, 'dtype': 'float32', 'filters': 32, 'kernel_size': (1, 1), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 16)}, 'name': 'stack1_c2f_output_conv', 'inbound_nodes': [[['stack1_c2f_pre_0_2', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack1_c2f_output_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 32)}, 'name': 'stack1_c2f_output_bn', 'inbound_nodes': [[['stack1_c2f_output_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack1_c2f_output', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 32)}, 'name': 'stack1_c2f_output', 'inbound_nodes': [[['stack1_c2f_output_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack2_downsample_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 16, 16, 32)}, 'name': 'stack2_downsample_pad', 'inbound_nodes': [[['stack1_c2f_output', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack2_downsample_conv', 'trainable': True, 'dtype': 'float32', 'filters': 64, 'kernel_size': (3, 3), 'strides': (2, 2), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 18, 18, 32)}, 'name': 'stack2_downsample_conv', 'inbound_nodes': [[['stack2_downsample_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack2_downsample_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 64)}, 'name': 'stack2_downsample_bn', 'inbound_nodes': [[['stack2_downsample_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack2_downsample', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 64)}, 'name': 'stack2_downsample', 'inbound_nodes': [[['stack2_downsample_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack2_c2f_pre_conv', 'trainable': True, 'dtype': 'float32', 'filters': 64, 'kernel_size': (1, 1), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 64)}, 'name': 'stack2_c2f_pre_conv', 'inbound_nodes': [[['stack2_downsample', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack2_c2f_pre_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 64)}, 'name': 'stack2_c2f_pre_bn', 'inbound_nodes': [[['stack2_c2f_pre_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack2_c2f_pre', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 64)}, 'name': 'stack2_c2f_pre', 'inbound_nodes': [[['stack2_c2f_pre_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack2_c2f_pre_0_1_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 64)}, 'name': 'stack2_c2f_pre_0_1_pad', 'inbound_nodes': [[['stack2_c2f_pre', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack2_c2f_pre_0_1_conv', 'trainable': True, 'dtype': 'float32', 'filters': 32, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 10, 10, 64)}, 'name': 'stack2_c2f_pre_0_1_conv', 'inbound_nodes': [[['stack2_c2f_pre_0_1_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack2_c2f_pre_0_1_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_0_1_bn', 'inbound_nodes': [[['stack2_c2f_pre_0_1_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack2_c2f_pre_0_1', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_0_1', 'inbound_nodes': [[['stack2_c2f_pre_0_1_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack2_c2f_pre_0_2_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_0_2_pad', 'inbound_nodes': [[['stack2_c2f_pre_0_1', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack2_c2f_pre_0_2_conv', 'trainable': True, 'dtype': 'float32', 'filters': 32, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 10, 10, 32)}, 'name': 'stack2_c2f_pre_0_2_conv', 'inbound_nodes': [[['stack2_c2f_pre_0_2_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack2_c2f_pre_0_2_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_0_2_bn', 'inbound_nodes': [[['stack2_c2f_pre_0_2_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack2_c2f_pre_0_2', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_0_2', 'inbound_nodes': [[['stack2_c2f_pre_0_2_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack2_c2f_pre_1_1_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_1_1_pad', 'inbound_nodes': [[['stack2_c2f_pre_0_2', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack2_c2f_pre_1_1_conv', 'trainable': True, 'dtype': 'float32', 'filters': 32, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 10, 10, 32)}, 'name': 'stack2_c2f_pre_1_1_conv', 'inbound_nodes': [[['stack2_c2f_pre_1_1_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack2_c2f_pre_1_1_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_1_1_bn', 'inbound_nodes': [[['stack2_c2f_pre_1_1_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack2_c2f_pre_1_1', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_1_1', 'inbound_nodes': [[['stack2_c2f_pre_1_1_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack2_c2f_pre_1_2_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_1_2_pad', 'inbound_nodes': [[['stack2_c2f_pre_1_1', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack2_c2f_pre_1_2_conv', 'trainable': True, 'dtype': 'float32', 'filters': 32, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 10, 10, 32)}, 'name': 'stack2_c2f_pre_1_2_conv', 'inbound_nodes': [[['stack2_c2f_pre_1_2_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack2_c2f_pre_1_2_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_1_2_bn', 'inbound_nodes': [[['stack2_c2f_pre_1_2_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack2_c2f_pre_1_2', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_pre_1_2', 'inbound_nodes': [[['stack2_c2f_pre_1_2_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack2_c2f_output_conv', 'trainable': True, 'dtype': 'float32', 'filters': 64, 'kernel_size': (1, 1), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 32)}, 'name': 'stack2_c2f_output_conv', 'inbound_nodes': [[['stack2_c2f_pre_1_2', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack2_c2f_output_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 64)}, 'name': 'stack2_c2f_output_bn', 'inbound_nodes': [[['stack2_c2f_output_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack2_c2f_output', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 64)}, 'name': 'stack2_c2f_output', 'inbound_nodes': [[['stack2_c2f_output_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack3_downsample_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 8, 8, 64)}, 'name': 'stack3_downsample_pad', 'inbound_nodes': [[['stack2_c2f_output', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack3_downsample_conv', 'trainable': True, 'dtype': 'float32', 'filters': 128, 'kernel_size': (3, 3), 'strides': (2, 2), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 10, 10, 64)}, 'name': 'stack3_downsample_conv', 'inbound_nodes': [[['stack3_downsample_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack3_downsample_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 128)}, 'name': 'stack3_downsample_bn', 'inbound_nodes': [[['stack3_downsample_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack3_downsample', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 128)}, 'name': 'stack3_downsample', 'inbound_nodes': [[['stack3_downsample_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack3_c2f_pre_conv', 'trainable': True, 'dtype': 'float32', 'filters': 128, 'kernel_size': (1, 1), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 128)}, 'name': 'stack3_c2f_pre_conv', 'inbound_nodes': [[['stack3_downsample', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack3_c2f_pre_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 128)}, 'name': 'stack3_c2f_pre_bn', 'inbound_nodes': [[['stack3_c2f_pre_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack3_c2f_pre', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 128)}, 'name': 'stack3_c2f_pre', 'inbound_nodes': [[['stack3_c2f_pre_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack3_c2f_pre_0_1_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 128)}, 'name': 'stack3_c2f_pre_0_1_pad', 'inbound_nodes': [[['stack3_c2f_pre', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack3_c2f_pre_0_1_conv', 'trainable': True, 'dtype': 'float32', 'filters': 64, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 6, 6, 128)}, 'name': 'stack3_c2f_pre_0_1_conv', 'inbound_nodes': [[['stack3_c2f_pre_0_1_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack3_c2f_pre_0_1_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_0_1_bn', 'inbound_nodes': [[['stack3_c2f_pre_0_1_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack3_c2f_pre_0_1', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_0_1', 'inbound_nodes': [[['stack3_c2f_pre_0_1_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack3_c2f_pre_0_2_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_0_2_pad', 'inbound_nodes': [[['stack3_c2f_pre_0_1', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack3_c2f_pre_0_2_conv', 'trainable': True, 'dtype': 'float32', 'filters': 64, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 6, 6, 64)}, 'name': 'stack3_c2f_pre_0_2_conv', 'inbound_nodes': [[['stack3_c2f_pre_0_2_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack3_c2f_pre_0_2_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_0_2_bn', 'inbound_nodes': [[['stack3_c2f_pre_0_2_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack3_c2f_pre_0_2', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_0_2', 'inbound_nodes': [[['stack3_c2f_pre_0_2_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack3_c2f_pre_1_1_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_1_1_pad', 'inbound_nodes': [[['stack3_c2f_pre_0_2', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack3_c2f_pre_1_1_conv', 'trainable': True, 'dtype': 'float32', 'filters': 64, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 6, 6, 64)}, 'name': 'stack3_c2f_pre_1_1_conv', 'inbound_nodes': [[['stack3_c2f_pre_1_1_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack3_c2f_pre_1_1_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_1_1_bn', 'inbound_nodes': [[['stack3_c2f_pre_1_1_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack3_c2f_pre_1_1', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_1_1', 'inbound_nodes': [[['stack3_c2f_pre_1_1_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack3_c2f_pre_1_2_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_1_2_pad', 'inbound_nodes': [[['stack3_c2f_pre_1_1', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack3_c2f_pre_1_2_conv', 'trainable': True, 'dtype': 'float32', 'filters': 64, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 6, 6, 64)}, 'name': 'stack3_c2f_pre_1_2_conv', 'inbound_nodes': [[['stack3_c2f_pre_1_2_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack3_c2f_pre_1_2_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_1_2_bn', 'inbound_nodes': [[['stack3_c2f_pre_1_2_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack3_c2f_pre_1_2', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_pre_1_2', 'inbound_nodes': [[['stack3_c2f_pre_1_2_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack3_c2f_output_conv', 'trainable': True, 'dtype': 'float32', 'filters': 128, 'kernel_size': (1, 1), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 64)}, 'name': 'stack3_c2f_output_conv', 'inbound_nodes': [[['stack3_c2f_pre_1_2', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack3_c2f_output_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 128)}, 'name': 'stack3_c2f_output_bn', 'inbound_nodes': [[['stack3_c2f_output_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack3_c2f_output', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 128)}, 'name': 'stack3_c2f_output', 'inbound_nodes': [[['stack3_c2f_output_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack4_downsample_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 128)}, 'name': 'stack4_downsample_pad', 'inbound_nodes': [[['stack3_c2f_output', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack4_downsample_conv', 'trainable': True, 'dtype': 'float32', 'filters': 256, 'kernel_size': (3, 3), 'strides': (2, 2), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 6, 6, 128)}, 'name': 'stack4_downsample_conv', 'inbound_nodes': [[['stack4_downsample_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack4_downsample_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_downsample_bn', 'inbound_nodes': [[['stack4_downsample_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack4_downsample', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_downsample', 'inbound_nodes': [[['stack4_downsample_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack4_c2f_pre_conv', 'trainable': True, 'dtype': 'float32', 'filters': 256, 'kernel_size': (1, 1), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_c2f_pre_conv', 'inbound_nodes': [[['stack4_downsample', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack4_c2f_pre_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_c2f_pre_bn', 'inbound_nodes': [[['stack4_c2f_pre_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack4_c2f_pre', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_c2f_pre', 'inbound_nodes': [[['stack4_c2f_pre_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack4_c2f_pre_0_1_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_c2f_pre_0_1_pad', 'inbound_nodes': [[['stack4_c2f_pre', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack4_c2f_pre_0_1_conv', 'trainable': True, 'dtype': 'float32', 'filters': 128, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 256)}, 'name': 'stack4_c2f_pre_0_1_conv', 'inbound_nodes': [[['stack4_c2f_pre_0_1_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack4_c2f_pre_0_1_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_c2f_pre_0_1_bn', 'inbound_nodes': [[['stack4_c2f_pre_0_1_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack4_c2f_pre_0_1', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_c2f_pre_0_1', 'inbound_nodes': [[['stack4_c2f_pre_0_1_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'ZeroPadding2D', 'config': {'name': 'stack4_c2f_pre_0_2_pad', 'trainable': True, 'dtype': 'float32', 'padding': ((1, 1), (1, 1)), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_c2f_pre_0_2_pad', 'inbound_nodes': [[['stack4_c2f_pre_0_1', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack4_c2f_pre_0_2_conv', 'trainable': True, 'dtype': 'float32', 'filters': 128, 'kernel_size': (3, 3), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 4, 4, 128)}, 'name': 'stack4_c2f_pre_0_2_conv', 'inbound_nodes': [[['stack4_c2f_pre_0_2_pad', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack4_c2f_pre_0_2_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_c2f_pre_0_2_bn', 'inbound_nodes': [[['stack4_c2f_pre_0_2_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack4_c2f_pre_0_2', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_c2f_pre_0_2', 'inbound_nodes': [[['stack4_c2f_pre_0_2_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack4_c2f_output_conv', 'trainable': True, 'dtype': 'float32', 'filters': 256, 'kernel_size': (1, 1), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_c2f_output_conv', 'inbound_nodes': [[['stack4_c2f_pre_0_2', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack4_c2f_output_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_c2f_output_bn', 'inbound_nodes': [[['stack4_c2f_output_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack4_c2f_output', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_c2f_output', 'inbound_nodes': [[['stack4_c2f_output_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack4_spp_fast_pre_conv', 'trainable': True, 'dtype': 'float32', 'filters': 128, 'kernel_size': (1, 1), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_spp_fast_pre_conv', 'inbound_nodes': [[['stack4_c2f_output', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack4_spp_fast_pre_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_spp_fast_pre_bn', 'inbound_nodes': [[['stack4_spp_fast_pre_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack4_spp_fast_pre', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_spp_fast_pre', 'inbound_nodes': [[['stack4_spp_fast_pre_bn', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'MaxPooling2D', 'config': {'name': 'stack4_spp_fast_pool1', 'trainable': True, 'dtype': 'float32', 'pool_size': (5, 5), 'padding': 'same', 'strides': (1, 1), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_spp_fast_pool1', 'inbound_nodes': [[['stack4_spp_fast_pre', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Concatenate', 'config': {'name': 'concatenate', 'trainable': True, 'dtype': 'float32', 'axis': -1}, 'registered_name': None, 'build_config': {'input_shape': [(None, 2, 2, 128), (None, 2, 2, 128)]}, 'name': 'concatenate', 'inbound_nodes': [[['stack4_spp_fast_pre', 0, 0, {}], ['stack4_spp_fast_pool1', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'MaxPooling2D', 'config': {'name': 'stack4_spp_fast_pool2', 'trainable': True, 'dtype': 'float32', 'pool_size': (5, 5), 'padding': 'same', 'strides': (1, 1), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_spp_fast_pool2', 'inbound_nodes': [[['stack4_spp_fast_pool1', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Concatenate', 'config': {'name': 'concatenate_1', 'trainable': True, 'dtype': 'float32', 'axis': -1}, 'registered_name': None, 'build_config': {'input_shape': [(None, 2, 2, 256), (None, 2, 2, 128)]}, 'name': 'concatenate_1', 'inbound_nodes': [[['concatenate', 0, 0, {}], ['stack4_spp_fast_pool2', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'MaxPooling2D', 'config': {'name': 'stack4_spp_fast_pool3', 'trainable': True, 'dtype': 'float32', 'pool_size': (5, 5), 'padding': 'same', 'strides': (1, 1), 'data_format': 'channels_last'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 128)}, 'name': 'stack4_spp_fast_pool3', 'inbound_nodes': [[['stack4_spp_fast_pool2', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Concatenate', 'config': {'name': 'concatenate_2', 'trainable': True, 'dtype': 'float32', 'axis': -1}, 'registered_name': None, 'build_config': {'input_shape': [(None, 2, 2, 384), (None, 2, 2, 128)]}, 'name': 'concatenate_2', 'inbound_nodes': [[['concatenate_1', 0, 0, {}], ['stack4_spp_fast_pool3', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Conv2D', 'config': {'name': 'stack4_spp_fast_output_conv', 'trainable': True, 'dtype': 'float32', 'filters': 256, 'kernel_size': (1, 1), 'strides': (1, 1), 'padding': 'valid', 'data_format': 'channels_last', 'dilation_rate': (1, 1), 'groups': 1, 'activation': 'linear', 'use_bias': False, 'kernel_initializer': {'module': 'keras.initializers', 'class_name': 'GlorotUniform', 'config': {'seed': None}, 'registered_name': None}, 'bias_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'kernel_regularizer': None, 'bias_regularizer': None, 'activity_regularizer': None, 'kernel_constraint': None, 'bias_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 512)}, 'name': 'stack4_spp_fast_output_conv', 'inbound_nodes': [[['concatenate_2', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'BatchNormalization', 'config': {'name': 'stack4_spp_fast_output_bn', 'trainable': True, 'dtype': 'float32', 'axis': [3], 'momentum': 0.97, 'epsilon': 0.001, 'center': True, 'scale': True, 'beta_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'gamma_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'moving_mean_initializer': {'module': 'keras.initializers', 'class_name': 'Zeros', 'config': {}, 'registered_name': None}, 'moving_variance_initializer': {'module': 'keras.initializers', 'class_name': 'Ones', 'config': {}, 'registered_name': None}, 'beta_regularizer': None, 'gamma_regularizer': None, 'beta_constraint': None, 'gamma_constraint': None}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_spp_fast_output_bn', 'inbound_nodes': [[['stack4_spp_fast_output_conv', 0, 0, {}]]]}, {'module': 'keras.layers', 'class_name': 'Activation', 'config': {'name': 'stack4_spp_fast_output', 'trainable': True, 'dtype': 'float32', 'activation': 'swish'}, 'registered_name': None, 'build_config': {'input_shape': (None, 2, 2, 256)}, 'name': 'stack4_spp_fast_output', 'inbound_nodes': [[['stack4_spp_fast_output_bn', 0, 0, {}]]]}], 'input_layers': [['input_1', 0, 0]], 'output_layers': {'P3': ['stack2_c2f_output', 0, 0], 'P4': ['stack3_c2f_output', 0, 0], 'P5': ['stack4_spp_fast_output', 0, 0]}}

That contains the layers, but also the input and output layers. Does anyone know how I can get that ?

Thanks for your help !

[jbischof]

Yes, we have overrode the get_config method. However, the recommended (much easier!) way to inspect functional layers is model.layers.