xmrt commited on
Commit
7c1031f
1 Parent(s): fc40636

make directory

Browse files
Files changed (1) hide show
  1. main.py +6 -3
main.py CHANGED
@@ -45,6 +45,7 @@ def tracking(video, model, boxes=True):
45
  return annotated_frame
46
 
47
  def show_tracking(video_content):
 
48
  video = cv2.VideoCapture(video_content)
49
 
50
  # Track
@@ -115,7 +116,7 @@ def pose2d(video):
115
 
116
 
117
  result_generator = human(video,
118
- vis_out_dir = vis_out_dir,
119
  return_vis=True,
120
  thickness=2,
121
  rebase_keypoint_height=True,
@@ -127,8 +128,10 @@ def pose2d(video):
127
  print(result)
128
 
129
 
130
- out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm"))
131
- out_file = os.path.join(vis_out_dir, "sample_flip.webm")
 
 
132
 
133
  return out_file
134
 
 
45
  return annotated_frame
46
 
47
  def show_tracking(video_content):
48
+ # https://docs.ultralytics.com/datasets/detect/coco/
49
  video = cv2.VideoCapture(video_content)
50
 
51
  # Track
 
116
 
117
 
118
  result_generator = human(video,
119
+ vis_out_dir = add_dir,
120
  return_vis=True,
121
  thickness=2,
122
  rebase_keypoint_height=True,
 
128
  print(result)
129
 
130
 
131
+ out_file = glob.glob(os.path.join(add_dir, "*.mp4")) + glob.glob(os.path.join(add_dir, "*.webm"))
132
+ #out_file = os.path.join(vis_out_dir, "sample_flip.webm")
133
+ print("OTFILE: ", out_file)
134
+ print("ADD DIR ", os.listdir(add_dir))
135
 
136
  return out_file
137