OpenSourceJoker/LivePortrait
3
0
Fork 0
mirror of https://github.com/KwaiVGI/LivePortrait.git synced 2024-12-22 12:22:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: audio stream set false without ffmepg and ffprobe (#100)

Browse Source
This commit is contained in:
Jianzhu Guo 2024-07-11 16:20:30 +08:00 committed by GitHub
parent 877295a776
commit 7c77e29442
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/utils/video.py
View File

@ -1,7 +1,9 @@
# coding: utf-8
"""
functions for processing video
Functions for processing video
ATTENTION: you need to install ffmpeg and ffprobe in your env!
"""
import os.path as osp
@ -9,19 +11,11 @@ import numpy as np
import subprocess
import imageio
import cv2
from rich.progress import track
from .rprint import rlog as log
# try:
# import ffmpeg
# except ImportError as e:
# log(f'Try to install ffmpeg by: pip install ffmpeg-python==0.2.0', style='bold red')
# raise(e)
from rich.progress import track
from .helper import prefix
from .rprint import rprint as print
from .helper import prefix
def exec_cmd(cmd):
@ -146,7 +140,7 @@ def get_fps(filepath, default_fps=25):
if fps in (0, None):
fps = default_fps
except Exception as e:
print(e)
log(e)
fps = default_fps
return fps
@ -171,13 +165,17 @@ def has_audio_stream(video_path: str) -> bool:
video_path
]
result = subprocess.run(cmd, capture_output=True, text=True)
if result.returncode != 0:
log(f"Error occurred while probing video: {result.stderr}")
return False
try:
result = subprocess.run(cmd, capture_output=True, text=True)
if result.returncode != 0:
log(f"Error occurred while probing video: {result.stderr}")
return False
# Check if there is any output from ffprobe command
return bool(result.stdout.strip())
# Check if there is any output from ffprobe command
return bool(result.stdout.strip())
except Exception as e:
log(f"Error occurred while probing video: {video_path}, you may need to install ffprobe! Now set audio to false!", style="bold red")
return False
def add_audio_to_video(silent_video_path: str, audio_video_path: str, output_video_path: str):