Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2020-01-27 17:03:27 -05:00
parent 72680a5992
commit 1961678177
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -233,6 +233,8 @@ class Darknet(nn.Module):
x = module(x) x = module(x)
elif mtype == 'route': elif mtype == 'route':
layers = [int(x) for x in mdef['layers'].split(',')] layers = [int(x) for x in mdef['layers'].split(',')]
if verbose:
print('route concatenating %s' % ([layer_outputs[i].shape for i in layers]))
if len(layers) == 1: if len(layers) == 1:
x = layer_outputs[layers[0]] x = layer_outputs[layers[0]]
else: else: