Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 20:19:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

support multi audio types

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11657396

    * support multi audio format

* SUPPORT_AUDIO_TYPE_SETS add .
This commit is contained in:
wucong.lyb
2023-02-14 08:42:19 +00:00
committed by wenmeng.zwm
parent 9f97259017
commit 61b1ee024a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modelscope/utils/audio/audio_utils.py
View File

@@ -12,6 +12,7 @@ import numpy as np
from modelscope.fileio.file import HTTPStorage
SEGMENT_LENGTH_TRAIN = 16000
SUPPORT_AUDIO_TYPE_SETS = ('.flac', '.mp3', '.ogg', '.opus', '.wav', '.pcm')
class TtsTrainType(object):
@@ -176,7 +177,7 @@ def generate_scp_from_url(url: str, key: str = None):
wav_scp_path = url
return wav_scp_path, raw_inputs
# for local wav file inputs
if os.path.exists(url) and (url.lower().endswith('.wav')):
if os.path.exists(url) and (url.lower().endswith(SUPPORT_AUDIO_TYPE_SETS)):
wav_scp_path = url
return wav_scp_path, raw_inputs
# for wav url, download bytes data
@@ -228,8 +229,7 @@ def generate_scp_for_sv(url: str, key: str = None):
wav_scp_path = url
return wav_scp_path
# for local wav file inputs
if os.path.exists(url) and (url.lower().endswith('.wav')
or url.lower().endswith('.pcm')):
if os.path.exists(url) and (url.lower().endswith(SUPPORT_AUDIO_TYPE_SETS)):
wav_path = url
work_dir = tempfile.TemporaryDirectory().name
if not os.path.exists(work_dir):