Update app.py
Browse files
app.py
CHANGED
@@ -25,17 +25,16 @@ ffmpeg_path = util.get_ffmpeg_path()
|
|
25 |
mediapy.set_ffmpeg(ffmpeg_path)
|
26 |
|
27 |
|
28 |
-
|
29 |
-
|
30 |
print(frame1, frame2)
|
31 |
input_frames = [frame1, frame2]
|
32 |
-
#times_to_interpolate = 2
|
33 |
frames = list(
|
34 |
util.interpolate_recursively_from_files(
|
35 |
-
input_frames,
|
36 |
|
37 |
#print(frames)
|
38 |
-
mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=
|
39 |
return f"{frame1}_to_{frame2}_out.mp4"
|
40 |
|
41 |
def get_frames(video_in, step, name):
|
@@ -108,13 +107,10 @@ def infer(url_in,interpolation,fps_output):
|
|
108 |
result_frames = []
|
109 |
print("set stop frames to: " + str(n_frame))
|
110 |
|
111 |
-
|
112 |
-
|
113 |
-
|
114 |
for idx, frame in enumerate(frames_list[0:int(n_frame)]):
|
115 |
if idx < len(frames_list) - 1:
|
116 |
next_frame = frames_list[idx+1]
|
117 |
-
interpolated_frames = do_interpolation(frame, next_frame,interpolation) # should return a list of 3 interpolated frames
|
118 |
break_interpolated_video = get_frames(interpolated_frames, "interpol",f"{idx}_")
|
119 |
print(break_interpolated_video[0])
|
120 |
for j, img in enumerate(break_interpolated_video[0][0:len(break_interpolated_video[0])-1]):
|
|
|
25 |
mediapy.set_ffmpeg(ffmpeg_path)
|
26 |
|
27 |
|
28 |
+
def do_interpolation(frame1, frame2, interpolation):
|
29 |
+
print("tween frames: " + str(interpolation))
|
30 |
print(frame1, frame2)
|
31 |
input_frames = [frame1, frame2]
|
|
|
32 |
frames = list(
|
33 |
util.interpolate_recursively_from_files(
|
34 |
+
input_frames, interpolation, interpolator))
|
35 |
|
36 |
#print(frames)
|
37 |
+
mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=25)
|
38 |
return f"{frame1}_to_{frame2}_out.mp4"
|
39 |
|
40 |
def get_frames(video_in, step, name):
|
|
|
107 |
result_frames = []
|
108 |
print("set stop frames to: " + str(n_frame))
|
109 |
|
|
|
|
|
|
|
110 |
for idx, frame in enumerate(frames_list[0:int(n_frame)]):
|
111 |
if idx < len(frames_list) - 1:
|
112 |
next_frame = frames_list[idx+1]
|
113 |
+
interpolated_frames = do_interpolation(frame, next_frame, interpolation) # should return a list of 3 interpolated frames
|
114 |
break_interpolated_video = get_frames(interpolated_frames, "interpol",f"{idx}_")
|
115 |
print(break_interpolated_video[0])
|
116 |
for j, img in enumerate(break_interpolated_video[0][0:len(break_interpolated_video[0])-1]):
|