fix(cli): correct import path for #75

src/python/pose_format/bin/directory.py

@@ -3,7 +3,7 @@
 import argparse
 import os
 
-from pose_format.bin.bin import pose_video
+from pose_format.bin.pose_estimation import pose_video
 from tqdm import tqdm
 
 

src/python/pose_format/bin/pose_estimation.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python
 import argparse
+import os
 
 import cv2
 from pose_format.utils.holistic import load_holistic
@@ -53,4 +54,7 @@ def main():
 
  args = parser.parse_args()
 
+ if not os.path.exists(args.i):
+ raise FileNotFoundError(f"Video file {args.i} not found")
+
  pose_video(args.i, args.o, args.format)

src/python/pose_format/bin/pose_visualizer.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 
 import argparse
+import os
 
 from pose_format import Pose
 from pose_format.pose_visualizer import PoseVisualizer
@@ -17,9 +18,12 @@ def visualize_pose(pose_path: str, video_path: str):
 
 def main():
  parser = argparse.ArgumentParser()
- parser.add_argument('-i', required=True, type=str, help='path to input video file')
- parser.add_argument('-o', required=True, type=str, help='path to output pose file')
+ parser.add_argument('-i', required=True, type=str, help='path to input pose file')
+ parser.add_argument('-o', required=True, type=str, help='path to output video file')
 
  args = parser.parse_args()
 
+ if not os.path.exists(args.i):
+ raise FileNotFoundError(f"Pose file {args.i} not found")
+
  visualize_pose(args.i, args.o)