Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
DeepSpeech
提交
f8375764
D
DeepSpeech
项目概览
PaddlePaddle
/
DeepSpeech
大约 1 年 前同步成功
通知
206
Star
8425
Fork
1598
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
245
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
D
DeepSpeech
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
245
Issue
245
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
f8375764
编写于
2月 25, 2022
作者:
L
lym0302
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add paddlespeech stats, test=doc
上级
e18802e1
变更
4
显示空白变更内容
内联
并排
Showing
4 changed file
with
161 addition
and
0 deletion
+161
-0
paddlespeech/cli/__init__.py
paddlespeech/cli/__init__.py
+1
-0
paddlespeech/cli/stats/__init__.py
paddlespeech/cli/stats/__init__.py
+14
-0
paddlespeech/cli/stats/infer.py
paddlespeech/cli/stats/infer.py
+145
-0
setup.py
setup.py
+1
-0
未找到文件。
paddlespeech/cli/__init__.py
浏览文件 @
f8375764
...
...
@@ -20,5 +20,6 @@ from .cls import CLSExecutor
from
.st
import
STExecutor
from
.text
import
TextExecutor
from
.tts
import
TTSExecutor
from
.stats
import
StatsExecutor
_locale
.
_getdefaultlocale
=
(
lambda
*
args
:
[
'en_US'
,
'utf8'
])
paddlespeech/cli/stats/__init__.py
0 → 100644
浏览文件 @
f8375764
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from
.infer
import
StatsExecutor
paddlespeech/cli/stats/infer.py
0 → 100644
浏览文件 @
f8375764
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import
argparse
from
typing
import
List
from
prettytable
import
PrettyTable
from
..log
import
logger
from
..utils
import
cli_register
from
..utils
import
stats_wrapper
__all__
=
[
'StatsExecutor'
]
model_name_format
=
{
'asr'
:
'Model-Language-Sample Rate'
,
'cls'
:
'Model-Sample Rate'
,
'st'
:
'Model-Source language-Target language'
,
'text'
:
'Model-Task-Sample Rate'
,
'tts'
:
'Model-Language'
}
@
cli_register
(
name
=
'paddlespeech.stats'
,
description
=
'Text infer command.'
)
class
StatsExecutor
():
def
__init__
(
self
):
super
(
StatsExecutor
,
self
).
__init__
()
self
.
parser
=
argparse
.
ArgumentParser
(
prog
=
'paddlespeech.stats'
,
add_help
=
True
)
self
.
parser
.
add_argument
(
'--task'
,
type
=
str
,
default
=
'asr'
,
choices
=
[
'asr'
,
'cls'
,
'st'
,
'text'
,
'tts'
],
help
=
'Choose speech task.'
,
required
=
True
)
self
.
task_choices
=
[
'asr'
,
'cls'
,
'st'
,
'text'
,
'tts'
]
def
show_support_models
(
self
,
pretrained_models
:
dict
):
fields
=
model_name_format
[
self
.
task
].
split
(
"-"
)
table
=
PrettyTable
(
fields
)
for
key
in
pretrained_models
:
table
.
add_row
(
key
.
split
(
"-"
))
print
(
table
)
def
execute
(
self
,
argv
:
List
[
str
])
->
bool
:
"""
Command line entry.
"""
parser_args
=
self
.
parser
.
parse_args
(
argv
)
self
.
task
=
parser_args
.
task
if
self
.
task
not
in
self
.
task_choices
:
logger
.
error
(
"Please input correct speech task, choices = ['asr', 'cls', 'st', 'text', 'tts']"
)
return
False
if
self
.
task
==
'asr'
:
try
:
from
..asr.infer
import
pretrained_models
logger
.
info
(
"Here is the list of ASR pretrained models released by PaddleSpeech that can be used by command line and python API"
)
self
.
show_support_models
(
pretrained_models
)
# TODO show pretrained static model
return
True
except
BaseException
:
logger
.
error
(
"Failed to get the list of ASR pretrained models."
)
return
False
elif
self
.
task
==
'cls'
:
try
:
from
..cls.infer
import
pretrained_models
logger
.
info
(
"Here is the list of CLS pretrained models released by PaddleSpeech that can be used by command line and python API"
)
self
.
show_support_models
(
pretrained_models
)
return
True
except
BaseException
:
logger
.
error
(
"Failed to get the list of CLS pretrained models."
)
return
False
elif
self
.
task
==
'st'
:
try
:
from
..st.infer
import
pretrained_models
logger
.
info
(
"Here is the list of ST pretrained models released by PaddleSpeech that can be used by command line and python API"
)
self
.
show_support_models
(
pretrained_models
)
return
True
except
BaseException
:
logger
.
error
(
"Failed to get the list of ST pretrained models."
)
return
False
elif
self
.
task
==
'text'
:
try
:
from
..text.infer
import
pretrained_models
logger
.
info
(
"Here is the list of TEXT pretrained models released by PaddleSpeech that can be used by command line and python API"
)
self
.
show_support_models
(
pretrained_models
)
return
True
except
BaseException
:
logger
.
error
(
"Failed to get the list of TEXT pretrained models."
)
return
False
elif
self
.
task
==
'tts'
:
try
:
from
..tts.infer
import
pretrained_models
logger
.
info
(
"Here is the list of TTS pretrained models released by PaddleSpeech that can be used by command line and python API"
)
self
.
show_support_models
(
pretrained_models
)
# TODO show pretrained static model
return
True
except
BaseException
:
logger
.
error
(
"Failed to get the list of TTS pretrained models."
)
return
False
@
stats_wrapper
def
__call__
(
self
,
task
:
str
=
None
,
):
"""
Python API to call an executor.
"""
if
task
not
in
[
'asr'
,
'cls'
,
'st'
,
'text'
,
'tts'
]:
print
(
"Please input correct speech task, choices = ['asr', 'cls', 'st', 'text', 'tts']"
)
res
=
""
return
res
setup.py
浏览文件 @
f8375764
...
...
@@ -66,6 +66,7 @@ requirements = {
# fastapi server
"fastapi"
,
"uvicorn"
,
"prettytable"
],
"develop"
:
[
"ConfigArgParse"
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录