modified demo script

John Harrington · John Harrington · commit 420bca884bff · 2021-05-27T22:52:58.000-04:00
diff --git a/demo/demo.py b/demo/demo.py
@@ -310,8 +310,8 @@ def main():
             print('Video ended')
             break
 
-    print(keypoints)
-    np.save(f"{args.output_dir}/keypoints.dat", keypoints)
+    np.save(f"{args.output_dir}/keypoints", keypoints)
+    print(f'keypoint saved to {args.output_dir}/keypoints.npy')
     cv2.destroyAllWindows()
     vidcap.release()
     if args.write:
