added return in check extension
Browse files- main_noweb.py +3 -1
main_noweb.py
CHANGED
@@ -130,13 +130,14 @@ def pose3d(video, kpt_threshold):
|
|
130 |
|
131 |
def pose2d(video, kpt_threshold):
|
132 |
#video = check_extension(video)
|
133 |
-
get_frames(video)
|
134 |
|
135 |
# Define new unique folder
|
136 |
add_dir = str(uuid.uuid4())
|
137 |
|
138 |
os.makedirs(add_dir)
|
139 |
print(check_fps(video))
|
|
|
140 |
result_generator = human(video,
|
141 |
vis_out_dir = add_dir,
|
142 |
radius = 5,
|
@@ -151,6 +152,7 @@ def pose2d(video, kpt_threshold):
|
|
151 |
out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm"))
|
152 |
kpoints = glob.glob(os.path.join(add_dir, "*.json"))
|
153 |
print(kpoints)
|
|
|
154 |
|
155 |
return "".join(out_file), "".join(kpoints)
|
156 |
|
|
|
130 |
|
131 |
def pose2d(video, kpt_threshold):
|
132 |
#video = check_extension(video)
|
133 |
+
video = get_frames(video)
|
134 |
|
135 |
# Define new unique folder
|
136 |
add_dir = str(uuid.uuid4())
|
137 |
|
138 |
os.makedirs(add_dir)
|
139 |
print(check_fps(video))
|
140 |
+
|
141 |
result_generator = human(video,
|
142 |
vis_out_dir = add_dir,
|
143 |
radius = 5,
|
|
|
152 |
out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm"))
|
153 |
kpoints = glob.glob(os.path.join(add_dir, "*.json"))
|
154 |
print(kpoints)
|
155 |
+
print(out_file)
|
156 |
|
157 |
return "".join(out_file), "".join(kpoints)
|
158 |
|