freealise commited on
Commit
98d8d0a
1 Parent(s): c98c478

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -9
app.py CHANGED
@@ -25,17 +25,16 @@ ffmpeg_path = util.get_ffmpeg_path()
25
  mediapy.set_ffmpeg(ffmpeg_path)
26
 
27
 
28
-
29
- def do_interpolation(frame1, frame2, times_to_interpolate):
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, times_to_interpolate, interpolator))
36
 
37
  #print(frames)
38
- mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=1)
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]):