xmrt commited on
Commit
ee152c3
1 Parent(s): e207c04

removed [] from return

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -50,7 +50,7 @@ def show_tracking(video_content, vis_out_dir, model):
50
 
51
  # Prepare to save video
52
  #out_file = os.path.join(vis_out_dir, "track.mp4")
53
- out_file = "track.mp4"
54
  print("[INFO]: TRACK", out_file)
55
 
56
  fourcc = cv2.VideoWriter_fourcc(*"mp4v") # Codec for MP4 video
@@ -103,7 +103,7 @@ def infer(video, check):
103
 
104
  if i == "Detect and track":
105
  #continue
106
- [out_file] = show_tracking(video, vis_out_dir, inferencer)
107
 
108
  else:
109
  out_file = poses(inferencer, video, vis_out_dir)
 
50
 
51
  # Prepare to save video
52
  #out_file = os.path.join(vis_out_dir, "track.mp4")
53
+ out_file = ["track.mp4"]
54
  print("[INFO]: TRACK", out_file)
55
 
56
  fourcc = cv2.VideoWriter_fourcc(*"mp4v") # Codec for MP4 video
 
103
 
104
  if i == "Detect and track":
105
  #continue
106
+ out_file = show_tracking(video, vis_out_dir, inferencer)
107
 
108
  else:
109
  out_file = poses(inferencer, video, vis_out_dir)