From 42d1dc89fd9588709651f3753c95c60cc1b39672 Mon Sep 17 00:00:00 2001 From: Chris Date: Tue, 12 Sep 2017 10:16:13 -0600 Subject: [PATCH] bug fix --- test_pretrained.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test_pretrained.py b/test_pretrained.py index 04d7993..e20a265 100644 --- a/test_pretrained.py +++ b/test_pretrained.py @@ -37,8 +37,8 @@ def predict(network, im, output_blob, args): def presolve(net, args): - net.blobs["data"].reshape(args.batch_size, 3, args.image_size, args.image_size) - net.blobs["gt"].reshape(args.batch_size, 1, args.image_size, args.image_size) + net.blobs["data"].reshape(1, 3, args.image_size, args.image_size) + net.blobs["gt"].reshape(1, 1, args.image_size, args.image_size) def main(args):