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')