interp multi

libs/film/predict.py

@@ -71,4 +71,9 @@ class Predictor(cog.BasePredictor):
         del self.interpolator
         tf.keras.backend.clear_session()
 
-        return frames
+        ffmpeg_path = util.get_ffmpeg_path()
+        mediapy.set_ffmpeg(ffmpeg_path)
+        out_path = Path(tempfile.mkdtemp()) / "out.mp4"
+        mediapy.write_video(str(out_path), frames, fps=30)
+
+        return out_path

main.py

@@ -1253,12 +1253,16 @@ def interpolate_frames(frame1, frame2, times_to_interp):
     frame1.thumbnail(thumb_size)
     frame2.thumbnail(thumb_size)
 
-    results = film.predict(frame1, frame2, times_to_interp)
-    
-    if not isinstance(results, list):
-        results = [results]
+    out_vid = film.predict(frame1, frame2, int(times_to_interp))
+    print(out_vid)
+    if str(out_vid).endswith('.mp4'):
+        results = extract_frames(out_vid, 30)
+        results = results[1:-1]
+    else:
+        results = [Image.open(out_vid)]
 
-    results = [Image.open(str(r)) for r in results]
+    print(results)
+    
     for r in results:
         r.thumbnail((width, height))
 
@@ -1266,6 +1270,7 @@ def interpolate_frames(frame1, frame2, times_to_interp):
 
     return results, getThumbnails(results)
 
+
 def run_interpolate_frames(frame1, frame2, times_to_interp):
     with Pool() as pool:
         results = pool.starmap(interpolate_frames, [(frame1, frame2, times_to_interp)])