This commit is contained in:
Glenn Jocher 2020-02-17 15:10:11 -08:00
parent 426d5b82c6
commit 9880dcd6cd
1 changed files with 1 additions and 1 deletions

View File

@ -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