Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

examples/speechcommand_v1/s0训练的模型在android系统上怎么运行? #163

Open
taorui-plus opened this issue May 7, 2024 · 2 comments

Comments

@taorui-plus
Copy link

taorui-plus commented May 7, 2024

背景:想在android系统上跑一个识别大约50个关键词的任务,想在这个项目上做调试,对android系统开发不太熟悉,在摸索阶段。

成功的例子:在runtime/android下面使用默认的ort模型文件(hey_snips)可以正常运行,可以准确检测到关键词,但是只有一个关键词,不满足需求。

目前的问题
尝试用examples/speechcommand_v1/s0和examples/hi_xiaowen/s0下面训练导出的ort模型文件替换之前的hey_snips模型报错:

LoadOrtModel The ORT format model version [6] is not supported this build 1.12.1
模型在android系统上没有加载成功。

在linux环境下检查了onnxruntime 的版本是1.17.3 ,降级到1.12.1重新生成ort文件还是这个问题,
查看onnx 没有1.12.1的版本,安装的是1.16.0

onnx                      1.16.0                               py38h12ddb61_0    defaults
onnxruntime        1.17.3-->1.12.1                pypi_0                      pypi

其他尝试
尝试用examples/hey_snips/s0训练模型测试,但是hey_snips_kws_4.0.tar.gz数据集下载遇到了问题,填写了表格,邮箱收到了下载链接,但是点击后无法下载。
所以不知道这个例子怎么才能跑起来。

@taorui-plus taorui-plus changed the title examples/speechcommand_v1/s0训练的模型在runtime/android上能直接替换模型运行吗? examples/speechcommand_v1/s0训练的模型在android系统上怎么运行? May 7, 2024
@mlxu995
Copy link
Collaborator

mlxu995 commented May 7, 2024

hey_snips_kws_4.0.tar.gz 的下载可以挂个代理试试。
或者也可以用 GSC(speechcommand_v1) 数据集按 max-pooling 的方式训练模型导出下在安卓上试试,注意原始 GSC 的训练配置是 softmax loss,需要改成 max-pooling loss 才能才安卓端跑(可以 compare 一下两个 example 下的 conf/mdtc.yaml )

@lin-xiaosheng
Copy link

礼貌问一下是怎么解决"LoadOrtModel The ORT format model version [6] is not supported this build 1.12.1"这个问题的?我也遇到了这个问题,深受困扰

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants