| | |
|---|
| | | |
|---|
| | | def is_ffmpeg_installed(): |
|---|
| | | try: |
|---|
| | | # 尝试运行ffmpeg命令并获取其版本信息 |
|---|
| | | output = subprocess.check_output(['ffmpeg', '-version'], stderr=subprocess.STDOUT) |
|---|
| | | return 'ffmpeg version' in output.decode('utf-8') |
|---|
| | | except (subprocess.CalledProcessError, FileNotFoundError): |
|---|
| | | # 若运行ffmpeg命令失败，则认为ffmpeg未安装 |
|---|
| | | return False |
|---|
| | | |
|---|
| | | use_ffmpeg=False |
|---|
| | | if is_ffmpeg_installed(): |
|---|
| | | use_ffmpeg = True |
|---|
| | | else: |
|---|
| | | print("Notice: ffmpeg is not installed. torchaudio is used to load audio") |
|---|
| | | print("Notice: ffmpeg is not installed. torchaudio is used to load audio\n" |
|---|
| | | "If you want to use ffmpeg backend to load audio, please install it by:" |
|---|
| | | "\n\tsudo apt install ffmpeg # ubuntu" |
|---|
| | | "\n\t# brew install ffmpeg # mac") |
|---|
| | | |
|---|
| | | def load_audio_text_image_video(data_or_path_or_list, fs: int = 16000, audio_fs: int = 16000, data_type="sound", tokenizer=None, **kwargs): |
|---|
| | | if isinstance(data_or_path_or_list, (list, tuple)): |
|---|
