wenet-e2e/wekws

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

taorui-plus opened this issue · 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数据集下载遇到了问题,填写了表格,邮箱收到了下载链接,但是点击后无法下载。
所以不知道这个例子怎么才能跑起来。

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

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