{ "cells": [ { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [], "source": [ "import os\n", "import sys\n", "sys.path += [os.path.abspath('keras-yolo3-master')]\n", "import argparse\n", "\n", "import numpy as np\n", "import json\n", "from voc import parse_voc_annotation\n", "from yolo import create_yolov3_model, dummy_loss\n", "from generator import BatchGenerator\n", "from utils.utils import normalize, evaluate, makedirs\n", "from tensorflow.keras.callbacks import EarlyStopping, ReduceLROnPlateau, ModelCheckpoint\n", "from tensorflow.keras.optimizers import Adam\n", "from callbacks import CustomModelCheckpoint, CustomTensorBoard\n", "from utils.multi_gpu_model import multi_gpu_model\n", "import tensorflow as tf\n", "from tensorflow import keras\n", "from tensorflow.keras.models import load_model\n", "\n", "\n", "tf.keras.backend.clear_session()\n", "tf.config.experimental_run_functions_eagerly(True)\n" ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [], "source": [ "def create_training_instances(\n", " train_annot_folder,\n", " train_image_folder,\n", " train_cache,\n", " valid_annot_folder,\n", " valid_image_folder,\n", " valid_cache,\n", " labels,\n", "):\n", " # parse annotations of the training set\n", " train_ints, train_labels = parse_voc_annotation(train_annot_folder, train_image_folder, train_cache, labels)\n", "\n", " # parse annotations of the validation set, if any, otherwise split the training set\n", " if os.path.exists(valid_annot_folder):\n", " valid_ints, valid_labels = parse_voc_annotation(valid_annot_folder, valid_image_folder, valid_cache, labels)\n", " else:\n", " print(\"valid_annot_folder not exists. Spliting the trainining set.\")\n", "\n", " train_valid_split = int(0.8*len(train_ints))\n", " np.random.seed(0)\n", " np.random.shuffle(train_ints)\n", " np.random.seed()\n", "\n", " valid_ints = train_ints[train_valid_split:]\n", " train_ints = train_ints[:train_valid_split]\n", "\n", " # compare the seen labels with the given labels in config.json\n", " if len(labels) > 0:\n", " overlap_labels = set(labels).intersection(set(train_labels.keys()))\n", "\n", " print('Seen labels: \\t' + str(train_labels) + '\\n')\n", " print('Given labels: \\t' + str(labels))\n", "\n", " # return None, None, None if some given label is not in the dataset\n", " if len(overlap_labels) < len(labels):\n", " print('Some labels have no annotations! Please revise the list of labels in the config.json.')\n", " return None, None, None\n", " else:\n", " print('No labels are provided. Train on all seen labels.')\n", " print(train_labels)\n", " labels = train_labels.keys()\n", "\n", " max_box_per_image = max([len(inst['object']) for inst in (train_ints + valid_ints)])\n", "\n", " return train_ints, valid_ints, sorted(labels), max_box_per_image\n", "\n", "\n", "def create_model(\n", " nb_class,\n", " anchors,\n", " max_box_per_image,\n", " max_grid, batch_size,\n", " warmup_batches,\n", " ignore_thresh,\n", " multi_gpu,\n", " saved_weights_name,\n", " lr,\n", " grid_scales,\n", " obj_scale,\n", " noobj_scale,\n", " xywh_scale,\n", " class_scale,\n", " backend\n", "):\n", " if multi_gpu > 1:\n", " with tf.device('/cpu:0'):\n", " template_model, infer_model = create_yolov3_model(\n", " nb_class = nb_class,\n", " anchors = anchors,\n", " max_box_per_image = max_box_per_image,\n", " max_grid = max_grid,\n", " batch_size = batch_size//multi_gpu,\n", " warmup_batches = warmup_batches,\n", " ignore_thresh = ignore_thresh,\n", " grid_scales = grid_scales,\n", " obj_scale = obj_scale,\n", " noobj_scale = noobj_scale,\n", " xywh_scale = xywh_scale,\n", " class_scale = class_scale\n", " )\n", " else:\n", " template_model, infer_model = create_yolov3_model(\n", " nb_class = nb_class,\n", " anchors = anchors,\n", " max_box_per_image = max_box_per_image,\n", " max_grid = max_grid,\n", " batch_size = batch_size,\n", " warmup_batches = warmup_batches,\n", " ignore_thresh = ignore_thresh,\n", " grid_scales = grid_scales,\n", " obj_scale = obj_scale,\n", " noobj_scale = noobj_scale,\n", " xywh_scale = xywh_scale,\n", " class_scale = class_scale\n", " )\n", "\n", " # load the pretrained weight if exists, otherwise load the backend weight only\n", " if os.path.exists(saved_weights_name):\n", " print(\"\\nLoading pretrained weights.\\n\")\n", " template_model.load_weights(saved_weights_name)\n", " else:\n", " template_model.load_weights(backend, by_name=True)\n", "\n", " if multi_gpu > 1:\n", " train_model = multi_gpu_model(template_model, gpus=multi_gpu)\n", " else:\n", " train_model = template_model\n", "\n", " optimizer = Adam(lr=lr, clipnorm=0.001)\n", " train_model.compile(loss=dummy_loss, optimizer=optimizer)\n", "\n", " return train_model, infer_model" ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Seen labels: \t{'2': 127}\n", "\n", "Given labels: \t['2']\n", "\n", "Training on: \t['2']\n", "\n", "multi_gpu:1\n" ] } ], "source": [ "config_path = 'config_full_yolo_fault_2_train.json'\n", "\n", "with open(config_path) as config_buffer:\n", " config = json.loads(config_buffer.read())\n", "\n", "###############################\n", "# Parse the annotations\n", "###############################\n", "train_ints, valid_ints, labels, max_box_per_image = create_training_instances(\n", " config['train']['train_annot_folder'],\n", " config['train']['train_image_folder'],\n", " config['train']['cache_name'],\n", " config['valid']['valid_annot_folder'],\n", " config['valid']['valid_image_folder'],\n", " config['valid']['cache_name'],\n", " config['model']['labels']\n", ")\n", "print('\\nTraining on: \\t' + str(labels) + '\\n')\n", "\n", "###############################\n", "# Create the generators\n", "###############################\n", "train_generator = BatchGenerator(\n", " instances = train_ints,\n", " anchors = config['model']['anchors'],\n", " labels = labels,\n", " downsample = 32, # ratio between network input's size and network output's size, 32 for YOLOv3\n", " max_box_per_image = max_box_per_image,\n", " batch_size = config['train']['batch_size'],\n", " min_net_size = config['model']['min_input_size'],\n", " max_net_size = config['model']['max_input_size'],\n", " shuffle = True,\n", " jitter = 0.3,\n", " norm = normalize\n", ")\n", "\n", "valid_generator = BatchGenerator(\n", " instances = valid_ints,\n", " anchors = config['model']['anchors'],\n", " labels = labels,\n", " downsample = 32, # ratio between network input's size and network output's size, 32 for YOLOv3\n", " max_box_per_image = max_box_per_image,\n", " batch_size = config['train']['batch_size'],\n", " min_net_size = config['model']['min_input_size'],\n", " max_net_size = config['model']['max_input_size'],\n", " shuffle = True,\n", " jitter = 0.0,\n", " norm = normalize\n", ")\n", "\n", "###############################\n", "# Create the model\n", "###############################\n", "if os.path.exists(config['train']['saved_weights_name']):\n", " config['train']['warmup_epochs'] = 0\n", "warmup_batches = config['train']['warmup_epochs'] * (config['train']['train_times']*len(train_generator))\n", "\n", "os.environ['CUDA_VISIBLE_DEVICES'] = config['train']['gpus']\n", "multi_gpu = len(config['train']['gpus'].split(','))\n", "print('multi_gpu:' + str(multi_gpu))" ] }, { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "WARNING:tensorflow:AutoGraph could not transform > and will run it as-is.\n", "Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.\n", "Cause: unexpected indent (, line 144)\n", "WARNING: AutoGraph could not transform > and will run it as-is.\n", "Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.\n", "Cause: unexpected indent (, line 144)\n", "WARNING:tensorflow:AutoGraph could not transform > and will run it as-is.\n", "Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.\n", "Cause: unexpected indent (, line 144)\n", "WARNING: AutoGraph could not transform > and will run it as-is.\n", "Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.\n", "Cause: unexpected indent (, line 144)\n", "WARNING:tensorflow:AutoGraph could not transform > and will run it as-is.\n", "Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.\n", "Cause: unexpected indent (, line 144)\n", "WARNING: AutoGraph could not transform > and will run it as-is.\n", "Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.\n", "Cause: unexpected indent (, line 144)\n" ] } ], "source": [ "train_model, infer_model = create_model(\n", " nb_class = len(labels),\n", " anchors = config['model']['anchors'],\n", " max_box_per_image = max_box_per_image,\n", " max_grid = [config['model']['max_input_size'], config['model']['max_input_size']],\n", " batch_size = config['train']['batch_size'],\n", " warmup_batches = warmup_batches,\n", " ignore_thresh = config['train']['ignore_thresh'],\n", " multi_gpu = len(config['train']['gpus'].split(',')),\n", " saved_weights_name = config['train']['saved_weights_name'],\n", " lr = config['train']['learning_rate'],\n", " grid_scales = config['train']['grid_scales'],\n", " obj_scale = config['train']['obj_scale'],\n", " noobj_scale = config['train']['noobj_scale'],\n", " xywh_scale = config['train']['xywh_scale'],\n", " class_scale = config['train']['class_scale'],\n", " backend\t\t = config['model']['backend']\n", " )" ] }, { "cell_type": "code", "execution_count": 5, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Model: \"model\"\n", "__________________________________________________________________________________________________\n", "Layer (type) Output Shape Param # Connected to \n", "==================================================================================================\n", "input_1 (InputLayer) [(None, None, None, 0 \n", "__________________________________________________________________________________________________\n", "conv_0 (Conv2D) (None, None, None, 3 864 input_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_0 (BatchNormalization) (None, None, None, 3 128 conv_0[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_0 (LeakyReLU) (None, None, None, 3 0 bnorm_0[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d (ZeroPadding2D) (None, None, None, 3 0 leaky_0[0][0] \n", "__________________________________________________________________________________________________\n", "conv_1 (Conv2D) (None, None, None, 6 18432 zero_padding2d[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_1 (BatchNormalization) (None, None, None, 6 256 conv_1[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_1 (LeakyReLU) (None, None, None, 6 0 bnorm_1[0][0] \n", "__________________________________________________________________________________________________\n", "conv_2 (Conv2D) (None, None, None, 3 2048 leaky_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_2 (BatchNormalization) (None, None, None, 3 128 conv_2[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_2 (LeakyReLU) (None, None, None, 3 0 bnorm_2[0][0] \n", "__________________________________________________________________________________________________\n", "conv_3 (Conv2D) (None, None, None, 6 18432 leaky_2[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_3 (BatchNormalization) (None, None, None, 6 256 conv_3[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_3 (LeakyReLU) (None, None, None, 6 0 bnorm_3[0][0] \n", "__________________________________________________________________________________________________\n", "add (Add) (None, None, None, 6 0 leaky_1[0][0] \n", " leaky_3[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_1 (ZeroPadding2D (None, None, None, 6 0 add[0][0] \n", "__________________________________________________________________________________________________\n", "conv_5 (Conv2D) (None, None, None, 1 73728 zero_padding2d_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_5 (BatchNormalization) (None, None, None, 1 512 conv_5[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_5 (LeakyReLU) (None, None, None, 1 0 bnorm_5[0][0] \n", "__________________________________________________________________________________________________\n", "conv_6 (Conv2D) (None, None, None, 6 8192 leaky_5[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_6 (BatchNormalization) (None, None, None, 6 256 conv_6[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_6 (LeakyReLU) (None, None, None, 6 0 bnorm_6[0][0] \n", "__________________________________________________________________________________________________\n", "conv_7 (Conv2D) (None, None, None, 1 73728 leaky_6[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_7 (BatchNormalization) (None, None, None, 1 512 conv_7[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_7 (LeakyReLU) (None, None, None, 1 0 bnorm_7[0][0] \n", "__________________________________________________________________________________________________\n", "add_1 (Add) (None, None, None, 1 0 leaky_5[0][0] \n", " leaky_7[0][0] \n", "__________________________________________________________________________________________________\n", "conv_9 (Conv2D) (None, None, None, 6 8192 add_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_9 (BatchNormalization) (None, None, None, 6 256 conv_9[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_9 (LeakyReLU) (None, None, None, 6 0 bnorm_9[0][0] \n", "__________________________________________________________________________________________________\n", "conv_10 (Conv2D) (None, None, None, 1 73728 leaky_9[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_10 (BatchNormalization) (None, None, None, 1 512 conv_10[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_10 (LeakyReLU) (None, None, None, 1 0 bnorm_10[0][0] \n", "__________________________________________________________________________________________________\n", "add_2 (Add) (None, None, None, 1 0 add_1[0][0] \n", " leaky_10[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_2 (ZeroPadding2D (None, None, None, 1 0 add_2[0][0] \n", "__________________________________________________________________________________________________\n", "conv_12 (Conv2D) (None, None, None, 2 294912 zero_padding2d_2[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_12 (BatchNormalization) (None, None, None, 2 1024 conv_12[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_12 (LeakyReLU) (None, None, None, 2 0 bnorm_12[0][0] \n", "__________________________________________________________________________________________________\n", "conv_13 (Conv2D) (None, None, None, 1 32768 leaky_12[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_13 (BatchNormalization) (None, None, None, 1 512 conv_13[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_13 (LeakyReLU) (None, None, None, 1 0 bnorm_13[0][0] \n", "__________________________________________________________________________________________________\n", "conv_14 (Conv2D) (None, None, None, 2 294912 leaky_13[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_14 (BatchNormalization) (None, None, None, 2 1024 conv_14[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_14 (LeakyReLU) (None, None, None, 2 0 bnorm_14[0][0] \n", "__________________________________________________________________________________________________\n", "add_3 (Add) (None, None, None, 2 0 leaky_12[0][0] \n", " leaky_14[0][0] \n", "__________________________________________________________________________________________________\n", "conv_16 (Conv2D) (None, None, None, 1 32768 add_3[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_16 (BatchNormalization) (None, None, None, 1 512 conv_16[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_16 (LeakyReLU) (None, None, None, 1 0 bnorm_16[0][0] \n", "__________________________________________________________________________________________________\n", "conv_17 (Conv2D) (None, None, None, 2 294912 leaky_16[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_17 (BatchNormalization) (None, None, None, 2 1024 conv_17[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_17 (LeakyReLU) (None, None, None, 2 0 bnorm_17[0][0] \n", "__________________________________________________________________________________________________\n", "add_4 (Add) (None, None, None, 2 0 add_3[0][0] \n", " leaky_17[0][0] \n", "__________________________________________________________________________________________________\n", "conv_19 (Conv2D) (None, None, None, 1 32768 add_4[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_19 (BatchNormalization) (None, None, None, 1 512 conv_19[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_19 (LeakyReLU) (None, None, None, 1 0 bnorm_19[0][0] \n", "__________________________________________________________________________________________________\n", "conv_20 (Conv2D) (None, None, None, 2 294912 leaky_19[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_20 (BatchNormalization) (None, None, None, 2 1024 conv_20[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_20 (LeakyReLU) (None, None, None, 2 0 bnorm_20[0][0] \n", "__________________________________________________________________________________________________\n", "add_5 (Add) (None, None, None, 2 0 add_4[0][0] \n", " leaky_20[0][0] \n", "__________________________________________________________________________________________________\n", "conv_22 (Conv2D) (None, None, None, 1 32768 add_5[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_22 (BatchNormalization) (None, None, None, 1 512 conv_22[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_22 (LeakyReLU) (None, None, None, 1 0 bnorm_22[0][0] \n", "__________________________________________________________________________________________________\n", "conv_23 (Conv2D) (None, None, None, 2 294912 leaky_22[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_23 (BatchNormalization) (None, None, None, 2 1024 conv_23[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_23 (LeakyReLU) (None, None, None, 2 0 bnorm_23[0][0] \n", "__________________________________________________________________________________________________\n", "add_6 (Add) (None, None, None, 2 0 add_5[0][0] \n", " leaky_23[0][0] \n", "__________________________________________________________________________________________________\n", "conv_25 (Conv2D) (None, None, None, 1 32768 add_6[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_25 (BatchNormalization) (None, None, None, 1 512 conv_25[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_25 (LeakyReLU) (None, None, None, 1 0 bnorm_25[0][0] \n", "__________________________________________________________________________________________________\n", "conv_26 (Conv2D) (None, None, None, 2 294912 leaky_25[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_26 (BatchNormalization) (None, None, None, 2 1024 conv_26[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_26 (LeakyReLU) (None, None, None, 2 0 bnorm_26[0][0] \n", "__________________________________________________________________________________________________\n", "add_7 (Add) (None, None, None, 2 0 add_6[0][0] \n", " leaky_26[0][0] \n", "__________________________________________________________________________________________________\n", "conv_28 (Conv2D) (None, None, None, 1 32768 add_7[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_28 (BatchNormalization) (None, None, None, 1 512 conv_28[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_28 (LeakyReLU) (None, None, None, 1 0 bnorm_28[0][0] \n", "__________________________________________________________________________________________________\n", "conv_29 (Conv2D) (None, None, None, 2 294912 leaky_28[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_29 (BatchNormalization) (None, None, None, 2 1024 conv_29[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_29 (LeakyReLU) (None, None, None, 2 0 bnorm_29[0][0] \n", "__________________________________________________________________________________________________\n", "add_8 (Add) (None, None, None, 2 0 add_7[0][0] \n", " leaky_29[0][0] \n", "__________________________________________________________________________________________________\n", "conv_31 (Conv2D) (None, None, None, 1 32768 add_8[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_31 (BatchNormalization) (None, None, None, 1 512 conv_31[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_31 (LeakyReLU) (None, None, None, 1 0 bnorm_31[0][0] \n", "__________________________________________________________________________________________________\n", "conv_32 (Conv2D) (None, None, None, 2 294912 leaky_31[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_32 (BatchNormalization) (None, None, None, 2 1024 conv_32[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_32 (LeakyReLU) (None, None, None, 2 0 bnorm_32[0][0] \n", "__________________________________________________________________________________________________\n", "add_9 (Add) (None, None, None, 2 0 add_8[0][0] \n", " leaky_32[0][0] \n", "__________________________________________________________________________________________________\n", "conv_34 (Conv2D) (None, None, None, 1 32768 add_9[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_34 (BatchNormalization) (None, None, None, 1 512 conv_34[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_34 (LeakyReLU) (None, None, None, 1 0 bnorm_34[0][0] \n", "__________________________________________________________________________________________________\n", "conv_35 (Conv2D) (None, None, None, 2 294912 leaky_34[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_35 (BatchNormalization) (None, None, None, 2 1024 conv_35[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_35 (LeakyReLU) (None, None, None, 2 0 bnorm_35[0][0] \n", "__________________________________________________________________________________________________\n", "add_10 (Add) (None, None, None, 2 0 add_9[0][0] \n", " leaky_35[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_3 (ZeroPadding2D (None, None, None, 2 0 add_10[0][0] \n", "__________________________________________________________________________________________________\n", "conv_37 (Conv2D) (None, None, None, 5 1179648 zero_padding2d_3[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_37 (BatchNormalization) (None, None, None, 5 2048 conv_37[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_37 (LeakyReLU) (None, None, None, 5 0 bnorm_37[0][0] \n", "__________________________________________________________________________________________________\n", "conv_38 (Conv2D) (None, None, None, 2 131072 leaky_37[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_38 (BatchNormalization) (None, None, None, 2 1024 conv_38[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_38 (LeakyReLU) (None, None, None, 2 0 bnorm_38[0][0] \n", "__________________________________________________________________________________________________\n", "conv_39 (Conv2D) (None, None, None, 5 1179648 leaky_38[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_39 (BatchNormalization) (None, None, None, 5 2048 conv_39[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_39 (LeakyReLU) (None, None, None, 5 0 bnorm_39[0][0] \n", "__________________________________________________________________________________________________\n", "add_11 (Add) (None, None, None, 5 0 leaky_37[0][0] \n", " leaky_39[0][0] \n", "__________________________________________________________________________________________________\n", "conv_41 (Conv2D) (None, None, None, 2 131072 add_11[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_41 (BatchNormalization) (None, None, None, 2 1024 conv_41[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_41 (LeakyReLU) (None, None, None, 2 0 bnorm_41[0][0] \n", "__________________________________________________________________________________________________\n", "conv_42 (Conv2D) (None, None, None, 5 1179648 leaky_41[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_42 (BatchNormalization) (None, None, None, 5 2048 conv_42[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_42 (LeakyReLU) (None, None, None, 5 0 bnorm_42[0][0] \n", "__________________________________________________________________________________________________\n", "add_12 (Add) (None, None, None, 5 0 add_11[0][0] \n", " leaky_42[0][0] \n", "__________________________________________________________________________________________________\n", "conv_44 (Conv2D) (None, None, None, 2 131072 add_12[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_44 (BatchNormalization) (None, None, None, 2 1024 conv_44[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_44 (LeakyReLU) (None, None, None, 2 0 bnorm_44[0][0] \n", "__________________________________________________________________________________________________\n", "conv_45 (Conv2D) (None, None, None, 5 1179648 leaky_44[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_45 (BatchNormalization) (None, None, None, 5 2048 conv_45[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_45 (LeakyReLU) (None, None, None, 5 0 bnorm_45[0][0] \n", "__________________________________________________________________________________________________\n", "add_13 (Add) (None, None, None, 5 0 add_12[0][0] \n", " leaky_45[0][0] \n", "__________________________________________________________________________________________________\n", "conv_47 (Conv2D) (None, None, None, 2 131072 add_13[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_47 (BatchNormalization) (None, None, None, 2 1024 conv_47[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_47 (LeakyReLU) (None, None, None, 2 0 bnorm_47[0][0] \n", "__________________________________________________________________________________________________\n", "conv_48 (Conv2D) (None, None, None, 5 1179648 leaky_47[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_48 (BatchNormalization) (None, None, None, 5 2048 conv_48[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_48 (LeakyReLU) (None, None, None, 5 0 bnorm_48[0][0] \n", "__________________________________________________________________________________________________\n", "add_14 (Add) (None, None, None, 5 0 add_13[0][0] \n", " leaky_48[0][0] \n", "__________________________________________________________________________________________________\n", "conv_50 (Conv2D) (None, None, None, 2 131072 add_14[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_50 (BatchNormalization) (None, None, None, 2 1024 conv_50[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_50 (LeakyReLU) (None, None, None, 2 0 bnorm_50[0][0] \n", "__________________________________________________________________________________________________\n", "conv_51 (Conv2D) (None, None, None, 5 1179648 leaky_50[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_51 (BatchNormalization) (None, None, None, 5 2048 conv_51[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_51 (LeakyReLU) (None, None, None, 5 0 bnorm_51[0][0] \n", "__________________________________________________________________________________________________\n", "add_15 (Add) (None, None, None, 5 0 add_14[0][0] \n", " leaky_51[0][0] \n", "__________________________________________________________________________________________________\n", "conv_53 (Conv2D) (None, None, None, 2 131072 add_15[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_53 (BatchNormalization) (None, None, None, 2 1024 conv_53[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_53 (LeakyReLU) (None, None, None, 2 0 bnorm_53[0][0] \n", "__________________________________________________________________________________________________\n", "conv_54 (Conv2D) (None, None, None, 5 1179648 leaky_53[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_54 (BatchNormalization) (None, None, None, 5 2048 conv_54[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_54 (LeakyReLU) (None, None, None, 5 0 bnorm_54[0][0] \n", "__________________________________________________________________________________________________\n", "add_16 (Add) (None, None, None, 5 0 add_15[0][0] \n", " leaky_54[0][0] \n", "__________________________________________________________________________________________________\n", "conv_56 (Conv2D) (None, None, None, 2 131072 add_16[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_56 (BatchNormalization) (None, None, None, 2 1024 conv_56[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_56 (LeakyReLU) (None, None, None, 2 0 bnorm_56[0][0] \n", "__________________________________________________________________________________________________\n", "conv_57 (Conv2D) (None, None, None, 5 1179648 leaky_56[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_57 (BatchNormalization) (None, None, None, 5 2048 conv_57[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_57 (LeakyReLU) (None, None, None, 5 0 bnorm_57[0][0] \n", "__________________________________________________________________________________________________\n", "add_17 (Add) (None, None, None, 5 0 add_16[0][0] \n", " leaky_57[0][0] \n", "__________________________________________________________________________________________________\n", "conv_59 (Conv2D) (None, None, None, 2 131072 add_17[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_59 (BatchNormalization) (None, None, None, 2 1024 conv_59[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_59 (LeakyReLU) (None, None, None, 2 0 bnorm_59[0][0] \n", "__________________________________________________________________________________________________\n", "conv_60 (Conv2D) (None, None, None, 5 1179648 leaky_59[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_60 (BatchNormalization) (None, None, None, 5 2048 conv_60[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_60 (LeakyReLU) (None, None, None, 5 0 bnorm_60[0][0] \n", "__________________________________________________________________________________________________\n", "add_18 (Add) (None, None, None, 5 0 add_17[0][0] \n", " leaky_60[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_4 (ZeroPadding2D (None, None, None, 5 0 add_18[0][0] \n", "__________________________________________________________________________________________________\n", "conv_62 (Conv2D) (None, None, None, 1 4718592 zero_padding2d_4[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_62 (BatchNormalization) (None, None, None, 1 4096 conv_62[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_62 (LeakyReLU) (None, None, None, 1 0 bnorm_62[0][0] \n", "__________________________________________________________________________________________________\n", "conv_63 (Conv2D) (None, None, None, 5 524288 leaky_62[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_63 (BatchNormalization) (None, None, None, 5 2048 conv_63[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_63 (LeakyReLU) (None, None, None, 5 0 bnorm_63[0][0] \n", "__________________________________________________________________________________________________\n", "conv_64 (Conv2D) (None, None, None, 1 4718592 leaky_63[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_64 (BatchNormalization) (None, None, None, 1 4096 conv_64[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_64 (LeakyReLU) (None, None, None, 1 0 bnorm_64[0][0] \n", "__________________________________________________________________________________________________\n", "add_19 (Add) (None, None, None, 1 0 leaky_62[0][0] \n", " leaky_64[0][0] \n", "__________________________________________________________________________________________________\n", "conv_66 (Conv2D) (None, None, None, 5 524288 add_19[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_66 (BatchNormalization) (None, None, None, 5 2048 conv_66[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_66 (LeakyReLU) (None, None, None, 5 0 bnorm_66[0][0] \n", "__________________________________________________________________________________________________\n", "conv_67 (Conv2D) (None, None, None, 1 4718592 leaky_66[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_67 (BatchNormalization) (None, None, None, 1 4096 conv_67[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_67 (LeakyReLU) (None, None, None, 1 0 bnorm_67[0][0] \n", "__________________________________________________________________________________________________\n", "add_20 (Add) (None, None, None, 1 0 add_19[0][0] \n", " leaky_67[0][0] \n", "__________________________________________________________________________________________________\n", "conv_69 (Conv2D) (None, None, None, 5 524288 add_20[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_69 (BatchNormalization) (None, None, None, 5 2048 conv_69[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_69 (LeakyReLU) (None, None, None, 5 0 bnorm_69[0][0] \n", "__________________________________________________________________________________________________\n", "conv_70 (Conv2D) (None, None, None, 1 4718592 leaky_69[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_70 (BatchNormalization) (None, None, None, 1 4096 conv_70[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_70 (LeakyReLU) (None, None, None, 1 0 bnorm_70[0][0] \n", "__________________________________________________________________________________________________\n", "add_21 (Add) (None, None, None, 1 0 add_20[0][0] \n", " leaky_70[0][0] \n", "__________________________________________________________________________________________________\n", "conv_72 (Conv2D) (None, None, None, 5 524288 add_21[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_72 (BatchNormalization) (None, None, None, 5 2048 conv_72[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_72 (LeakyReLU) (None, None, None, 5 0 bnorm_72[0][0] \n", "__________________________________________________________________________________________________\n", "conv_73 (Conv2D) (None, None, None, 1 4718592 leaky_72[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_73 (BatchNormalization) (None, None, None, 1 4096 conv_73[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_73 (LeakyReLU) (None, None, None, 1 0 bnorm_73[0][0] \n", "__________________________________________________________________________________________________\n", "add_22 (Add) (None, None, None, 1 0 add_21[0][0] \n", " leaky_73[0][0] \n", "__________________________________________________________________________________________________\n", "conv_75 (Conv2D) (None, None, None, 5 524288 add_22[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_75 (BatchNormalization) (None, None, None, 5 2048 conv_75[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_75 (LeakyReLU) (None, None, None, 5 0 bnorm_75[0][0] \n", "__________________________________________________________________________________________________\n", "conv_76 (Conv2D) (None, None, None, 1 4718592 leaky_75[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_76 (BatchNormalization) (None, None, None, 1 4096 conv_76[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_76 (LeakyReLU) (None, None, None, 1 0 bnorm_76[0][0] \n", "__________________________________________________________________________________________________\n", "conv_77 (Conv2D) (None, None, None, 5 524288 leaky_76[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_77 (BatchNormalization) (None, None, None, 5 2048 conv_77[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_77 (LeakyReLU) (None, None, None, 5 0 bnorm_77[0][0] \n", "__________________________________________________________________________________________________\n", "conv_78 (Conv2D) (None, None, None, 1 4718592 leaky_77[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_78 (BatchNormalization) (None, None, None, 1 4096 conv_78[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_78 (LeakyReLU) (None, None, None, 1 0 bnorm_78[0][0] \n", "__________________________________________________________________________________________________\n", "conv_79 (Conv2D) (None, None, None, 5 524288 leaky_78[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_79 (BatchNormalization) (None, None, None, 5 2048 conv_79[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_79 (LeakyReLU) (None, None, None, 5 0 bnorm_79[0][0] \n", "__________________________________________________________________________________________________\n", "conv_84 (Conv2D) (None, None, None, 2 131072 leaky_79[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_84 (BatchNormalization) (None, None, None, 2 1024 conv_84[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_84 (LeakyReLU) (None, None, None, 2 0 bnorm_84[0][0] \n", "__________________________________________________________________________________________________\n", "up_sampling2d (UpSampling2D) (None, None, None, 2 0 leaky_84[0][0] \n", "__________________________________________________________________________________________________\n", "concatenate (Concatenate) (None, None, None, 7 0 up_sampling2d[0][0] \n", " add_18[0][0] \n", "__________________________________________________________________________________________________\n", "conv_87 (Conv2D) (None, None, None, 2 196608 concatenate[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_87 (BatchNormalization) (None, None, None, 2 1024 conv_87[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_87 (LeakyReLU) (None, None, None, 2 0 bnorm_87[0][0] \n", "__________________________________________________________________________________________________\n", "conv_88 (Conv2D) (None, None, None, 5 1179648 leaky_87[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_88 (BatchNormalization) (None, None, None, 5 2048 conv_88[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_88 (LeakyReLU) (None, None, None, 5 0 bnorm_88[0][0] \n", "__________________________________________________________________________________________________\n", "conv_89 (Conv2D) (None, None, None, 2 131072 leaky_88[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_89 (BatchNormalization) (None, None, None, 2 1024 conv_89[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_89 (LeakyReLU) (None, None, None, 2 0 bnorm_89[0][0] \n", "__________________________________________________________________________________________________\n", "conv_90 (Conv2D) (None, None, None, 5 1179648 leaky_89[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_90 (BatchNormalization) (None, None, None, 5 2048 conv_90[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_90 (LeakyReLU) (None, None, None, 5 0 bnorm_90[0][0] \n", "__________________________________________________________________________________________________\n", "conv_91 (Conv2D) (None, None, None, 2 131072 leaky_90[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_91 (BatchNormalization) (None, None, None, 2 1024 conv_91[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_91 (LeakyReLU) (None, None, None, 2 0 bnorm_91[0][0] \n", "__________________________________________________________________________________________________\n", "conv_96 (Conv2D) (None, None, None, 1 32768 leaky_91[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_96 (BatchNormalization) (None, None, None, 1 512 conv_96[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_96 (LeakyReLU) (None, None, None, 1 0 bnorm_96[0][0] \n", "__________________________________________________________________________________________________\n", "up_sampling2d_1 (UpSampling2D) (None, None, None, 1 0 leaky_96[0][0] \n", "__________________________________________________________________________________________________\n", "concatenate_1 (Concatenate) (None, None, None, 3 0 up_sampling2d_1[0][0] \n", " add_10[0][0] \n", "__________________________________________________________________________________________________\n", "conv_99 (Conv2D) (None, None, None, 1 49152 concatenate_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_99 (BatchNormalization) (None, None, None, 1 512 conv_99[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_99 (LeakyReLU) (None, None, None, 1 0 bnorm_99[0][0] \n", "__________________________________________________________________________________________________\n", "conv_100 (Conv2D) (None, None, None, 2 294912 leaky_99[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_100 (BatchNormalization) (None, None, None, 2 1024 conv_100[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_100 (LeakyReLU) (None, None, None, 2 0 bnorm_100[0][0] \n", "__________________________________________________________________________________________________\n", "conv_101 (Conv2D) (None, None, None, 1 32768 leaky_100[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_101 (BatchNormalization) (None, None, None, 1 512 conv_101[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_101 (LeakyReLU) (None, None, None, 1 0 bnorm_101[0][0] \n", "__________________________________________________________________________________________________\n", "conv_102 (Conv2D) (None, None, None, 2 294912 leaky_101[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_102 (BatchNormalization) (None, None, None, 2 1024 conv_102[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_102 (LeakyReLU) (None, None, None, 2 0 bnorm_102[0][0] \n", "__________________________________________________________________________________________________\n", "conv_103 (Conv2D) (None, None, None, 1 32768 leaky_102[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_103 (BatchNormalization) (None, None, None, 1 512 conv_103[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_103 (LeakyReLU) (None, None, None, 1 0 bnorm_103[0][0] \n", "__________________________________________________________________________________________________\n", "conv_80 (Conv2D) (None, None, None, 1 4718592 leaky_79[0][0] \n", "__________________________________________________________________________________________________\n", "conv_92 (Conv2D) (None, None, None, 5 1179648 leaky_91[0][0] \n", "__________________________________________________________________________________________________\n", "conv_104 (Conv2D) (None, None, None, 2 294912 leaky_103[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_80 (BatchNormalization) (None, None, None, 1 4096 conv_80[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_92 (BatchNormalization) (None, None, None, 5 2048 conv_92[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_104 (BatchNormalization) (None, None, None, 2 1024 conv_104[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_80 (LeakyReLU) (None, None, None, 1 0 bnorm_80[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_92 (LeakyReLU) (None, None, None, 5 0 bnorm_92[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_104 (LeakyReLU) (None, None, None, 2 0 bnorm_104[0][0] \n", "__________________________________________________________________________________________________\n", "conv_81 (Conv2D) (None, None, None, 1 18450 leaky_80[0][0] \n", "__________________________________________________________________________________________________\n", "input_3 (InputLayer) [(None, None, None, 0 \n", "__________________________________________________________________________________________________\n", "input_2 (InputLayer) [(None, 1, 1, 1, 5, 0 \n", "__________________________________________________________________________________________________\n", "conv_93 (Conv2D) (None, None, None, 1 9234 leaky_92[0][0] \n", "__________________________________________________________________________________________________\n", "input_4 (InputLayer) [(None, None, None, 0 \n", "__________________________________________________________________________________________________\n", "conv_105 (Conv2D) (None, None, None, 1 4626 leaky_104[0][0] \n", "__________________________________________________________________________________________________\n", "input_5 (InputLayer) [(None, None, None, 0 \n", "__________________________________________________________________________________________________\n", "yolo_layer (YoloLayer) (2,) 0 input_1[0][0] \n", " conv_81[0][0] \n", " input_3[0][0] \n", " input_2[0][0] \n", "__________________________________________________________________________________________________\n", "yolo_layer_1 (YoloLayer) (2,) 0 input_1[0][0] \n", " conv_93[0][0] \n", " input_4[0][0] \n", " input_2[0][0] \n", "__________________________________________________________________________________________________\n", "yolo_layer_2 (YoloLayer) (2,) 0 input_1[0][0] \n", " conv_105[0][0] \n", " input_5[0][0] \n", " input_2[0][0] \n", "==================================================================================================\n", "Total params: 61,576,342\n", "Trainable params: 61,523,734\n", "Non-trainable params: 52,608\n", "__________________________________________________________________________________________________\n" ] } ], "source": [ "train_model.summary()" ] }, { "cell_type": "code", "execution_count": 6, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Model: \"model_1\"\n", "__________________________________________________________________________________________________\n", "Layer (type) Output Shape Param # Connected to \n", "==================================================================================================\n", "input_1 (InputLayer) [(None, None, None, 0 \n", "__________________________________________________________________________________________________\n", "conv_0 (Conv2D) (None, None, None, 3 864 input_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_0 (BatchNormalization) (None, None, None, 3 128 conv_0[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_0 (LeakyReLU) (None, None, None, 3 0 bnorm_0[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d (ZeroPadding2D) (None, None, None, 3 0 leaky_0[0][0] \n", "__________________________________________________________________________________________________\n", "conv_1 (Conv2D) (None, None, None, 6 18432 zero_padding2d[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_1 (BatchNormalization) (None, None, None, 6 256 conv_1[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_1 (LeakyReLU) (None, None, None, 6 0 bnorm_1[0][0] \n", "__________________________________________________________________________________________________\n", "conv_2 (Conv2D) (None, None, None, 3 2048 leaky_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_2 (BatchNormalization) (None, None, None, 3 128 conv_2[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_2 (LeakyReLU) (None, None, None, 3 0 bnorm_2[0][0] \n", "__________________________________________________________________________________________________\n", "conv_3 (Conv2D) (None, None, None, 6 18432 leaky_2[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_3 (BatchNormalization) (None, None, None, 6 256 conv_3[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_3 (LeakyReLU) (None, None, None, 6 0 bnorm_3[0][0] \n", "__________________________________________________________________________________________________\n", "add (Add) (None, None, None, 6 0 leaky_1[0][0] \n", " leaky_3[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_1 (ZeroPadding2D (None, None, None, 6 0 add[0][0] \n", "__________________________________________________________________________________________________\n", "conv_5 (Conv2D) (None, None, None, 1 73728 zero_padding2d_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_5 (BatchNormalization) (None, None, None, 1 512 conv_5[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_5 (LeakyReLU) (None, None, None, 1 0 bnorm_5[0][0] \n", "__________________________________________________________________________________________________\n", "conv_6 (Conv2D) (None, None, None, 6 8192 leaky_5[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_6 (BatchNormalization) (None, None, None, 6 256 conv_6[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_6 (LeakyReLU) (None, None, None, 6 0 bnorm_6[0][0] \n", "__________________________________________________________________________________________________\n", "conv_7 (Conv2D) (None, None, None, 1 73728 leaky_6[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_7 (BatchNormalization) (None, None, None, 1 512 conv_7[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_7 (LeakyReLU) (None, None, None, 1 0 bnorm_7[0][0] \n", "__________________________________________________________________________________________________\n", "add_1 (Add) (None, None, None, 1 0 leaky_5[0][0] \n", " leaky_7[0][0] \n", "__________________________________________________________________________________________________\n", "conv_9 (Conv2D) (None, None, None, 6 8192 add_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_9 (BatchNormalization) (None, None, None, 6 256 conv_9[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_9 (LeakyReLU) (None, None, None, 6 0 bnorm_9[0][0] \n", "__________________________________________________________________________________________________\n", "conv_10 (Conv2D) (None, None, None, 1 73728 leaky_9[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_10 (BatchNormalization) (None, None, None, 1 512 conv_10[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_10 (LeakyReLU) (None, None, None, 1 0 bnorm_10[0][0] \n", "__________________________________________________________________________________________________\n", "add_2 (Add) (None, None, None, 1 0 add_1[0][0] \n", " leaky_10[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_2 (ZeroPadding2D (None, None, None, 1 0 add_2[0][0] \n", "__________________________________________________________________________________________________\n", "conv_12 (Conv2D) (None, None, None, 2 294912 zero_padding2d_2[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_12 (BatchNormalization) (None, None, None, 2 1024 conv_12[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_12 (LeakyReLU) (None, None, None, 2 0 bnorm_12[0][0] \n", "__________________________________________________________________________________________________\n", "conv_13 (Conv2D) (None, None, None, 1 32768 leaky_12[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_13 (BatchNormalization) (None, None, None, 1 512 conv_13[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_13 (LeakyReLU) (None, None, None, 1 0 bnorm_13[0][0] \n", "__________________________________________________________________________________________________\n", "conv_14 (Conv2D) (None, None, None, 2 294912 leaky_13[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_14 (BatchNormalization) (None, None, None, 2 1024 conv_14[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_14 (LeakyReLU) (None, None, None, 2 0 bnorm_14[0][0] \n", "__________________________________________________________________________________________________\n", "add_3 (Add) (None, None, None, 2 0 leaky_12[0][0] \n", " leaky_14[0][0] \n", "__________________________________________________________________________________________________\n", "conv_16 (Conv2D) (None, None, None, 1 32768 add_3[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_16 (BatchNormalization) (None, None, None, 1 512 conv_16[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_16 (LeakyReLU) (None, None, None, 1 0 bnorm_16[0][0] \n", "__________________________________________________________________________________________________\n", "conv_17 (Conv2D) (None, None, None, 2 294912 leaky_16[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_17 (BatchNormalization) (None, None, None, 2 1024 conv_17[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_17 (LeakyReLU) (None, None, None, 2 0 bnorm_17[0][0] \n", "__________________________________________________________________________________________________\n", "add_4 (Add) (None, None, None, 2 0 add_3[0][0] \n", " leaky_17[0][0] \n", "__________________________________________________________________________________________________\n", "conv_19 (Conv2D) (None, None, None, 1 32768 add_4[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_19 (BatchNormalization) (None, None, None, 1 512 conv_19[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_19 (LeakyReLU) (None, None, None, 1 0 bnorm_19[0][0] \n", "__________________________________________________________________________________________________\n", "conv_20 (Conv2D) (None, None, None, 2 294912 leaky_19[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_20 (BatchNormalization) (None, None, None, 2 1024 conv_20[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_20 (LeakyReLU) (None, None, None, 2 0 bnorm_20[0][0] \n", "__________________________________________________________________________________________________\n", "add_5 (Add) (None, None, None, 2 0 add_4[0][0] \n", " leaky_20[0][0] \n", "__________________________________________________________________________________________________\n", "conv_22 (Conv2D) (None, None, None, 1 32768 add_5[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_22 (BatchNormalization) (None, None, None, 1 512 conv_22[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_22 (LeakyReLU) (None, None, None, 1 0 bnorm_22[0][0] \n", "__________________________________________________________________________________________________\n", "conv_23 (Conv2D) (None, None, None, 2 294912 leaky_22[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_23 (BatchNormalization) (None, None, None, 2 1024 conv_23[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_23 (LeakyReLU) (None, None, None, 2 0 bnorm_23[0][0] \n", "__________________________________________________________________________________________________\n", "add_6 (Add) (None, None, None, 2 0 add_5[0][0] \n", " leaky_23[0][0] \n", "__________________________________________________________________________________________________\n", "conv_25 (Conv2D) (None, None, None, 1 32768 add_6[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_25 (BatchNormalization) (None, None, None, 1 512 conv_25[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_25 (LeakyReLU) (None, None, None, 1 0 bnorm_25[0][0] \n", "__________________________________________________________________________________________________\n", "conv_26 (Conv2D) (None, None, None, 2 294912 leaky_25[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_26 (BatchNormalization) (None, None, None, 2 1024 conv_26[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_26 (LeakyReLU) (None, None, None, 2 0 bnorm_26[0][0] \n", "__________________________________________________________________________________________________\n", "add_7 (Add) (None, None, None, 2 0 add_6[0][0] \n", " leaky_26[0][0] \n", "__________________________________________________________________________________________________\n", "conv_28 (Conv2D) (None, None, None, 1 32768 add_7[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_28 (BatchNormalization) (None, None, None, 1 512 conv_28[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_28 (LeakyReLU) (None, None, None, 1 0 bnorm_28[0][0] \n", "__________________________________________________________________________________________________\n", "conv_29 (Conv2D) (None, None, None, 2 294912 leaky_28[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_29 (BatchNormalization) (None, None, None, 2 1024 conv_29[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_29 (LeakyReLU) (None, None, None, 2 0 bnorm_29[0][0] \n", "__________________________________________________________________________________________________\n", "add_8 (Add) (None, None, None, 2 0 add_7[0][0] \n", " leaky_29[0][0] \n", "__________________________________________________________________________________________________\n", "conv_31 (Conv2D) (None, None, None, 1 32768 add_8[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_31 (BatchNormalization) (None, None, None, 1 512 conv_31[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_31 (LeakyReLU) (None, None, None, 1 0 bnorm_31[0][0] \n", "__________________________________________________________________________________________________\n", "conv_32 (Conv2D) (None, None, None, 2 294912 leaky_31[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_32 (BatchNormalization) (None, None, None, 2 1024 conv_32[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_32 (LeakyReLU) (None, None, None, 2 0 bnorm_32[0][0] \n", "__________________________________________________________________________________________________\n", "add_9 (Add) (None, None, None, 2 0 add_8[0][0] \n", " leaky_32[0][0] \n", "__________________________________________________________________________________________________\n", "conv_34 (Conv2D) (None, None, None, 1 32768 add_9[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_34 (BatchNormalization) (None, None, None, 1 512 conv_34[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_34 (LeakyReLU) (None, None, None, 1 0 bnorm_34[0][0] \n", "__________________________________________________________________________________________________\n", "conv_35 (Conv2D) (None, None, None, 2 294912 leaky_34[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_35 (BatchNormalization) (None, None, None, 2 1024 conv_35[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_35 (LeakyReLU) (None, None, None, 2 0 bnorm_35[0][0] \n", "__________________________________________________________________________________________________\n", "add_10 (Add) (None, None, None, 2 0 add_9[0][0] \n", " leaky_35[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_3 (ZeroPadding2D (None, None, None, 2 0 add_10[0][0] \n", "__________________________________________________________________________________________________\n", "conv_37 (Conv2D) (None, None, None, 5 1179648 zero_padding2d_3[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_37 (BatchNormalization) (None, None, None, 5 2048 conv_37[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_37 (LeakyReLU) (None, None, None, 5 0 bnorm_37[0][0] \n", "__________________________________________________________________________________________________\n", "conv_38 (Conv2D) (None, None, None, 2 131072 leaky_37[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_38 (BatchNormalization) (None, None, None, 2 1024 conv_38[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_38 (LeakyReLU) (None, None, None, 2 0 bnorm_38[0][0] \n", "__________________________________________________________________________________________________\n", "conv_39 (Conv2D) (None, None, None, 5 1179648 leaky_38[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_39 (BatchNormalization) (None, None, None, 5 2048 conv_39[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_39 (LeakyReLU) (None, None, None, 5 0 bnorm_39[0][0] \n", "__________________________________________________________________________________________________\n", "add_11 (Add) (None, None, None, 5 0 leaky_37[0][0] \n", " leaky_39[0][0] \n", "__________________________________________________________________________________________________\n", "conv_41 (Conv2D) (None, None, None, 2 131072 add_11[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_41 (BatchNormalization) (None, None, None, 2 1024 conv_41[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_41 (LeakyReLU) (None, None, None, 2 0 bnorm_41[0][0] \n", "__________________________________________________________________________________________________\n", "conv_42 (Conv2D) (None, None, None, 5 1179648 leaky_41[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_42 (BatchNormalization) (None, None, None, 5 2048 conv_42[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_42 (LeakyReLU) (None, None, None, 5 0 bnorm_42[0][0] \n", "__________________________________________________________________________________________________\n", "add_12 (Add) (None, None, None, 5 0 add_11[0][0] \n", " leaky_42[0][0] \n", "__________________________________________________________________________________________________\n", "conv_44 (Conv2D) (None, None, None, 2 131072 add_12[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_44 (BatchNormalization) (None, None, None, 2 1024 conv_44[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_44 (LeakyReLU) (None, None, None, 2 0 bnorm_44[0][0] \n", "__________________________________________________________________________________________________\n", "conv_45 (Conv2D) (None, None, None, 5 1179648 leaky_44[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_45 (BatchNormalization) (None, None, None, 5 2048 conv_45[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_45 (LeakyReLU) (None, None, None, 5 0 bnorm_45[0][0] \n", "__________________________________________________________________________________________________\n", "add_13 (Add) (None, None, None, 5 0 add_12[0][0] \n", " leaky_45[0][0] \n", "__________________________________________________________________________________________________\n", "conv_47 (Conv2D) (None, None, None, 2 131072 add_13[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_47 (BatchNormalization) (None, None, None, 2 1024 conv_47[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_47 (LeakyReLU) (None, None, None, 2 0 bnorm_47[0][0] \n", "__________________________________________________________________________________________________\n", "conv_48 (Conv2D) (None, None, None, 5 1179648 leaky_47[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_48 (BatchNormalization) (None, None, None, 5 2048 conv_48[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_48 (LeakyReLU) (None, None, None, 5 0 bnorm_48[0][0] \n", "__________________________________________________________________________________________________\n", "add_14 (Add) (None, None, None, 5 0 add_13[0][0] \n", " leaky_48[0][0] \n", "__________________________________________________________________________________________________\n", "conv_50 (Conv2D) (None, None, None, 2 131072 add_14[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_50 (BatchNormalization) (None, None, None, 2 1024 conv_50[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_50 (LeakyReLU) (None, None, None, 2 0 bnorm_50[0][0] \n", "__________________________________________________________________________________________________\n", "conv_51 (Conv2D) (None, None, None, 5 1179648 leaky_50[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_51 (BatchNormalization) (None, None, None, 5 2048 conv_51[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_51 (LeakyReLU) (None, None, None, 5 0 bnorm_51[0][0] \n", "__________________________________________________________________________________________________\n", "add_15 (Add) (None, None, None, 5 0 add_14[0][0] \n", " leaky_51[0][0] \n", "__________________________________________________________________________________________________\n", "conv_53 (Conv2D) (None, None, None, 2 131072 add_15[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_53 (BatchNormalization) (None, None, None, 2 1024 conv_53[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_53 (LeakyReLU) (None, None, None, 2 0 bnorm_53[0][0] \n", "__________________________________________________________________________________________________\n", "conv_54 (Conv2D) (None, None, None, 5 1179648 leaky_53[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_54 (BatchNormalization) (None, None, None, 5 2048 conv_54[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_54 (LeakyReLU) (None, None, None, 5 0 bnorm_54[0][0] \n", "__________________________________________________________________________________________________\n", "add_16 (Add) (None, None, None, 5 0 add_15[0][0] \n", " leaky_54[0][0] \n", "__________________________________________________________________________________________________\n", "conv_56 (Conv2D) (None, None, None, 2 131072 add_16[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_56 (BatchNormalization) (None, None, None, 2 1024 conv_56[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_56 (LeakyReLU) (None, None, None, 2 0 bnorm_56[0][0] \n", "__________________________________________________________________________________________________\n", "conv_57 (Conv2D) (None, None, None, 5 1179648 leaky_56[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_57 (BatchNormalization) (None, None, None, 5 2048 conv_57[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_57 (LeakyReLU) (None, None, None, 5 0 bnorm_57[0][0] \n", "__________________________________________________________________________________________________\n", "add_17 (Add) (None, None, None, 5 0 add_16[0][0] \n", " leaky_57[0][0] \n", "__________________________________________________________________________________________________\n", "conv_59 (Conv2D) (None, None, None, 2 131072 add_17[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_59 (BatchNormalization) (None, None, None, 2 1024 conv_59[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_59 (LeakyReLU) (None, None, None, 2 0 bnorm_59[0][0] \n", "__________________________________________________________________________________________________\n", "conv_60 (Conv2D) (None, None, None, 5 1179648 leaky_59[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_60 (BatchNormalization) (None, None, None, 5 2048 conv_60[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_60 (LeakyReLU) (None, None, None, 5 0 bnorm_60[0][0] \n", "__________________________________________________________________________________________________\n", "add_18 (Add) (None, None, None, 5 0 add_17[0][0] \n", " leaky_60[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_4 (ZeroPadding2D (None, None, None, 5 0 add_18[0][0] \n", "__________________________________________________________________________________________________\n", "conv_62 (Conv2D) (None, None, None, 1 4718592 zero_padding2d_4[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_62 (BatchNormalization) (None, None, None, 1 4096 conv_62[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_62 (LeakyReLU) (None, None, None, 1 0 bnorm_62[0][0] \n", "__________________________________________________________________________________________________\n", "conv_63 (Conv2D) (None, None, None, 5 524288 leaky_62[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_63 (BatchNormalization) (None, None, None, 5 2048 conv_63[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_63 (LeakyReLU) (None, None, None, 5 0 bnorm_63[0][0] \n", "__________________________________________________________________________________________________\n", "conv_64 (Conv2D) (None, None, None, 1 4718592 leaky_63[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_64 (BatchNormalization) (None, None, None, 1 4096 conv_64[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_64 (LeakyReLU) (None, None, None, 1 0 bnorm_64[0][0] \n", "__________________________________________________________________________________________________\n", "add_19 (Add) (None, None, None, 1 0 leaky_62[0][0] \n", " leaky_64[0][0] \n", "__________________________________________________________________________________________________\n", "conv_66 (Conv2D) (None, None, None, 5 524288 add_19[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_66 (BatchNormalization) (None, None, None, 5 2048 conv_66[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_66 (LeakyReLU) (None, None, None, 5 0 bnorm_66[0][0] \n", "__________________________________________________________________________________________________\n", "conv_67 (Conv2D) (None, None, None, 1 4718592 leaky_66[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_67 (BatchNormalization) (None, None, None, 1 4096 conv_67[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_67 (LeakyReLU) (None, None, None, 1 0 bnorm_67[0][0] \n", "__________________________________________________________________________________________________\n", "add_20 (Add) (None, None, None, 1 0 add_19[0][0] \n", " leaky_67[0][0] \n", "__________________________________________________________________________________________________\n", "conv_69 (Conv2D) (None, None, None, 5 524288 add_20[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_69 (BatchNormalization) (None, None, None, 5 2048 conv_69[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_69 (LeakyReLU) (None, None, None, 5 0 bnorm_69[0][0] \n", "__________________________________________________________________________________________________\n", "conv_70 (Conv2D) (None, None, None, 1 4718592 leaky_69[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_70 (BatchNormalization) (None, None, None, 1 4096 conv_70[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_70 (LeakyReLU) (None, None, None, 1 0 bnorm_70[0][0] \n", "__________________________________________________________________________________________________\n", "add_21 (Add) (None, None, None, 1 0 add_20[0][0] \n", " leaky_70[0][0] \n", "__________________________________________________________________________________________________\n", "conv_72 (Conv2D) (None, None, None, 5 524288 add_21[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_72 (BatchNormalization) (None, None, None, 5 2048 conv_72[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_72 (LeakyReLU) (None, None, None, 5 0 bnorm_72[0][0] \n", "__________________________________________________________________________________________________\n", "conv_73 (Conv2D) (None, None, None, 1 4718592 leaky_72[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_73 (BatchNormalization) (None, None, None, 1 4096 conv_73[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_73 (LeakyReLU) (None, None, None, 1 0 bnorm_73[0][0] \n", "__________________________________________________________________________________________________\n", "add_22 (Add) (None, None, None, 1 0 add_21[0][0] \n", " leaky_73[0][0] \n", "__________________________________________________________________________________________________\n", "conv_75 (Conv2D) (None, None, None, 5 524288 add_22[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_75 (BatchNormalization) (None, None, None, 5 2048 conv_75[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_75 (LeakyReLU) (None, None, None, 5 0 bnorm_75[0][0] \n", "__________________________________________________________________________________________________\n", "conv_76 (Conv2D) (None, None, None, 1 4718592 leaky_75[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_76 (BatchNormalization) (None, None, None, 1 4096 conv_76[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_76 (LeakyReLU) (None, None, None, 1 0 bnorm_76[0][0] \n", "__________________________________________________________________________________________________\n", "conv_77 (Conv2D) (None, None, None, 5 524288 leaky_76[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_77 (BatchNormalization) (None, None, None, 5 2048 conv_77[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_77 (LeakyReLU) (None, None, None, 5 0 bnorm_77[0][0] \n", "__________________________________________________________________________________________________\n", "conv_78 (Conv2D) (None, None, None, 1 4718592 leaky_77[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_78 (BatchNormalization) (None, None, None, 1 4096 conv_78[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_78 (LeakyReLU) (None, None, None, 1 0 bnorm_78[0][0] \n", "__________________________________________________________________________________________________\n", "conv_79 (Conv2D) (None, None, None, 5 524288 leaky_78[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_79 (BatchNormalization) (None, None, None, 5 2048 conv_79[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_79 (LeakyReLU) (None, None, None, 5 0 bnorm_79[0][0] \n", "__________________________________________________________________________________________________\n", "conv_84 (Conv2D) (None, None, None, 2 131072 leaky_79[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_84 (BatchNormalization) (None, None, None, 2 1024 conv_84[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_84 (LeakyReLU) (None, None, None, 2 0 bnorm_84[0][0] \n", "__________________________________________________________________________________________________\n", "up_sampling2d (UpSampling2D) (None, None, None, 2 0 leaky_84[0][0] \n", "__________________________________________________________________________________________________\n", "concatenate (Concatenate) (None, None, None, 7 0 up_sampling2d[0][0] \n", " add_18[0][0] \n", "__________________________________________________________________________________________________\n", "conv_87 (Conv2D) (None, None, None, 2 196608 concatenate[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_87 (BatchNormalization) (None, None, None, 2 1024 conv_87[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_87 (LeakyReLU) (None, None, None, 2 0 bnorm_87[0][0] \n", "__________________________________________________________________________________________________\n", "conv_88 (Conv2D) (None, None, None, 5 1179648 leaky_87[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_88 (BatchNormalization) (None, None, None, 5 2048 conv_88[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_88 (LeakyReLU) (None, None, None, 5 0 bnorm_88[0][0] \n", "__________________________________________________________________________________________________\n", "conv_89 (Conv2D) (None, None, None, 2 131072 leaky_88[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_89 (BatchNormalization) (None, None, None, 2 1024 conv_89[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_89 (LeakyReLU) (None, None, None, 2 0 bnorm_89[0][0] \n", "__________________________________________________________________________________________________\n", "conv_90 (Conv2D) (None, None, None, 5 1179648 leaky_89[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_90 (BatchNormalization) (None, None, None, 5 2048 conv_90[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_90 (LeakyReLU) (None, None, None, 5 0 bnorm_90[0][0] \n", "__________________________________________________________________________________________________\n", "conv_91 (Conv2D) (None, None, None, 2 131072 leaky_90[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_91 (BatchNormalization) (None, None, None, 2 1024 conv_91[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_91 (LeakyReLU) (None, None, None, 2 0 bnorm_91[0][0] \n", "__________________________________________________________________________________________________\n", "conv_96 (Conv2D) (None, None, None, 1 32768 leaky_91[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_96 (BatchNormalization) (None, None, None, 1 512 conv_96[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_96 (LeakyReLU) (None, None, None, 1 0 bnorm_96[0][0] \n", "__________________________________________________________________________________________________\n", "up_sampling2d_1 (UpSampling2D) (None, None, None, 1 0 leaky_96[0][0] \n", "__________________________________________________________________________________________________\n", "concatenate_1 (Concatenate) (None, None, None, 3 0 up_sampling2d_1[0][0] \n", " add_10[0][0] \n", "__________________________________________________________________________________________________\n", "conv_99 (Conv2D) (None, None, None, 1 49152 concatenate_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_99 (BatchNormalization) (None, None, None, 1 512 conv_99[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_99 (LeakyReLU) (None, None, None, 1 0 bnorm_99[0][0] \n", "__________________________________________________________________________________________________\n", "conv_100 (Conv2D) (None, None, None, 2 294912 leaky_99[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_100 (BatchNormalization) (None, None, None, 2 1024 conv_100[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_100 (LeakyReLU) (None, None, None, 2 0 bnorm_100[0][0] \n", "__________________________________________________________________________________________________\n", "conv_101 (Conv2D) (None, None, None, 1 32768 leaky_100[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_101 (BatchNormalization) (None, None, None, 1 512 conv_101[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_101 (LeakyReLU) (None, None, None, 1 0 bnorm_101[0][0] \n", "__________________________________________________________________________________________________\n", "conv_102 (Conv2D) (None, None, None, 2 294912 leaky_101[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_102 (BatchNormalization) (None, None, None, 2 1024 conv_102[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_102 (LeakyReLU) (None, None, None, 2 0 bnorm_102[0][0] \n", "__________________________________________________________________________________________________\n", "conv_103 (Conv2D) (None, None, None, 1 32768 leaky_102[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_103 (BatchNormalization) (None, None, None, 1 512 conv_103[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_103 (LeakyReLU) (None, None, None, 1 0 bnorm_103[0][0] \n", "__________________________________________________________________________________________________\n", "conv_80 (Conv2D) (None, None, None, 1 4718592 leaky_79[0][0] \n", "__________________________________________________________________________________________________\n", "conv_92 (Conv2D) (None, None, None, 5 1179648 leaky_91[0][0] \n", "__________________________________________________________________________________________________\n", "conv_104 (Conv2D) (None, None, None, 2 294912 leaky_103[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_80 (BatchNormalization) (None, None, None, 1 4096 conv_80[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_92 (BatchNormalization) (None, None, None, 5 2048 conv_92[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_104 (BatchNormalization) (None, None, None, 2 1024 conv_104[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_80 (LeakyReLU) (None, None, None, 1 0 bnorm_80[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_92 (LeakyReLU) (None, None, None, 5 0 bnorm_92[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_104 (LeakyReLU) (None, None, None, 2 0 bnorm_104[0][0] \n", "__________________________________________________________________________________________________\n", "conv_81 (Conv2D) (None, None, None, 1 18450 leaky_80[0][0] \n", "__________________________________________________________________________________________________\n", "conv_93 (Conv2D) (None, None, None, 1 9234 leaky_92[0][0] \n", "__________________________________________________________________________________________________\n", "conv_105 (Conv2D) (None, None, None, 1 4626 leaky_104[0][0] \n", "==================================================================================================\n", "Total params: 61,576,342\n", "Trainable params: 61,523,734\n", "Non-trainable params: 52,608\n", "__________________________________________________________________________________________________\n" ] } ], "source": [ "infer_model.summary()" ] }, { "cell_type": "code", "execution_count": 7, "metadata": {}, "outputs": [], "source": [ "train_model.save_weights('Prueba1.h5')" ] }, { "cell_type": "code", "execution_count": 8, "metadata": {}, "outputs": [], "source": [ "infer_model.load_weights('Prueba1.h5')" ] }, { "cell_type": "code", "execution_count": 9, "metadata": {}, "outputs": [], "source": [ "infer_model.save('Prueba1.h5')" ] }, { "cell_type": "code", "execution_count": 10, "metadata": {}, "outputs": [], "source": [ "train_model.load_weights('Prueba1.h5')" ] }, { "cell_type": "code", "execution_count": 12, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Model: \"model\"\n", "__________________________________________________________________________________________________\n", "Layer (type) Output Shape Param # Connected to \n", "==================================================================================================\n", "input_1 (InputLayer) [(None, None, None, 0 \n", "__________________________________________________________________________________________________\n", "conv_0 (Conv2D) (None, None, None, 3 864 input_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_0 (BatchNormalization) (None, None, None, 3 128 conv_0[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_0 (LeakyReLU) (None, None, None, 3 0 bnorm_0[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d (ZeroPadding2D) (None, None, None, 3 0 leaky_0[0][0] \n", "__________________________________________________________________________________________________\n", "conv_1 (Conv2D) (None, None, None, 6 18432 zero_padding2d[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_1 (BatchNormalization) (None, None, None, 6 256 conv_1[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_1 (LeakyReLU) (None, None, None, 6 0 bnorm_1[0][0] \n", "__________________________________________________________________________________________________\n", "conv_2 (Conv2D) (None, None, None, 3 2048 leaky_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_2 (BatchNormalization) (None, None, None, 3 128 conv_2[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_2 (LeakyReLU) (None, None, None, 3 0 bnorm_2[0][0] \n", "__________________________________________________________________________________________________\n", "conv_3 (Conv2D) (None, None, None, 6 18432 leaky_2[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_3 (BatchNormalization) (None, None, None, 6 256 conv_3[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_3 (LeakyReLU) (None, None, None, 6 0 bnorm_3[0][0] \n", "__________________________________________________________________________________________________\n", "add (Add) (None, None, None, 6 0 leaky_1[0][0] \n", " leaky_3[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_1 (ZeroPadding2D (None, None, None, 6 0 add[0][0] \n", "__________________________________________________________________________________________________\n", "conv_5 (Conv2D) (None, None, None, 1 73728 zero_padding2d_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_5 (BatchNormalization) (None, None, None, 1 512 conv_5[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_5 (LeakyReLU) (None, None, None, 1 0 bnorm_5[0][0] \n", "__________________________________________________________________________________________________\n", "conv_6 (Conv2D) (None, None, None, 6 8192 leaky_5[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_6 (BatchNormalization) (None, None, None, 6 256 conv_6[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_6 (LeakyReLU) (None, None, None, 6 0 bnorm_6[0][0] \n", "__________________________________________________________________________________________________\n", "conv_7 (Conv2D) (None, None, None, 1 73728 leaky_6[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_7 (BatchNormalization) (None, None, None, 1 512 conv_7[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_7 (LeakyReLU) (None, None, None, 1 0 bnorm_7[0][0] \n", "__________________________________________________________________________________________________\n", "add_1 (Add) (None, None, None, 1 0 leaky_5[0][0] \n", " leaky_7[0][0] \n", "__________________________________________________________________________________________________\n", "conv_9 (Conv2D) (None, None, None, 6 8192 add_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_9 (BatchNormalization) (None, None, None, 6 256 conv_9[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_9 (LeakyReLU) (None, None, None, 6 0 bnorm_9[0][0] \n", "__________________________________________________________________________________________________\n", "conv_10 (Conv2D) (None, None, None, 1 73728 leaky_9[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_10 (BatchNormalization) (None, None, None, 1 512 conv_10[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_10 (LeakyReLU) (None, None, None, 1 0 bnorm_10[0][0] \n", "__________________________________________________________________________________________________\n", "add_2 (Add) (None, None, None, 1 0 add_1[0][0] \n", " leaky_10[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_2 (ZeroPadding2D (None, None, None, 1 0 add_2[0][0] \n", "__________________________________________________________________________________________________\n", "conv_12 (Conv2D) (None, None, None, 2 294912 zero_padding2d_2[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_12 (BatchNormalization) (None, None, None, 2 1024 conv_12[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_12 (LeakyReLU) (None, None, None, 2 0 bnorm_12[0][0] \n", "__________________________________________________________________________________________________\n", "conv_13 (Conv2D) (None, None, None, 1 32768 leaky_12[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_13 (BatchNormalization) (None, None, None, 1 512 conv_13[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_13 (LeakyReLU) (None, None, None, 1 0 bnorm_13[0][0] \n", "__________________________________________________________________________________________________\n", "conv_14 (Conv2D) (None, None, None, 2 294912 leaky_13[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_14 (BatchNormalization) (None, None, None, 2 1024 conv_14[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_14 (LeakyReLU) (None, None, None, 2 0 bnorm_14[0][0] \n", "__________________________________________________________________________________________________\n", "add_3 (Add) (None, None, None, 2 0 leaky_12[0][0] \n", " leaky_14[0][0] \n", "__________________________________________________________________________________________________\n", "conv_16 (Conv2D) (None, None, None, 1 32768 add_3[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_16 (BatchNormalization) (None, None, None, 1 512 conv_16[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_16 (LeakyReLU) (None, None, None, 1 0 bnorm_16[0][0] \n", "__________________________________________________________________________________________________\n", "conv_17 (Conv2D) (None, None, None, 2 294912 leaky_16[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_17 (BatchNormalization) (None, None, None, 2 1024 conv_17[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_17 (LeakyReLU) (None, None, None, 2 0 bnorm_17[0][0] \n", "__________________________________________________________________________________________________\n", "add_4 (Add) (None, None, None, 2 0 add_3[0][0] \n", " leaky_17[0][0] \n", "__________________________________________________________________________________________________\n", "conv_19 (Conv2D) (None, None, None, 1 32768 add_4[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_19 (BatchNormalization) (None, None, None, 1 512 conv_19[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_19 (LeakyReLU) (None, None, None, 1 0 bnorm_19[0][0] \n", "__________________________________________________________________________________________________\n", "conv_20 (Conv2D) (None, None, None, 2 294912 leaky_19[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_20 (BatchNormalization) (None, None, None, 2 1024 conv_20[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_20 (LeakyReLU) (None, None, None, 2 0 bnorm_20[0][0] \n", "__________________________________________________________________________________________________\n", "add_5 (Add) (None, None, None, 2 0 add_4[0][0] \n", " leaky_20[0][0] \n", "__________________________________________________________________________________________________\n", "conv_22 (Conv2D) (None, None, None, 1 32768 add_5[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_22 (BatchNormalization) (None, None, None, 1 512 conv_22[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_22 (LeakyReLU) (None, None, None, 1 0 bnorm_22[0][0] \n", "__________________________________________________________________________________________________\n", "conv_23 (Conv2D) (None, None, None, 2 294912 leaky_22[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_23 (BatchNormalization) (None, None, None, 2 1024 conv_23[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_23 (LeakyReLU) (None, None, None, 2 0 bnorm_23[0][0] \n", "__________________________________________________________________________________________________\n", "add_6 (Add) (None, None, None, 2 0 add_5[0][0] \n", " leaky_23[0][0] \n", "__________________________________________________________________________________________________\n", "conv_25 (Conv2D) (None, None, None, 1 32768 add_6[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_25 (BatchNormalization) (None, None, None, 1 512 conv_25[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_25 (LeakyReLU) (None, None, None, 1 0 bnorm_25[0][0] \n", "__________________________________________________________________________________________________\n", "conv_26 (Conv2D) (None, None, None, 2 294912 leaky_25[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_26 (BatchNormalization) (None, None, None, 2 1024 conv_26[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_26 (LeakyReLU) (None, None, None, 2 0 bnorm_26[0][0] \n", "__________________________________________________________________________________________________\n", "add_7 (Add) (None, None, None, 2 0 add_6[0][0] \n", " leaky_26[0][0] \n", "__________________________________________________________________________________________________\n", "conv_28 (Conv2D) (None, None, None, 1 32768 add_7[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_28 (BatchNormalization) (None, None, None, 1 512 conv_28[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_28 (LeakyReLU) (None, None, None, 1 0 bnorm_28[0][0] \n", "__________________________________________________________________________________________________\n", "conv_29 (Conv2D) (None, None, None, 2 294912 leaky_28[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_29 (BatchNormalization) (None, None, None, 2 1024 conv_29[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_29 (LeakyReLU) (None, None, None, 2 0 bnorm_29[0][0] \n", "__________________________________________________________________________________________________\n", "add_8 (Add) (None, None, None, 2 0 add_7[0][0] \n", " leaky_29[0][0] \n", "__________________________________________________________________________________________________\n", "conv_31 (Conv2D) (None, None, None, 1 32768 add_8[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_31 (BatchNormalization) (None, None, None, 1 512 conv_31[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_31 (LeakyReLU) (None, None, None, 1 0 bnorm_31[0][0] \n", "__________________________________________________________________________________________________\n", "conv_32 (Conv2D) (None, None, None, 2 294912 leaky_31[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_32 (BatchNormalization) (None, None, None, 2 1024 conv_32[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_32 (LeakyReLU) (None, None, None, 2 0 bnorm_32[0][0] \n", "__________________________________________________________________________________________________\n", "add_9 (Add) (None, None, None, 2 0 add_8[0][0] \n", " leaky_32[0][0] \n", "__________________________________________________________________________________________________\n", "conv_34 (Conv2D) (None, None, None, 1 32768 add_9[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_34 (BatchNormalization) (None, None, None, 1 512 conv_34[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_34 (LeakyReLU) (None, None, None, 1 0 bnorm_34[0][0] \n", "__________________________________________________________________________________________________\n", "conv_35 (Conv2D) (None, None, None, 2 294912 leaky_34[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_35 (BatchNormalization) (None, None, None, 2 1024 conv_35[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_35 (LeakyReLU) (None, None, None, 2 0 bnorm_35[0][0] \n", "__________________________________________________________________________________________________\n", "add_10 (Add) (None, None, None, 2 0 add_9[0][0] \n", " leaky_35[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_3 (ZeroPadding2D (None, None, None, 2 0 add_10[0][0] \n", "__________________________________________________________________________________________________\n", "conv_37 (Conv2D) (None, None, None, 5 1179648 zero_padding2d_3[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_37 (BatchNormalization) (None, None, None, 5 2048 conv_37[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_37 (LeakyReLU) (None, None, None, 5 0 bnorm_37[0][0] \n", "__________________________________________________________________________________________________\n", "conv_38 (Conv2D) (None, None, None, 2 131072 leaky_37[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_38 (BatchNormalization) (None, None, None, 2 1024 conv_38[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_38 (LeakyReLU) (None, None, None, 2 0 bnorm_38[0][0] \n", "__________________________________________________________________________________________________\n", "conv_39 (Conv2D) (None, None, None, 5 1179648 leaky_38[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_39 (BatchNormalization) (None, None, None, 5 2048 conv_39[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_39 (LeakyReLU) (None, None, None, 5 0 bnorm_39[0][0] \n", "__________________________________________________________________________________________________\n", "add_11 (Add) (None, None, None, 5 0 leaky_37[0][0] \n", " leaky_39[0][0] \n", "__________________________________________________________________________________________________\n", "conv_41 (Conv2D) (None, None, None, 2 131072 add_11[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_41 (BatchNormalization) (None, None, None, 2 1024 conv_41[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_41 (LeakyReLU) (None, None, None, 2 0 bnorm_41[0][0] \n", "__________________________________________________________________________________________________\n", "conv_42 (Conv2D) (None, None, None, 5 1179648 leaky_41[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_42 (BatchNormalization) (None, None, None, 5 2048 conv_42[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_42 (LeakyReLU) (None, None, None, 5 0 bnorm_42[0][0] \n", "__________________________________________________________________________________________________\n", "add_12 (Add) (None, None, None, 5 0 add_11[0][0] \n", " leaky_42[0][0] \n", "__________________________________________________________________________________________________\n", "conv_44 (Conv2D) (None, None, None, 2 131072 add_12[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_44 (BatchNormalization) (None, None, None, 2 1024 conv_44[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_44 (LeakyReLU) (None, None, None, 2 0 bnorm_44[0][0] \n", "__________________________________________________________________________________________________\n", "conv_45 (Conv2D) (None, None, None, 5 1179648 leaky_44[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_45 (BatchNormalization) (None, None, None, 5 2048 conv_45[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_45 (LeakyReLU) (None, None, None, 5 0 bnorm_45[0][0] \n", "__________________________________________________________________________________________________\n", "add_13 (Add) (None, None, None, 5 0 add_12[0][0] \n", " leaky_45[0][0] \n", "__________________________________________________________________________________________________\n", "conv_47 (Conv2D) (None, None, None, 2 131072 add_13[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_47 (BatchNormalization) (None, None, None, 2 1024 conv_47[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_47 (LeakyReLU) (None, None, None, 2 0 bnorm_47[0][0] \n", "__________________________________________________________________________________________________\n", "conv_48 (Conv2D) (None, None, None, 5 1179648 leaky_47[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_48 (BatchNormalization) (None, None, None, 5 2048 conv_48[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_48 (LeakyReLU) (None, None, None, 5 0 bnorm_48[0][0] \n", "__________________________________________________________________________________________________\n", "add_14 (Add) (None, None, None, 5 0 add_13[0][0] \n", " leaky_48[0][0] \n", "__________________________________________________________________________________________________\n", "conv_50 (Conv2D) (None, None, None, 2 131072 add_14[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_50 (BatchNormalization) (None, None, None, 2 1024 conv_50[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_50 (LeakyReLU) (None, None, None, 2 0 bnorm_50[0][0] \n", "__________________________________________________________________________________________________\n", "conv_51 (Conv2D) (None, None, None, 5 1179648 leaky_50[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_51 (BatchNormalization) (None, None, None, 5 2048 conv_51[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_51 (LeakyReLU) (None, None, None, 5 0 bnorm_51[0][0] \n", "__________________________________________________________________________________________________\n", "add_15 (Add) (None, None, None, 5 0 add_14[0][0] \n", " leaky_51[0][0] \n", "__________________________________________________________________________________________________\n", "conv_53 (Conv2D) (None, None, None, 2 131072 add_15[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_53 (BatchNormalization) (None, None, None, 2 1024 conv_53[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_53 (LeakyReLU) (None, None, None, 2 0 bnorm_53[0][0] \n", "__________________________________________________________________________________________________\n", "conv_54 (Conv2D) (None, None, None, 5 1179648 leaky_53[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_54 (BatchNormalization) (None, None, None, 5 2048 conv_54[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_54 (LeakyReLU) (None, None, None, 5 0 bnorm_54[0][0] \n", "__________________________________________________________________________________________________\n", "add_16 (Add) (None, None, None, 5 0 add_15[0][0] \n", " leaky_54[0][0] \n", "__________________________________________________________________________________________________\n", "conv_56 (Conv2D) (None, None, None, 2 131072 add_16[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_56 (BatchNormalization) (None, None, None, 2 1024 conv_56[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_56 (LeakyReLU) (None, None, None, 2 0 bnorm_56[0][0] \n", "__________________________________________________________________________________________________\n", "conv_57 (Conv2D) (None, None, None, 5 1179648 leaky_56[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_57 (BatchNormalization) (None, None, None, 5 2048 conv_57[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_57 (LeakyReLU) (None, None, None, 5 0 bnorm_57[0][0] \n", "__________________________________________________________________________________________________\n", "add_17 (Add) (None, None, None, 5 0 add_16[0][0] \n", " leaky_57[0][0] \n", "__________________________________________________________________________________________________\n", "conv_59 (Conv2D) (None, None, None, 2 131072 add_17[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_59 (BatchNormalization) (None, None, None, 2 1024 conv_59[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_59 (LeakyReLU) (None, None, None, 2 0 bnorm_59[0][0] \n", "__________________________________________________________________________________________________\n", "conv_60 (Conv2D) (None, None, None, 5 1179648 leaky_59[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_60 (BatchNormalization) (None, None, None, 5 2048 conv_60[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_60 (LeakyReLU) (None, None, None, 5 0 bnorm_60[0][0] \n", "__________________________________________________________________________________________________\n", "add_18 (Add) (None, None, None, 5 0 add_17[0][0] \n", " leaky_60[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_4 (ZeroPadding2D (None, None, None, 5 0 add_18[0][0] \n", "__________________________________________________________________________________________________\n", "conv_62 (Conv2D) (None, None, None, 1 4718592 zero_padding2d_4[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_62 (BatchNormalization) (None, None, None, 1 4096 conv_62[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_62 (LeakyReLU) (None, None, None, 1 0 bnorm_62[0][0] \n", "__________________________________________________________________________________________________\n", "conv_63 (Conv2D) (None, None, None, 5 524288 leaky_62[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_63 (BatchNormalization) (None, None, None, 5 2048 conv_63[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_63 (LeakyReLU) (None, None, None, 5 0 bnorm_63[0][0] \n", "__________________________________________________________________________________________________\n", "conv_64 (Conv2D) (None, None, None, 1 4718592 leaky_63[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_64 (BatchNormalization) (None, None, None, 1 4096 conv_64[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_64 (LeakyReLU) (None, None, None, 1 0 bnorm_64[0][0] \n", "__________________________________________________________________________________________________\n", "add_19 (Add) (None, None, None, 1 0 leaky_62[0][0] \n", " leaky_64[0][0] \n", "__________________________________________________________________________________________________\n", "conv_66 (Conv2D) (None, None, None, 5 524288 add_19[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_66 (BatchNormalization) (None, None, None, 5 2048 conv_66[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_66 (LeakyReLU) (None, None, None, 5 0 bnorm_66[0][0] \n", "__________________________________________________________________________________________________\n", "conv_67 (Conv2D) (None, None, None, 1 4718592 leaky_66[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_67 (BatchNormalization) (None, None, None, 1 4096 conv_67[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_67 (LeakyReLU) (None, None, None, 1 0 bnorm_67[0][0] \n", "__________________________________________________________________________________________________\n", "add_20 (Add) (None, None, None, 1 0 add_19[0][0] \n", " leaky_67[0][0] \n", "__________________________________________________________________________________________________\n", "conv_69 (Conv2D) (None, None, None, 5 524288 add_20[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_69 (BatchNormalization) (None, None, None, 5 2048 conv_69[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_69 (LeakyReLU) (None, None, None, 5 0 bnorm_69[0][0] \n", "__________________________________________________________________________________________________\n", "conv_70 (Conv2D) (None, None, None, 1 4718592 leaky_69[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_70 (BatchNormalization) (None, None, None, 1 4096 conv_70[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_70 (LeakyReLU) (None, None, None, 1 0 bnorm_70[0][0] \n", "__________________________________________________________________________________________________\n", "add_21 (Add) (None, None, None, 1 0 add_20[0][0] \n", " leaky_70[0][0] \n", "__________________________________________________________________________________________________\n", "conv_72 (Conv2D) (None, None, None, 5 524288 add_21[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_72 (BatchNormalization) (None, None, None, 5 2048 conv_72[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_72 (LeakyReLU) (None, None, None, 5 0 bnorm_72[0][0] \n", "__________________________________________________________________________________________________\n", "conv_73 (Conv2D) (None, None, None, 1 4718592 leaky_72[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_73 (BatchNormalization) (None, None, None, 1 4096 conv_73[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_73 (LeakyReLU) (None, None, None, 1 0 bnorm_73[0][0] \n", "__________________________________________________________________________________________________\n", "add_22 (Add) (None, None, None, 1 0 add_21[0][0] \n", " leaky_73[0][0] \n", "__________________________________________________________________________________________________\n", "conv_75 (Conv2D) (None, None, None, 5 524288 add_22[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_75 (BatchNormalization) (None, None, None, 5 2048 conv_75[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_75 (LeakyReLU) (None, None, None, 5 0 bnorm_75[0][0] \n", "__________________________________________________________________________________________________\n", "conv_76 (Conv2D) (None, None, None, 1 4718592 leaky_75[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_76 (BatchNormalization) (None, None, None, 1 4096 conv_76[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_76 (LeakyReLU) (None, None, None, 1 0 bnorm_76[0][0] \n", "__________________________________________________________________________________________________\n", "conv_77 (Conv2D) (None, None, None, 5 524288 leaky_76[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_77 (BatchNormalization) (None, None, None, 5 2048 conv_77[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_77 (LeakyReLU) (None, None, None, 5 0 bnorm_77[0][0] \n", "__________________________________________________________________________________________________\n", "conv_78 (Conv2D) (None, None, None, 1 4718592 leaky_77[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_78 (BatchNormalization) (None, None, None, 1 4096 conv_78[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_78 (LeakyReLU) (None, None, None, 1 0 bnorm_78[0][0] \n", "__________________________________________________________________________________________________\n", "conv_79 (Conv2D) (None, None, None, 5 524288 leaky_78[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_79 (BatchNormalization) (None, None, None, 5 2048 conv_79[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_79 (LeakyReLU) (None, None, None, 5 0 bnorm_79[0][0] \n", "__________________________________________________________________________________________________\n", "conv_84 (Conv2D) (None, None, None, 2 131072 leaky_79[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_84 (BatchNormalization) (None, None, None, 2 1024 conv_84[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_84 (LeakyReLU) (None, None, None, 2 0 bnorm_84[0][0] \n", "__________________________________________________________________________________________________\n", "up_sampling2d (UpSampling2D) (None, None, None, 2 0 leaky_84[0][0] \n", "__________________________________________________________________________________________________\n", "concatenate (Concatenate) (None, None, None, 7 0 up_sampling2d[0][0] \n", " add_18[0][0] \n", "__________________________________________________________________________________________________\n", "conv_87 (Conv2D) (None, None, None, 2 196608 concatenate[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_87 (BatchNormalization) (None, None, None, 2 1024 conv_87[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_87 (LeakyReLU) (None, None, None, 2 0 bnorm_87[0][0] \n", "__________________________________________________________________________________________________\n", "conv_88 (Conv2D) (None, None, None, 5 1179648 leaky_87[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_88 (BatchNormalization) (None, None, None, 5 2048 conv_88[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_88 (LeakyReLU) (None, None, None, 5 0 bnorm_88[0][0] \n", "__________________________________________________________________________________________________\n", "conv_89 (Conv2D) (None, None, None, 2 131072 leaky_88[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_89 (BatchNormalization) (None, None, None, 2 1024 conv_89[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_89 (LeakyReLU) (None, None, None, 2 0 bnorm_89[0][0] \n", "__________________________________________________________________________________________________\n", "conv_90 (Conv2D) (None, None, None, 5 1179648 leaky_89[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_90 (BatchNormalization) (None, None, None, 5 2048 conv_90[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_90 (LeakyReLU) (None, None, None, 5 0 bnorm_90[0][0] \n", "__________________________________________________________________________________________________\n", "conv_91 (Conv2D) (None, None, None, 2 131072 leaky_90[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_91 (BatchNormalization) (None, None, None, 2 1024 conv_91[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_91 (LeakyReLU) (None, None, None, 2 0 bnorm_91[0][0] \n", "__________________________________________________________________________________________________\n", "conv_96 (Conv2D) (None, None, None, 1 32768 leaky_91[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_96 (BatchNormalization) (None, None, None, 1 512 conv_96[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_96 (LeakyReLU) (None, None, None, 1 0 bnorm_96[0][0] \n", "__________________________________________________________________________________________________\n", "up_sampling2d_1 (UpSampling2D) (None, None, None, 1 0 leaky_96[0][0] \n", "__________________________________________________________________________________________________\n", "concatenate_1 (Concatenate) (None, None, None, 3 0 up_sampling2d_1[0][0] \n", " add_10[0][0] \n", "__________________________________________________________________________________________________\n", "conv_99 (Conv2D) (None, None, None, 1 49152 concatenate_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_99 (BatchNormalization) (None, None, None, 1 512 conv_99[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_99 (LeakyReLU) (None, None, None, 1 0 bnorm_99[0][0] \n", "__________________________________________________________________________________________________\n", "conv_100 (Conv2D) (None, None, None, 2 294912 leaky_99[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_100 (BatchNormalization) (None, None, None, 2 1024 conv_100[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_100 (LeakyReLU) (None, None, None, 2 0 bnorm_100[0][0] \n", "__________________________________________________________________________________________________\n", "conv_101 (Conv2D) (None, None, None, 1 32768 leaky_100[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_101 (BatchNormalization) (None, None, None, 1 512 conv_101[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_101 (LeakyReLU) (None, None, None, 1 0 bnorm_101[0][0] \n", "__________________________________________________________________________________________________\n", "conv_102 (Conv2D) (None, None, None, 2 294912 leaky_101[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_102 (BatchNormalization) (None, None, None, 2 1024 conv_102[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_102 (LeakyReLU) (None, None, None, 2 0 bnorm_102[0][0] \n", "__________________________________________________________________________________________________\n", "conv_103 (Conv2D) (None, None, None, 1 32768 leaky_102[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_103 (BatchNormalization) (None, None, None, 1 512 conv_103[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_103 (LeakyReLU) (None, None, None, 1 0 bnorm_103[0][0] \n", "__________________________________________________________________________________________________\n", "conv_80 (Conv2D) (None, None, None, 1 4718592 leaky_79[0][0] \n", "__________________________________________________________________________________________________\n", "conv_92 (Conv2D) (None, None, None, 5 1179648 leaky_91[0][0] \n", "__________________________________________________________________________________________________\n", "conv_104 (Conv2D) (None, None, None, 2 294912 leaky_103[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_80 (BatchNormalization) (None, None, None, 1 4096 conv_80[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_92 (BatchNormalization) (None, None, None, 5 2048 conv_92[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_104 (BatchNormalization) (None, None, None, 2 1024 conv_104[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_80 (LeakyReLU) (None, None, None, 1 0 bnorm_80[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_92 (LeakyReLU) (None, None, None, 5 0 bnorm_92[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_104 (LeakyReLU) (None, None, None, 2 0 bnorm_104[0][0] \n", "__________________________________________________________________________________________________\n", "conv_81 (Conv2D) (None, None, None, 1 18450 leaky_80[0][0] \n", "__________________________________________________________________________________________________\n", "input_3 (InputLayer) [(None, None, None, 0 \n", "__________________________________________________________________________________________________\n", "input_2 (InputLayer) [(None, 1, 1, 1, 5, 0 \n", "__________________________________________________________________________________________________\n", "conv_93 (Conv2D) (None, None, None, 1 9234 leaky_92[0][0] \n", "__________________________________________________________________________________________________\n", "input_4 (InputLayer) [(None, None, None, 0 \n", "__________________________________________________________________________________________________\n", "conv_105 (Conv2D) (None, None, None, 1 4626 leaky_104[0][0] \n", "__________________________________________________________________________________________________\n", "input_5 (InputLayer) [(None, None, None, 0 \n", "__________________________________________________________________________________________________\n", "yolo_layer (YoloLayer) (2,) 0 input_1[0][0] \n", " conv_81[0][0] \n", " input_3[0][0] \n", " input_2[0][0] \n", "__________________________________________________________________________________________________\n", "yolo_layer_1 (YoloLayer) (2,) 0 input_1[0][0] \n", " conv_93[0][0] \n", " input_4[0][0] \n", " input_2[0][0] \n", "__________________________________________________________________________________________________\n", "yolo_layer_2 (YoloLayer) (2,) 0 input_1[0][0] \n", " conv_105[0][0] \n", " input_5[0][0] \n", " input_2[0][0] \n", "==================================================================================================\n", "Total params: 61,576,342\n", "Trainable params: 61,523,734\n", "Non-trainable params: 52,608\n", "__________________________________________________________________________________________________\n" ] } ], "source": [ "train_model.summary()\n" ] }, { "cell_type": "code", "execution_count": 13, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "WARNING:tensorflow:No training configuration found in save file: the model was *not* compiled. Compile it manually.\n" ] } ], "source": [ "A = load_model('Prueba1.h5')" ] }, { "cell_type": "code", "execution_count": 14, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Model: \"model_1\"\n", "__________________________________________________________________________________________________\n", "Layer (type) Output Shape Param # Connected to \n", "==================================================================================================\n", "input_1 (InputLayer) [(None, None, None, 0 \n", "__________________________________________________________________________________________________\n", "conv_0 (Conv2D) (None, None, None, 3 864 input_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_0 (BatchNormalization) (None, None, None, 3 128 conv_0[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_0 (LeakyReLU) (None, None, None, 3 0 bnorm_0[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d (ZeroPadding2D) (None, None, None, 3 0 leaky_0[0][0] \n", "__________________________________________________________________________________________________\n", "conv_1 (Conv2D) (None, None, None, 6 18432 zero_padding2d[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_1 (BatchNormalization) (None, None, None, 6 256 conv_1[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_1 (LeakyReLU) (None, None, None, 6 0 bnorm_1[0][0] \n", "__________________________________________________________________________________________________\n", "conv_2 (Conv2D) (None, None, None, 3 2048 leaky_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_2 (BatchNormalization) (None, None, None, 3 128 conv_2[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_2 (LeakyReLU) (None, None, None, 3 0 bnorm_2[0][0] \n", "__________________________________________________________________________________________________\n", "conv_3 (Conv2D) (None, None, None, 6 18432 leaky_2[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_3 (BatchNormalization) (None, None, None, 6 256 conv_3[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_3 (LeakyReLU) (None, None, None, 6 0 bnorm_3[0][0] \n", "__________________________________________________________________________________________________\n", "add (Add) (None, None, None, 6 0 leaky_1[0][0] \n", " leaky_3[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_1 (ZeroPadding2D (None, None, None, 6 0 add[0][0] \n", "__________________________________________________________________________________________________\n", "conv_5 (Conv2D) (None, None, None, 1 73728 zero_padding2d_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_5 (BatchNormalization) (None, None, None, 1 512 conv_5[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_5 (LeakyReLU) (None, None, None, 1 0 bnorm_5[0][0] \n", "__________________________________________________________________________________________________\n", "conv_6 (Conv2D) (None, None, None, 6 8192 leaky_5[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_6 (BatchNormalization) (None, None, None, 6 256 conv_6[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_6 (LeakyReLU) (None, None, None, 6 0 bnorm_6[0][0] \n", "__________________________________________________________________________________________________\n", "conv_7 (Conv2D) (None, None, None, 1 73728 leaky_6[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_7 (BatchNormalization) (None, None, None, 1 512 conv_7[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_7 (LeakyReLU) (None, None, None, 1 0 bnorm_7[0][0] \n", "__________________________________________________________________________________________________\n", "add_1 (Add) (None, None, None, 1 0 leaky_5[0][0] \n", " leaky_7[0][0] \n", "__________________________________________________________________________________________________\n", "conv_9 (Conv2D) (None, None, None, 6 8192 add_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_9 (BatchNormalization) (None, None, None, 6 256 conv_9[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_9 (LeakyReLU) (None, None, None, 6 0 bnorm_9[0][0] \n", "__________________________________________________________________________________________________\n", "conv_10 (Conv2D) (None, None, None, 1 73728 leaky_9[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_10 (BatchNormalization) (None, None, None, 1 512 conv_10[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_10 (LeakyReLU) (None, None, None, 1 0 bnorm_10[0][0] \n", "__________________________________________________________________________________________________\n", "add_2 (Add) (None, None, None, 1 0 add_1[0][0] \n", " leaky_10[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_2 (ZeroPadding2D (None, None, None, 1 0 add_2[0][0] \n", "__________________________________________________________________________________________________\n", "conv_12 (Conv2D) (None, None, None, 2 294912 zero_padding2d_2[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_12 (BatchNormalization) (None, None, None, 2 1024 conv_12[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_12 (LeakyReLU) (None, None, None, 2 0 bnorm_12[0][0] \n", "__________________________________________________________________________________________________\n", "conv_13 (Conv2D) (None, None, None, 1 32768 leaky_12[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_13 (BatchNormalization) (None, None, None, 1 512 conv_13[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_13 (LeakyReLU) (None, None, None, 1 0 bnorm_13[0][0] \n", "__________________________________________________________________________________________________\n", "conv_14 (Conv2D) (None, None, None, 2 294912 leaky_13[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_14 (BatchNormalization) (None, None, None, 2 1024 conv_14[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_14 (LeakyReLU) (None, None, None, 2 0 bnorm_14[0][0] \n", "__________________________________________________________________________________________________\n", "add_3 (Add) (None, None, None, 2 0 leaky_12[0][0] \n", " leaky_14[0][0] \n", "__________________________________________________________________________________________________\n", "conv_16 (Conv2D) (None, None, None, 1 32768 add_3[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_16 (BatchNormalization) (None, None, None, 1 512 conv_16[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_16 (LeakyReLU) (None, None, None, 1 0 bnorm_16[0][0] \n", "__________________________________________________________________________________________________\n", "conv_17 (Conv2D) (None, None, None, 2 294912 leaky_16[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_17 (BatchNormalization) (None, None, None, 2 1024 conv_17[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_17 (LeakyReLU) (None, None, None, 2 0 bnorm_17[0][0] \n", "__________________________________________________________________________________________________\n", "add_4 (Add) (None, None, None, 2 0 add_3[0][0] \n", " leaky_17[0][0] \n", "__________________________________________________________________________________________________\n", "conv_19 (Conv2D) (None, None, None, 1 32768 add_4[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_19 (BatchNormalization) (None, None, None, 1 512 conv_19[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_19 (LeakyReLU) (None, None, None, 1 0 bnorm_19[0][0] \n", "__________________________________________________________________________________________________\n", "conv_20 (Conv2D) (None, None, None, 2 294912 leaky_19[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_20 (BatchNormalization) (None, None, None, 2 1024 conv_20[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_20 (LeakyReLU) (None, None, None, 2 0 bnorm_20[0][0] \n", "__________________________________________________________________________________________________\n", "add_5 (Add) (None, None, None, 2 0 add_4[0][0] \n", " leaky_20[0][0] \n", "__________________________________________________________________________________________________\n", "conv_22 (Conv2D) (None, None, None, 1 32768 add_5[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_22 (BatchNormalization) (None, None, None, 1 512 conv_22[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_22 (LeakyReLU) (None, None, None, 1 0 bnorm_22[0][0] \n", "__________________________________________________________________________________________________\n", "conv_23 (Conv2D) (None, None, None, 2 294912 leaky_22[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_23 (BatchNormalization) (None, None, None, 2 1024 conv_23[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_23 (LeakyReLU) (None, None, None, 2 0 bnorm_23[0][0] \n", "__________________________________________________________________________________________________\n", "add_6 (Add) (None, None, None, 2 0 add_5[0][0] \n", " leaky_23[0][0] \n", "__________________________________________________________________________________________________\n", "conv_25 (Conv2D) (None, None, None, 1 32768 add_6[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_25 (BatchNormalization) (None, None, None, 1 512 conv_25[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_25 (LeakyReLU) (None, None, None, 1 0 bnorm_25[0][0] \n", "__________________________________________________________________________________________________\n", "conv_26 (Conv2D) (None, None, None, 2 294912 leaky_25[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_26 (BatchNormalization) (None, None, None, 2 1024 conv_26[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_26 (LeakyReLU) (None, None, None, 2 0 bnorm_26[0][0] \n", "__________________________________________________________________________________________________\n", "add_7 (Add) (None, None, None, 2 0 add_6[0][0] \n", " leaky_26[0][0] \n", "__________________________________________________________________________________________________\n", "conv_28 (Conv2D) (None, None, None, 1 32768 add_7[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_28 (BatchNormalization) (None, None, None, 1 512 conv_28[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_28 (LeakyReLU) (None, None, None, 1 0 bnorm_28[0][0] \n", "__________________________________________________________________________________________________\n", "conv_29 (Conv2D) (None, None, None, 2 294912 leaky_28[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_29 (BatchNormalization) (None, None, None, 2 1024 conv_29[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_29 (LeakyReLU) (None, None, None, 2 0 bnorm_29[0][0] \n", "__________________________________________________________________________________________________\n", "add_8 (Add) (None, None, None, 2 0 add_7[0][0] \n", " leaky_29[0][0] \n", "__________________________________________________________________________________________________\n", "conv_31 (Conv2D) (None, None, None, 1 32768 add_8[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_31 (BatchNormalization) (None, None, None, 1 512 conv_31[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_31 (LeakyReLU) (None, None, None, 1 0 bnorm_31[0][0] \n", "__________________________________________________________________________________________________\n", "conv_32 (Conv2D) (None, None, None, 2 294912 leaky_31[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_32 (BatchNormalization) (None, None, None, 2 1024 conv_32[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_32 (LeakyReLU) (None, None, None, 2 0 bnorm_32[0][0] \n", "__________________________________________________________________________________________________\n", "add_9 (Add) (None, None, None, 2 0 add_8[0][0] \n", " leaky_32[0][0] \n", "__________________________________________________________________________________________________\n", "conv_34 (Conv2D) (None, None, None, 1 32768 add_9[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_34 (BatchNormalization) (None, None, None, 1 512 conv_34[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_34 (LeakyReLU) (None, None, None, 1 0 bnorm_34[0][0] \n", "__________________________________________________________________________________________________\n", "conv_35 (Conv2D) (None, None, None, 2 294912 leaky_34[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_35 (BatchNormalization) (None, None, None, 2 1024 conv_35[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_35 (LeakyReLU) (None, None, None, 2 0 bnorm_35[0][0] \n", "__________________________________________________________________________________________________\n", "add_10 (Add) (None, None, None, 2 0 add_9[0][0] \n", " leaky_35[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_3 (ZeroPadding2D (None, None, None, 2 0 add_10[0][0] \n", "__________________________________________________________________________________________________\n", "conv_37 (Conv2D) (None, None, None, 5 1179648 zero_padding2d_3[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_37 (BatchNormalization) (None, None, None, 5 2048 conv_37[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_37 (LeakyReLU) (None, None, None, 5 0 bnorm_37[0][0] \n", "__________________________________________________________________________________________________\n", "conv_38 (Conv2D) (None, None, None, 2 131072 leaky_37[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_38 (BatchNormalization) (None, None, None, 2 1024 conv_38[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_38 (LeakyReLU) (None, None, None, 2 0 bnorm_38[0][0] \n", "__________________________________________________________________________________________________\n", "conv_39 (Conv2D) (None, None, None, 5 1179648 leaky_38[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_39 (BatchNormalization) (None, None, None, 5 2048 conv_39[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_39 (LeakyReLU) (None, None, None, 5 0 bnorm_39[0][0] \n", "__________________________________________________________________________________________________\n", "add_11 (Add) (None, None, None, 5 0 leaky_37[0][0] \n", " leaky_39[0][0] \n", "__________________________________________________________________________________________________\n", "conv_41 (Conv2D) (None, None, None, 2 131072 add_11[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_41 (BatchNormalization) (None, None, None, 2 1024 conv_41[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_41 (LeakyReLU) (None, None, None, 2 0 bnorm_41[0][0] \n", "__________________________________________________________________________________________________\n", "conv_42 (Conv2D) (None, None, None, 5 1179648 leaky_41[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_42 (BatchNormalization) (None, None, None, 5 2048 conv_42[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_42 (LeakyReLU) (None, None, None, 5 0 bnorm_42[0][0] \n", "__________________________________________________________________________________________________\n", "add_12 (Add) (None, None, None, 5 0 add_11[0][0] \n", " leaky_42[0][0] \n", "__________________________________________________________________________________________________\n", "conv_44 (Conv2D) (None, None, None, 2 131072 add_12[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_44 (BatchNormalization) (None, None, None, 2 1024 conv_44[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_44 (LeakyReLU) (None, None, None, 2 0 bnorm_44[0][0] \n", "__________________________________________________________________________________________________\n", "conv_45 (Conv2D) (None, None, None, 5 1179648 leaky_44[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_45 (BatchNormalization) (None, None, None, 5 2048 conv_45[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_45 (LeakyReLU) (None, None, None, 5 0 bnorm_45[0][0] \n", "__________________________________________________________________________________________________\n", "add_13 (Add) (None, None, None, 5 0 add_12[0][0] \n", " leaky_45[0][0] \n", "__________________________________________________________________________________________________\n", "conv_47 (Conv2D) (None, None, None, 2 131072 add_13[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_47 (BatchNormalization) (None, None, None, 2 1024 conv_47[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_47 (LeakyReLU) (None, None, None, 2 0 bnorm_47[0][0] \n", "__________________________________________________________________________________________________\n", "conv_48 (Conv2D) (None, None, None, 5 1179648 leaky_47[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_48 (BatchNormalization) (None, None, None, 5 2048 conv_48[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_48 (LeakyReLU) (None, None, None, 5 0 bnorm_48[0][0] \n", "__________________________________________________________________________________________________\n", "add_14 (Add) (None, None, None, 5 0 add_13[0][0] \n", " leaky_48[0][0] \n", "__________________________________________________________________________________________________\n", "conv_50 (Conv2D) (None, None, None, 2 131072 add_14[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_50 (BatchNormalization) (None, None, None, 2 1024 conv_50[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_50 (LeakyReLU) (None, None, None, 2 0 bnorm_50[0][0] \n", "__________________________________________________________________________________________________\n", "conv_51 (Conv2D) (None, None, None, 5 1179648 leaky_50[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_51 (BatchNormalization) (None, None, None, 5 2048 conv_51[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_51 (LeakyReLU) (None, None, None, 5 0 bnorm_51[0][0] \n", "__________________________________________________________________________________________________\n", "add_15 (Add) (None, None, None, 5 0 add_14[0][0] \n", " leaky_51[0][0] \n", "__________________________________________________________________________________________________\n", "conv_53 (Conv2D) (None, None, None, 2 131072 add_15[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_53 (BatchNormalization) (None, None, None, 2 1024 conv_53[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_53 (LeakyReLU) (None, None, None, 2 0 bnorm_53[0][0] \n", "__________________________________________________________________________________________________\n", "conv_54 (Conv2D) (None, None, None, 5 1179648 leaky_53[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_54 (BatchNormalization) (None, None, None, 5 2048 conv_54[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_54 (LeakyReLU) (None, None, None, 5 0 bnorm_54[0][0] \n", "__________________________________________________________________________________________________\n", "add_16 (Add) (None, None, None, 5 0 add_15[0][0] \n", " leaky_54[0][0] \n", "__________________________________________________________________________________________________\n", "conv_56 (Conv2D) (None, None, None, 2 131072 add_16[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_56 (BatchNormalization) (None, None, None, 2 1024 conv_56[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_56 (LeakyReLU) (None, None, None, 2 0 bnorm_56[0][0] \n", "__________________________________________________________________________________________________\n", "conv_57 (Conv2D) (None, None, None, 5 1179648 leaky_56[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_57 (BatchNormalization) (None, None, None, 5 2048 conv_57[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_57 (LeakyReLU) (None, None, None, 5 0 bnorm_57[0][0] \n", "__________________________________________________________________________________________________\n", "add_17 (Add) (None, None, None, 5 0 add_16[0][0] \n", " leaky_57[0][0] \n", "__________________________________________________________________________________________________\n", "conv_59 (Conv2D) (None, None, None, 2 131072 add_17[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_59 (BatchNormalization) (None, None, None, 2 1024 conv_59[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_59 (LeakyReLU) (None, None, None, 2 0 bnorm_59[0][0] \n", "__________________________________________________________________________________________________\n", "conv_60 (Conv2D) (None, None, None, 5 1179648 leaky_59[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_60 (BatchNormalization) (None, None, None, 5 2048 conv_60[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_60 (LeakyReLU) (None, None, None, 5 0 bnorm_60[0][0] \n", "__________________________________________________________________________________________________\n", "add_18 (Add) (None, None, None, 5 0 add_17[0][0] \n", " leaky_60[0][0] \n", "__________________________________________________________________________________________________\n", "zero_padding2d_4 (ZeroPadding2D (None, None, None, 5 0 add_18[0][0] \n", "__________________________________________________________________________________________________\n", "conv_62 (Conv2D) (None, None, None, 1 4718592 zero_padding2d_4[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_62 (BatchNormalization) (None, None, None, 1 4096 conv_62[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_62 (LeakyReLU) (None, None, None, 1 0 bnorm_62[0][0] \n", "__________________________________________________________________________________________________\n", "conv_63 (Conv2D) (None, None, None, 5 524288 leaky_62[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_63 (BatchNormalization) (None, None, None, 5 2048 conv_63[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_63 (LeakyReLU) (None, None, None, 5 0 bnorm_63[0][0] \n", "__________________________________________________________________________________________________\n", "conv_64 (Conv2D) (None, None, None, 1 4718592 leaky_63[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_64 (BatchNormalization) (None, None, None, 1 4096 conv_64[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_64 (LeakyReLU) (None, None, None, 1 0 bnorm_64[0][0] \n", "__________________________________________________________________________________________________\n", "add_19 (Add) (None, None, None, 1 0 leaky_62[0][0] \n", " leaky_64[0][0] \n", "__________________________________________________________________________________________________\n", "conv_66 (Conv2D) (None, None, None, 5 524288 add_19[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_66 (BatchNormalization) (None, None, None, 5 2048 conv_66[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_66 (LeakyReLU) (None, None, None, 5 0 bnorm_66[0][0] \n", "__________________________________________________________________________________________________\n", "conv_67 (Conv2D) (None, None, None, 1 4718592 leaky_66[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_67 (BatchNormalization) (None, None, None, 1 4096 conv_67[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_67 (LeakyReLU) (None, None, None, 1 0 bnorm_67[0][0] \n", "__________________________________________________________________________________________________\n", "add_20 (Add) (None, None, None, 1 0 add_19[0][0] \n", " leaky_67[0][0] \n", "__________________________________________________________________________________________________\n", "conv_69 (Conv2D) (None, None, None, 5 524288 add_20[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_69 (BatchNormalization) (None, None, None, 5 2048 conv_69[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_69 (LeakyReLU) (None, None, None, 5 0 bnorm_69[0][0] \n", "__________________________________________________________________________________________________\n", "conv_70 (Conv2D) (None, None, None, 1 4718592 leaky_69[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_70 (BatchNormalization) (None, None, None, 1 4096 conv_70[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_70 (LeakyReLU) (None, None, None, 1 0 bnorm_70[0][0] \n", "__________________________________________________________________________________________________\n", "add_21 (Add) (None, None, None, 1 0 add_20[0][0] \n", " leaky_70[0][0] \n", "__________________________________________________________________________________________________\n", "conv_72 (Conv2D) (None, None, None, 5 524288 add_21[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_72 (BatchNormalization) (None, None, None, 5 2048 conv_72[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_72 (LeakyReLU) (None, None, None, 5 0 bnorm_72[0][0] \n", "__________________________________________________________________________________________________\n", "conv_73 (Conv2D) (None, None, None, 1 4718592 leaky_72[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_73 (BatchNormalization) (None, None, None, 1 4096 conv_73[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_73 (LeakyReLU) (None, None, None, 1 0 bnorm_73[0][0] \n", "__________________________________________________________________________________________________\n", "add_22 (Add) (None, None, None, 1 0 add_21[0][0] \n", " leaky_73[0][0] \n", "__________________________________________________________________________________________________\n", "conv_75 (Conv2D) (None, None, None, 5 524288 add_22[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_75 (BatchNormalization) (None, None, None, 5 2048 conv_75[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_75 (LeakyReLU) (None, None, None, 5 0 bnorm_75[0][0] \n", "__________________________________________________________________________________________________\n", "conv_76 (Conv2D) (None, None, None, 1 4718592 leaky_75[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_76 (BatchNormalization) (None, None, None, 1 4096 conv_76[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_76 (LeakyReLU) (None, None, None, 1 0 bnorm_76[0][0] \n", "__________________________________________________________________________________________________\n", "conv_77 (Conv2D) (None, None, None, 5 524288 leaky_76[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_77 (BatchNormalization) (None, None, None, 5 2048 conv_77[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_77 (LeakyReLU) (None, None, None, 5 0 bnorm_77[0][0] \n", "__________________________________________________________________________________________________\n", "conv_78 (Conv2D) (None, None, None, 1 4718592 leaky_77[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_78 (BatchNormalization) (None, None, None, 1 4096 conv_78[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_78 (LeakyReLU) (None, None, None, 1 0 bnorm_78[0][0] \n", "__________________________________________________________________________________________________\n", "conv_79 (Conv2D) (None, None, None, 5 524288 leaky_78[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_79 (BatchNormalization) (None, None, None, 5 2048 conv_79[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_79 (LeakyReLU) (None, None, None, 5 0 bnorm_79[0][0] \n", "__________________________________________________________________________________________________\n", "conv_84 (Conv2D) (None, None, None, 2 131072 leaky_79[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_84 (BatchNormalization) (None, None, None, 2 1024 conv_84[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_84 (LeakyReLU) (None, None, None, 2 0 bnorm_84[0][0] \n", "__________________________________________________________________________________________________\n", "up_sampling2d (UpSampling2D) (None, None, None, 2 0 leaky_84[0][0] \n", "__________________________________________________________________________________________________\n", "concatenate (Concatenate) (None, None, None, 7 0 up_sampling2d[0][0] \n", " add_18[0][0] \n", "__________________________________________________________________________________________________\n", "conv_87 (Conv2D) (None, None, None, 2 196608 concatenate[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_87 (BatchNormalization) (None, None, None, 2 1024 conv_87[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_87 (LeakyReLU) (None, None, None, 2 0 bnorm_87[0][0] \n", "__________________________________________________________________________________________________\n", "conv_88 (Conv2D) (None, None, None, 5 1179648 leaky_87[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_88 (BatchNormalization) (None, None, None, 5 2048 conv_88[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_88 (LeakyReLU) (None, None, None, 5 0 bnorm_88[0][0] \n", "__________________________________________________________________________________________________\n", "conv_89 (Conv2D) (None, None, None, 2 131072 leaky_88[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_89 (BatchNormalization) (None, None, None, 2 1024 conv_89[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_89 (LeakyReLU) (None, None, None, 2 0 bnorm_89[0][0] \n", "__________________________________________________________________________________________________\n", "conv_90 (Conv2D) (None, None, None, 5 1179648 leaky_89[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_90 (BatchNormalization) (None, None, None, 5 2048 conv_90[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_90 (LeakyReLU) (None, None, None, 5 0 bnorm_90[0][0] \n", "__________________________________________________________________________________________________\n", "conv_91 (Conv2D) (None, None, None, 2 131072 leaky_90[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_91 (BatchNormalization) (None, None, None, 2 1024 conv_91[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_91 (LeakyReLU) (None, None, None, 2 0 bnorm_91[0][0] \n", "__________________________________________________________________________________________________\n", "conv_96 (Conv2D) (None, None, None, 1 32768 leaky_91[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_96 (BatchNormalization) (None, None, None, 1 512 conv_96[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_96 (LeakyReLU) (None, None, None, 1 0 bnorm_96[0][0] \n", "__________________________________________________________________________________________________\n", "up_sampling2d_1 (UpSampling2D) (None, None, None, 1 0 leaky_96[0][0] \n", "__________________________________________________________________________________________________\n", "concatenate_1 (Concatenate) (None, None, None, 3 0 up_sampling2d_1[0][0] \n", " add_10[0][0] \n", "__________________________________________________________________________________________________\n", "conv_99 (Conv2D) (None, None, None, 1 49152 concatenate_1[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_99 (BatchNormalization) (None, None, None, 1 512 conv_99[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_99 (LeakyReLU) (None, None, None, 1 0 bnorm_99[0][0] \n", "__________________________________________________________________________________________________\n", "conv_100 (Conv2D) (None, None, None, 2 294912 leaky_99[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_100 (BatchNormalization) (None, None, None, 2 1024 conv_100[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_100 (LeakyReLU) (None, None, None, 2 0 bnorm_100[0][0] \n", "__________________________________________________________________________________________________\n", "conv_101 (Conv2D) (None, None, None, 1 32768 leaky_100[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_101 (BatchNormalization) (None, None, None, 1 512 conv_101[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_101 (LeakyReLU) (None, None, None, 1 0 bnorm_101[0][0] \n", "__________________________________________________________________________________________________\n", "conv_102 (Conv2D) (None, None, None, 2 294912 leaky_101[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_102 (BatchNormalization) (None, None, None, 2 1024 conv_102[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_102 (LeakyReLU) (None, None, None, 2 0 bnorm_102[0][0] \n", "__________________________________________________________________________________________________\n", "conv_103 (Conv2D) (None, None, None, 1 32768 leaky_102[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_103 (BatchNormalization) (None, None, None, 1 512 conv_103[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_103 (LeakyReLU) (None, None, None, 1 0 bnorm_103[0][0] \n", "__________________________________________________________________________________________________\n", "conv_80 (Conv2D) (None, None, None, 1 4718592 leaky_79[0][0] \n", "__________________________________________________________________________________________________\n", "conv_92 (Conv2D) (None, None, None, 5 1179648 leaky_91[0][0] \n", "__________________________________________________________________________________________________\n", "conv_104 (Conv2D) (None, None, None, 2 294912 leaky_103[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_80 (BatchNormalization) (None, None, None, 1 4096 conv_80[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_92 (BatchNormalization) (None, None, None, 5 2048 conv_92[0][0] \n", "__________________________________________________________________________________________________\n", "bnorm_104 (BatchNormalization) (None, None, None, 2 1024 conv_104[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_80 (LeakyReLU) (None, None, None, 1 0 bnorm_80[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_92 (LeakyReLU) (None, None, None, 5 0 bnorm_92[0][0] \n", "__________________________________________________________________________________________________\n", "leaky_104 (LeakyReLU) (None, None, None, 2 0 bnorm_104[0][0] \n", "__________________________________________________________________________________________________\n", "conv_81 (Conv2D) (None, None, None, 1 18450 leaky_80[0][0] \n", "__________________________________________________________________________________________________\n", "conv_93 (Conv2D) (None, None, None, 1 9234 leaky_92[0][0] \n", "__________________________________________________________________________________________________\n", "conv_105 (Conv2D) (None, None, None, 1 4626 leaky_104[0][0] \n", "==================================================================================================\n", "Total params: 61,576,342\n", "Trainable params: 61,523,734\n", "Non-trainable params: 52,608\n", "__________________________________________________________________________________________________\n" ] } ], "source": [ "A.summary()" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.7.7" } }, "nbformat": 4, "nbformat_minor": 4 }