updates

2019-07-29 00:42:03 +02:00 · 2019-07-29 00:42:03 +02:00 · 9cf7f2215a
parent 1cfd370455
commit 9cf7f2215a
1 changed files with 2 additions and 0 deletions
--- 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)