make directory
Browse files
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 |
|