blocks
Browse files- main_blocks.py +2 -2
main_blocks.py
CHANGED
@@ -44,11 +44,11 @@ def tracking(video, model, boxes=True):
|
|
44 |
|
45 |
return annotated_frame
|
46 |
|
47 |
-
def show_tracking(video_content
|
48 |
video = cv2.VideoCapture(video_content)
|
49 |
|
50 |
# Track
|
51 |
-
video_track = tracking(video_content,
|
52 |
|
53 |
# Prepare to save video
|
54 |
#out_file = os.path.join(vis_out_dir, "track.mp4")
|
|
|
44 |
|
45 |
return annotated_frame
|
46 |
|
47 |
+
def show_tracking(video_content):
|
48 |
video = cv2.VideoCapture(video_content)
|
49 |
|
50 |
# Track
|
51 |
+
video_track = tracking(video_content, track_model.track)
|
52 |
|
53 |
# Prepare to save video
|
54 |
#out_file = os.path.join(vis_out_dir, "track.mp4")
|