Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,15 +45,15 @@ def get_frames(video_in):
|
|
| 45 |
|
| 46 |
return frames, fps
|
| 47 |
|
| 48 |
-
def
|
| 49 |
image = Image.open(i)
|
| 50 |
|
| 51 |
#image = np.array(image)
|
| 52 |
|
| 53 |
image = openpose(image)
|
| 54 |
#image = Image.fromarray(image)
|
| 55 |
-
image.save("
|
| 56 |
-
return "
|
| 57 |
|
| 58 |
def create_video(frames, fps, type):
|
| 59 |
print("building video result")
|
|
@@ -86,7 +86,7 @@ def infer(video_in):
|
|
| 86 |
print("set stop frames to: " + str(n_frame))
|
| 87 |
|
| 88 |
for i in frames_list[0:int(n_frame)]:
|
| 89 |
-
|
| 90 |
result_frames.append(canny_frame)
|
| 91 |
print("frame " + i + "/" + str(n_frame) + ": done;")
|
| 92 |
|
|
@@ -109,7 +109,7 @@ title="""
|
|
| 109 |
"
|
| 110 |
>
|
| 111 |
<h1 style="font-weight: 600; margin-bottom: 7px;">
|
| 112 |
-
Video to
|
| 113 |
</h1>
|
| 114 |
</div>
|
| 115 |
|
|
|
|
| 45 |
|
| 46 |
return frames, fps
|
| 47 |
|
| 48 |
+
def get_openpose_filter(i):
|
| 49 |
image = Image.open(i)
|
| 50 |
|
| 51 |
#image = np.array(image)
|
| 52 |
|
| 53 |
image = openpose(image)
|
| 54 |
#image = Image.fromarray(image)
|
| 55 |
+
image.save("openpose_frame_" + str(i) + ".jpeg")
|
| 56 |
+
return "openpose_frame_" + str(i) + ".jpeg"
|
| 57 |
|
| 58 |
def create_video(frames, fps, type):
|
| 59 |
print("building video result")
|
|
|
|
| 86 |
print("set stop frames to: " + str(n_frame))
|
| 87 |
|
| 88 |
for i in frames_list[0:int(n_frame)]:
|
| 89 |
+
openpose_frame = get_openpose_filter(i)
|
| 90 |
result_frames.append(canny_frame)
|
| 91 |
print("frame " + i + "/" + str(n_frame) + ": done;")
|
| 92 |
|
|
|
|
| 109 |
"
|
| 110 |
>
|
| 111 |
<h1 style="font-weight: 600; margin-bottom: 7px;">
|
| 112 |
+
Video to OpenPose
|
| 113 |
</h1>
|
| 114 |
</div>
|
| 115 |
|