Browse Source

bug fix

master
Chris 8 years ago
parent
commit
8f810faaab
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      test_pretrained.py

+ 1
- 0
test_pretrained.py View File

@@ -63,6 +63,7 @@ def main(args):
cv2.imwrite(out_fn, raw) cv2.imwrite(out_fn, raw)


post, coords = post_process(raw) post, coords = post_process(raw)
print coords


out_fn = os.path.join(args.out_dir, f.replace('/','_')[:-4] + "_post.png") out_fn = os.path.join(args.out_dir, f.replace('/','_')[:-4] + "_post.png")
cv2.imwrite(out_fn, post) cv2.imwrite(out_fn, post)


Loading…
Cancel
Save