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}")
|