Added ffmpeg installation check

MrinalNextGen · Jan 19, 2025 · c59cf07 · c59cf07
1 parent 21e36b6
commit c59cf07
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/latentsync/pipelines/lipsync_pipeline.py b/latentsync/pipelines/lipsync_pipeline.py
@@ -31,7 +31,7 @@
 
 from ..models.unet import UNet3DConditionModel
 from ..utils.image_processor import ImageProcessor
-from ..utils.util import read_video, read_audio, write_video
+from ..utils.util import read_video, read_audio, write_video, check_ffmpeg_installed
 from ..whisper.audio2feature import Audio2Feature
 import tqdm
 import soundfile as sf
@@ -317,6 +317,8 @@ def __call__(
         is_train = self.unet.training
         self.unet.eval()
 
+        check_ffmpeg_installed()
+
         # 0. Define call parameters
         batch_size = 1
         device = self._execution_device

diff --git a/latentsync/utils/util.py b/latentsync/utils/util.py
@@ -363,3 +363,10 @@ def count_video_time(video_path):
     frame_count = video.get(cv2.CAP_PROP_FRAME_COUNT)
     fps = video.get(cv2.CAP_PROP_FPS)
     return frame_count / fps
+
+
+def check_ffmpeg_installed():
+    # Run the ffmpeg command with the -version argument to check if it's installed
+    result = subprocess.run("ffmpeg -version", stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
+    if not result.returncode == 0:
+        raise FileNotFoundError("ffmpeg not found, please install it by:\n    $ conda install -c conda-forge ffmpeg")