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

updates

This commit is contained in:
Glenn Jocher 2019-08-12 12:25:26 +02:00
parent 89d0aa7164
commit daaa8194a9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -201,7 +201,7 @@ class Darknet(nn.Module):
x = torch.cat([layer_outputs[i] for i in layer_i], 1)
# print(''), [print(layer_outputs[i].shape) for i in layer_i], print(x.shape)
elif mtype == 'shortcut':
x += layer_outputs[int(mdef['from'])]
x = layer_outputs[-1] + layer_outputs[int(mdef['from']) ]
elif mtype == 'yolo':
x = module(x, img_size)
output.append(x)