Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
DeepSpeech
提交
83465d2a
D
DeepSpeech
项目概览
PaddlePaddle
/
DeepSpeech
12 个月 前同步成功
通知
205
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,发现更多精彩内容 >>
未验证
提交
83465d2a
编写于
2月 28, 2022
作者:
H
Hui Zhang
提交者:
GitHub
2月 28, 2022
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #1497 from lym0302/paddlespeech_stats
[CLI] add paddlespeech stats
上级
29a41a40
96abb33b
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
210 addition
and
0 deletion
+210
-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
+194
-0
setup.py
setup.py
+1
-0
未找到文件。
paddlespeech/cli/__init__.py
浏览文件 @
83465d2a
...
...
@@ -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
浏览文件 @
83465d2a
# 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
浏览文件 @
83465d2a
# 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
=
'Get speech tasks support models list.'
)
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
)
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
)
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.
"""
self
.
task
=
task
if
self
.
task
not
in
self
.
task_choices
:
print
(
"Please input correct speech task, choices = ['asr', 'cls', 'st', 'text', 'tts']"
)
elif
self
.
task
==
'asr'
:
try
:
from
..asr.infer
import
pretrained_models
print
(
"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
)
except
BaseException
:
print
(
"Failed to get the list of ASR pretrained models."
)
elif
self
.
task
==
'cls'
:
try
:
from
..cls.infer
import
pretrained_models
print
(
"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
)
except
BaseException
:
print
(
"Failed to get the list of CLS pretrained models."
)
elif
self
.
task
==
'st'
:
try
:
from
..st.infer
import
pretrained_models
print
(
"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
)
except
BaseException
:
print
(
"Failed to get the list of ST pretrained models."
)
elif
self
.
task
==
'text'
:
try
:
from
..text.infer
import
pretrained_models
print
(
"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
)
except
BaseException
:
print
(
"Failed to get the list of TEXT pretrained models."
)
elif
self
.
task
==
'tts'
:
try
:
from
..tts.infer
import
pretrained_models
print
(
"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
)
except
BaseException
:
print
(
"Failed to get the list of TTS pretrained models."
)
\ No newline at end of file
setup.py
浏览文件 @
83465d2a
...
...
@@ -62,6 +62,7 @@ base = [
"visualdl"
,
"webrtcvad"
,
"yacs~=0.1.8"
,
"prettytable"
,
]
server
=
[
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录