Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ def do_interpolation(frame1, frame2, interpolation):
|
|
38 |
mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=25)
|
39 |
return f"{frame1}_to_{frame2}_out.mp4"
|
40 |
|
41 |
-
def get_frames(video_in, step, name,
|
42 |
frames = []
|
43 |
cap = cv2.VideoCapture(video_in)
|
44 |
cframes = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
|
@@ -70,7 +70,7 @@ def get_frames(video_in, step, name, resize):
|
|
70 |
ret, frame = cap.read()
|
71 |
if ret == False:
|
72 |
break
|
73 |
-
frame = cv2.resize(frame, (int(
|
74 |
cv2.imwrite(f"{name}_{step}{str(i)}.jpg",frame)
|
75 |
frames.append(f"{name}_{step}{str(i)}.jpg")
|
76 |
i+=1
|
@@ -90,11 +90,11 @@ def create_video(frames, fps, type):
|
|
90 |
return type + "_result.mp4"
|
91 |
|
92 |
|
93 |
-
def infer(url_in,interpolation,fps_output,
|
94 |
|
95 |
fps_output = logscale(fps_output)
|
96 |
# 1. break video into frames and get FPS
|
97 |
-
break_vid = get_frames(url_in, "vid_input_frame", "origin",
|
98 |
frames_list= break_vid[0]
|
99 |
fps = break_vid[1]
|
100 |
print(f"ORIGIN FPS: {fps}")
|
|
|
38 |
mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=25)
|
39 |
return f"{frame1}_to_{frame2}_out.mp4"
|
40 |
|
41 |
+
def get_frames(video_in, step, name, resize_n):
|
42 |
frames = []
|
43 |
cap = cv2.VideoCapture(video_in)
|
44 |
cframes = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
|
|
|
70 |
ret, frame = cap.read()
|
71 |
if ret == False:
|
72 |
break
|
73 |
+
frame = cv2.resize(frame, (int(resize_n), int(resize_n/2)))
|
74 |
cv2.imwrite(f"{name}_{step}{str(i)}.jpg",frame)
|
75 |
frames.append(f"{name}_{step}{str(i)}.jpg")
|
76 |
i+=1
|
|
|
90 |
return type + "_result.mp4"
|
91 |
|
92 |
|
93 |
+
def infer(url_in,interpolation,fps_output,resize_n):
|
94 |
|
95 |
fps_output = logscale(fps_output)
|
96 |
# 1. break video into frames and get FPS
|
97 |
+
break_vid = get_frames(url_in, "vid_input_frame", "origin", resize_n)
|
98 |
frames_list= break_vid[0]
|
99 |
fps = break_vid[1]
|
100 |
print(f"ORIGIN FPS: {fps}")
|