From 396e3b86b938be28a48ec8d9a46ae4739961aae1 Mon Sep 17 00:00:00 2001
From: Chris <christopher.tensmeyer@gmail.com>
Date: Tue, 12 Sep 2017 10:35:26 -0600
Subject: [PATCH] bug fix

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

diff --git a/test_pretrained.py b/test_pretrained.py
index 87aeed2..a021430 100644
--- a/test_pretrained.py
+++ b/test_pretrained.py
@@ -67,6 +67,7 @@ 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('\n')