mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
66 lines
1.3 KiB
Python
66 lines
1.3 KiB
Python
from collections.abc import Sequence
|
|
from distutils.util import strtobool as dist_strtobool
|
|
import sys
|
|
|
|
import numpy
|
|
|
|
|
|
def strtobool(x):
|
|
# distutils.util.strtobool returns integer, but it's confusing,
|
|
return bool(dist_strtobool(x))
|
|
|
|
|
|
def get_commandline_args():
|
|
extra_chars = [
|
|
" ",
|
|
";",
|
|
"&",
|
|
"(",
|
|
")",
|
|
"|",
|
|
"^",
|
|
"<",
|
|
">",
|
|
"?",
|
|
"*",
|
|
"[",
|
|
"]",
|
|
"$",
|
|
"`",
|
|
'"',
|
|
"\\",
|
|
"!",
|
|
"{",
|
|
"}",
|
|
]
|
|
|
|
# Escape the extra characters for shell
|
|
argv = [
|
|
arg.replace("'", "'\\''")
|
|
if all(char not in arg for char in extra_chars)
|
|
else "'" + arg.replace("'", "'\\''") + "'"
|
|
for arg in sys.argv
|
|
]
|
|
|
|
return sys.executable + " " + " ".join(argv)
|
|
|
|
|
|
def is_scipy_wav_style(value):
|
|
# If Tuple[int, numpy.ndarray] or not
|
|
return (
|
|
isinstance(value, Sequence)
|
|
and len(value) == 2
|
|
and isinstance(value[0], int)
|
|
and isinstance(value[1], numpy.ndarray)
|
|
)
|
|
|
|
|
|
def assert_scipy_wav_style(value):
|
|
assert is_scipy_wav_style(
|
|
value
|
|
), "Must be Tuple[int, numpy.ndarray], but got {}".format(
|
|
type(value)
|
|
if not isinstance(value, Sequence)
|
|
else "{}[{}]".format(type(value), ", ".join(str(type(v)) for v in value))
|
|
)
|