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 |
|