From daaa8194a9c082766a297fefb4e473e8f49a8359 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 12 Aug 2019 12:25:26 +0200 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 109b336c..6d36f1ef 100755 --- a/models.py +++ b/models.py @@ -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)