Add VGG networks.

This commit is contained in:
2018-03-20 16:14:10 +01:00
parent 42a5d882d6
commit 0a8aa4e2eb
9 changed files with 1544 additions and 3 deletions

View File

@@ -33,7 +33,8 @@ def load_net(args):
def deinplace(args, net):
outputs = {}
for layer in net.layer:
layers = net.layer if net.layer else net.layers
for layer in layers:
for idx, bottom in enumerate(layer.bottom):
if bottom in outputs and bottom != outputs[bottom]:
if args.verbose: