xmrt commited on
Commit
5e4a9da
·
1 Parent(s): 8d4e075

added return in check extension

Browse files
Files changed (1) hide show
  1. main_noweb.py +3 -1
main_noweb.py CHANGED
@@ -130,13 +130,14 @@ def pose3d(video, kpt_threshold):
130
 
131
  def pose2d(video, kpt_threshold):
132
  #video = check_extension(video)
133
- get_frames(video)
134
 
135
  # Define new unique folder
136
  add_dir = str(uuid.uuid4())
137
 
138
  os.makedirs(add_dir)
139
  print(check_fps(video))
 
140
  result_generator = human(video,
141
  vis_out_dir = add_dir,
142
  radius = 5,
@@ -151,6 +152,7 @@ def pose2d(video, kpt_threshold):
151
  out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm"))
152
  kpoints = glob.glob(os.path.join(add_dir, "*.json"))
153
  print(kpoints)
 
154
 
155
  return "".join(out_file), "".join(kpoints)
156
 
 
130
 
131
  def pose2d(video, kpt_threshold):
132
  #video = check_extension(video)
133
+ video = get_frames(video)
134
 
135
  # Define new unique folder
136
  add_dir = str(uuid.uuid4())
137
 
138
  os.makedirs(add_dir)
139
  print(check_fps(video))
140
+
141
  result_generator = human(video,
142
  vis_out_dir = add_dir,
143
  radius = 5,
 
152
  out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm"))
153
  kpoints = glob.glob(os.path.join(add_dir, "*.json"))
154
  print(kpoints)
155
+ print(out_file)
156
 
157
  return "".join(out_file), "".join(kpoints)
158