From 9880dcd6cde81d277486b452dba266b23eeae0e6 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 17 Feb 2020 15:10:11 -0800 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 69bf0c9b..a48f35d2 100755 --- a/models.py +++ b/models.py @@ -67,8 +67,8 @@ def create_modules(module_defs, img_size, arc): # modules = nn.Upsample(scale_factor=1/float(mdef[i+1]['stride']), mode='nearest') # reorg3d elif mdef['type'] == 'shortcut': # nn.Sequential() placeholder for 'shortcut' layer - filters = output_filters[int(mdef['from'])] layer = int(mdef['from']) + filters = output_filters[layer] routs.extend([i + layer if layer < 0 else layer]) elif mdef['type'] == 'reorg3d': # yolov3-spp-pan-scale