From 5d55a4253efe59e397d2ec0d426bfd23c9538c7d Mon Sep 17 00:00:00 2001
From: Chris <christopher.tensmeyer@gmail.com>
Date: Tue, 12 Sep 2017 10:36:25 -0600
Subject: [PATCH] bug fix

---
 test_pretrained.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/test_pretrained.py b/test_pretrained.py
index a021430..967a388 100644
--- a/test_pretrained.py
+++ b/test_pretrained.py
@@ -67,9 +67,8 @@ def main(args):
 		cv2.imwrite(out_fn, raw)
 
 		post, coords = post_process(raw)
-		print coords.shape
 		for idx2 in [1, 2, 3, 0]:
-			fd.write('%d,%d,' % (width * coords[0][idx2][0] / 256., height * coords[0][idx2][1] / 256.))
+			fd.write('%d,%d,' % (width * coords[idx2][0] / 256., height * coords[idx2][1] / 256.))
 		fd.write('\n')
 
 		out_fn = os.path.join(args.out_dir, f.replace('/','_')[:-4] + "_post.png")