From 9cf7f2215a9f50575ee962f35fc86cb4fd673513 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 29 Jul 2019 00:42:03 +0200 Subject: [PATCH] updates --- models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/models.py b/models.py index 4e5d487c..562306d1 100755 --- a/models.py +++ b/models.py @@ -69,6 +69,8 @@ def create_modules(module_defs): img_size = hyperparams['height'] # Define detection layer modules.add_module('yolo_%d' % i, YOLOLayer(anchors, nc, img_size, yolo_index)) + else: + print('Warning: Unrecognized Layer Type: ' + module_def['type']) # Register module list and number of output filters module_list.append(modules)