xmrt commited on
Commit
fc40636
1 Parent(s): 559fc59

make directory

Browse files
Files changed (1) hide show
  1. main.py +7 -1
main.py CHANGED
@@ -105,6 +105,8 @@ def pose3d(video):
105
 
106
 
107
  def pose2d(video):
 
 
108
  add_dir = str(uuid.uuid4())
109
  vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
110
  os.makedirs(vis_out_dir)
@@ -121,8 +123,12 @@ def pose2d(video):
121
  device="cuda"
122
  )
123
 
124
- result = [result for result in result_generator] #next(result_generator)
 
 
 
125
  out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm"))
 
126
 
127
  return out_file
128
 
 
105
 
106
 
107
  def pose2d(video):
108
+ vidname = video.split("/")[-1]
109
+ print("VIDNAME", vidname)
110
  add_dir = str(uuid.uuid4())
111
  vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
112
  os.makedirs(vis_out_dir)
 
123
  device="cuda"
124
  )
125
 
126
+ result = [result for result in result_generator] #next(result_generator)
127
+ print(result)
128
+
129
+
130
  out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm"))
131
+ out_file = os.path.join(vis_out_dir, "sample_flip.webm")
132
 
133
  return out_file
134