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

make directory

Browse files
Files changed (1) hide show
  1. main.py +6 -1
main.py CHANGED
@@ -65,6 +65,7 @@ def show_tracking(video_content):
65
  # Go through frames and write them
66
  for frame_track in video_track:
67
  result_track = frame_track[0].plot() # plot a BGR numpy array of predictions
 
68
  print("[INFO] Done with frames")
69
  #print(type(result_pose)) numpy ndarray
70
  out_track.write(result_track)
@@ -82,6 +83,7 @@ def pose3d(video):
82
  #vidname = video.split("/")[-1]
83
  vis_out_dir = os.path.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)
@@ -105,8 +107,10 @@ def pose3d(video):
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,
@@ -129,6 +133,7 @@ def pose2dhand(video):
129
  os.makedirs(vis_out_dir)
130
 
131
  print("[INFO]: CURRENT OUT DIR: ", vis_out_dir)
 
132
 
133
  vis_out_dir = str(uuid.uuid4())
134
 
 
65
  # Go through frames and write them
66
  for frame_track in video_track:
67
  result_track = frame_track[0].plot() # plot a BGR numpy array of predictions
68
+
69
  print("[INFO] Done with frames")
70
  #print(type(result_pose)) numpy ndarray
71
  out_track.write(result_track)
 
83
  #vidname = video.split("/")[-1]
84
  vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
85
  print("[INFO]: CURRENT OUT DIR: ", vis_out_dir)
86
+ print("[INFO]: LIST CURRENT OUT DIR: ", os.listdir("/".join(video.split("/")[:-1])))
87
  os.makedirs(vis_out_dir)
88
 
89
  #full name = os.path.join(vis_out_dir, vidname)
 
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)
111
+ print("[INFO]: CURRENT OUT DIR: ", vis_out_dir)
112
+ print("[INFO]: LIST CURRENT OUT DIR: ", os.listdir("/".join(video.split("/")[:-1])))
113
+
114
 
115
  result_generator = human(video,
116
  vis_out_dir = vis_out_dir,
 
133
  os.makedirs(vis_out_dir)
134
 
135
  print("[INFO]: CURRENT OUT DIR: ", vis_out_dir)
136
+ print("[INFO]: LIST CURRENT OUT DIR: ", os.listdir("/".join(video.split("/")[:-1])))
137
 
138
  vis_out_dir = str(uuid.uuid4())
139