xmrt commited on
Commit
1e4500f
·
1 Parent(s): 582b742
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -94,6 +94,7 @@ def pose3d(video):
94
 
95
  result = [result for result in result_generator] #next(result_generator)
96
  out_file = glob.glob(os.path.join(vis_out_dir, "*"))
 
97
  print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
98
 
99
  return out_file
@@ -116,6 +117,7 @@ def pose2d(video):
116
 
117
  out_file = glob.glob(os.path.join(vis_out_dir, "*"))
118
  print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
 
119
 
120
  return out_file
121
 
@@ -138,6 +140,7 @@ def pose2dhand(video):
138
  result = [result for result in result_generator] #next(result_generator)
139
 
140
  out_file = glob.glob(os.path.join(vis_out_dir, "*"))
 
141
  print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
142
 
143
  return out_file
 
94
 
95
  result = [result for result in result_generator] #next(result_generator)
96
  out_file = glob.glob(os.path.join(vis_out_dir, "*"))
97
+ out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm"))
98
  print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
99
 
100
  return out_file
 
117
 
118
  out_file = glob.glob(os.path.join(vis_out_dir, "*"))
119
  print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
120
+ out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm"))
121
 
122
  return out_file
123
 
 
140
  result = [result for result in result_generator] #next(result_generator)
141
 
142
  out_file = glob.glob(os.path.join(vis_out_dir, "*"))
143
+ out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm"))
144
  print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
145
 
146
  return out_file