This commit is contained in:
Glenn Jocher 2019-07-29 00:42:03 +02:00
parent 1cfd370455
commit 9cf7f2215a
1 changed files with 2 additions and 0 deletions

View File

@ -69,6 +69,8 @@ def create_modules(module_defs):
img_size = hyperparams['height'] img_size = hyperparams['height']
# Define detection layer # Define detection layer
modules.add_module('yolo_%d' % i, YOLOLayer(anchors, nc, img_size, yolo_index)) 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 # Register module list and number of output filters
module_list.append(modules) module_list.append(modules)