wekws icon indicating copy to clipboard operation
wekws copied to clipboard

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

Open taorui-plus opened this issue 1 year ago • 2 comments

背景:想在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 avatar May 07 '24 08:05 taorui-plus

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

mlxu995 avatar May 07 '24 14:05 mlxu995

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

lin-xiaosheng avatar May 28 '24 02:05 lin-xiaosheng