switched to imageio

Dockerfile

@@ -16,7 +16,6 @@ RUN apt-get update -o Acquire::AllowInsecureRepositories=true \
 RUN pip install --upgrade pip \
     && pip install --upgrade pip setuptools wheel \
     && pip install --no-cache-dir -r requirements.txt \
-    && pip install opencv-python --verbose \
     && pip install jupyter
 
 WORKDIR /data

app.py

@@ -5,7 +5,7 @@ import numpy as np
 from pathlib import Path
 import dnnlib
 from dnnlib import tflib
-import cv2
+import imageio
 import os
 import subprocess
 
@@ -69,14 +69,14 @@ random_img_latent_code = Gs.components.mapping.run(z,None)
 random_img_latent_code -= 0.7*latent_controls['time']
 vid = frame_to_frame(random_img_latent_code)
 
-fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Codec for .mp4
-temp_video_path="output.mp4"
-out = cv2.VideoWriter(temp_video_path, fourcc, 20.0, (256, 256), isColor=False)
 
+temp_video_path="output.mp4"
+writer=imageio.get_writer(temp_video_path, fps=20)
 for i in range(vid.shape[0]):
     frame = vid[i]
-    out.write(frame)
-out.release()
+    writer.append_data(frame)
+
+writer.close()
 out_path = "fixed_out.mp4"
 command = ["ffmpeg", "-i", temp_video_path, "-vcodec", "libx264", out_path]
 subprocess.run(command)

requirements.txt

@@ -5,3 +5,4 @@ ipywidgets
 streamlit
 numpy
 pathlib
+imagio