make directory
Browse files
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 =
|
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(
|
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 |
|