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 00:43:04 +02:00
parent 3fca16d3ce
commit 89d0aa7164
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models.py
View File

@ -201,8 +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':
layer_i = int(mdef['from'])
x = layer_outputs[-1] + layer_outputs[layer_i]
x += layer_outputs[int(mdef['from'])]
elif mtype == 'yolo':
x = module(x, img_size)
output.append(x)