xmrt commited on
Commit
435673b
1 Parent(s): d151311

make directory

Browse files
Files changed (1) hide show
  1. main.py +5 -4
main.py CHANGED
@@ -82,6 +82,7 @@ def pose3d(video):
82
  #vidname = video.split("/")[-1]
83
  vis_out_dir = "/".join(["/".join(video.split("/")[:-1]), add_dir])
84
  print("[INFO]: CURRENT OUT DIR: ", vis_out_dir)
 
85
 
86
  #full name = os.path.join(vis_out_dir, vidname)
87
 
@@ -103,9 +104,9 @@ def pose3d(video):
103
 
104
  def pose2d(video):
105
  add_dir = str(uuid.uuid4())
106
- vis_out_dir = "/".join(["/".join(video.split("/")[:-1]), add_dir])
107
  print("[INFO]: CURRENT OUT DIR: ", vis_out_dir)
108
-
109
 
110
  result_generator = human(video,
111
  vis_out_dir = vis_out_dir,
@@ -118,7 +119,6 @@ def pose2d(video):
118
 
119
  result = [result for result in result_generator] #next(result_generator)
120
 
121
- out_file = glob.glob(os.path.join(vis_out_dir, "*"))
122
  print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
123
  out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm"))
124
 
@@ -128,7 +128,8 @@ def pose2d(video):
128
  def pose2dhand(video):
129
  add_dir = str(uuid.uuid4())
130
  vis_out_dir = "/".join(["/".join(video.split("/")[:-1]), add_dir])
131
-
 
132
  print("[INFO]: CURRENT OUT DIR: ", vis_out_dir)
133
 
134
  vis_out_dir = str(uuid.uuid4())
 
82
  #vidname = video.split("/")[-1]
83
  vis_out_dir = "/".join(["/".join(video.split("/")[:-1]), add_dir])
84
  print("[INFO]: CURRENT OUT DIR: ", vis_out_dir)
85
+ os.makedirs(vis_out_dir)
86
 
87
  #full name = os.path.join(vis_out_dir, vidname)
88
 
 
104
 
105
  def pose2d(video):
106
  add_dir = str(uuid.uuid4())
107
+ vis_out_dir = os.path.join(["/".join(video.split("/")[:-1]), add_dir])
108
  print("[INFO]: CURRENT OUT DIR: ", vis_out_dir)
109
+ os.makedirs(vis_out_dir)
110
 
111
  result_generator = human(video,
112
  vis_out_dir = vis_out_dir,
 
119
 
120
  result = [result for result in result_generator] #next(result_generator)
121
 
 
122
  print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
123
  out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm"))
124
 
 
128
  def pose2dhand(video):
129
  add_dir = str(uuid.uuid4())
130
  vis_out_dir = "/".join(["/".join(video.split("/")[:-1]), add_dir])
131
+ os.makedirs(vis_out_dir)
132
+
133
  print("[INFO]: CURRENT OUT DIR: ", vis_out_dir)
134
 
135
  vis_out_dir = str(uuid.uuid4())