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

updates

This commit is contained in:
Glenn Jocher 2019-12-29 14:28:56 -08:00
parent f964f29567
commit 894218390b
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

@ -260,8 +260,7 @@ class Darknet(nn.Module):
elif mtype == 'shortcut': elif mtype == 'shortcut':
x = x + layer_outputs[int(mdef['from'])] x = x + layer_outputs[int(mdef['from'])]
elif mtype == 'yolo': elif mtype == 'yolo':
x = module(x, img_size) output.append(module(x, img_size))
output.append(x)
layer_outputs.append(x if i in self.routs else []) layer_outputs.append(x if i in self.routs else [])
if self.training: if self.training: