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