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

updates

This commit is contained in:
Glenn Jocher 2020-02-19 18:06:53 -08:00
parent 6fbab656c8
commit 7f1b2bfe08
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
models.py
View File

@ -274,7 +274,9 @@ class Darknet(nn.Module):
x = module(x)
elif mtype == 'shortcut': # sum
if verbose:
print('shortcut/add %s + %s' % (list(x.shape), [list(out[i].shape) for i in module.layers]))
l = [i] + module.layers # layers
s = [list(x.shape)] + [list(out[i].shape) for i in module.layers] # shapes
print('shortcut/add: ' + ' + '.join(['layer %g %s' % x for x in zip(l, s)]))
x = module(x, out) # weightedFeatureFusion()
elif mtype == 'route': # concat
layers = mdef['layers']
@ -293,7 +295,7 @@ class Darknet(nn.Module):
yolo_out.append(module(x, img_size))
out.append(x if i in self.routs else [])
if verbose:
print(i, list(x.shape))
print('%g/%g -' % (i, len(self.module_list)), list(x.shape))
if self.training: # train
return yolo_out