15 Commits

Author SHA1 Message Date
Jean Du
b233d46552
[ctc] KWS with CTCloss training and CTC prefix beam search detection. (#135)
* add ctcloss training scripts.

* update compute_det_ctc

* fix typo.

* add fsmn model, can use pretrained kws model from modelscope.

* Add streaming detection of CTC model. Add CTC model onnx export. Add CTC model's result in README; For now CTC model runtime is not supported yet.

* QA run.sh, maxpooling training scripts is compatible. Ready to PR.

* Add a streaming kws demo, support fsmn online forward

* fix typo.

* Align Stream FSMN and Non-Stream FSMN, both in feature extraction and model forward.

* fix repeat activation, add a interval restrict.

* fix timestamp when subsampling!=1.

* fix flake8, update training script and README, give pretrained ckpt.

* fix quickcheck and flake8

* Add realtime CTC-KWS demo in README.

---------

Co-authored-by: dujing <dujing@xmov.ai>
2023-08-16 10:07:04 +08:00
Binbin Zhang
9f29e033aa
[examples] remove static quantization (#87) 2022-09-12 15:49:28 +08:00
Binbin Zhang
c9a262866f
[wekws] rename kws to wekws (#76)
* [wekws] rename kws to wekws

* fix lint
2022-08-27 11:57:44 +08:00
ryoha
41a3432198
fix export in export_onnx (#71) 2022-05-29 09:30:53 +08:00
Cyan
7d142b9528
[examples] refactor FAR computation to support long audio test (#64)
* add .gitattributes

* add long wav

* fix some bugs

* updated lint error

* back the hi_xiaowen/run.sh to the same

* remove the space

* better one

* remove 'num_keyword' parameter

* remove files

* flask8 examine

* override the score and compute_det file

* remove defaultdict

* remove import defaultdict
2022-03-24 14:35:07 +08:00
Binbin Zhang
57021924cb
[kws] support onnx export (#53) 2022-01-15 13:50:34 +08:00
Binbin Zhang
f86a797b10
[kws] add static quantize (#44)
* [kws] add static quantize

* refine lint error in shuffle_list.py

* refine lint

* fix topo
2021-12-14 14:32:54 +08:00
Binbin Zhang
171309bd9e
[bin] use torchrun to launch ddp training (#42) 2021-12-13 19:45:55 +08:00
Binbin Zhang
92a4c19ffe
[examples] use ds_tcn as default model (#34)
* [examples] use ds_tcn as default model

* fix scoring gpu id
2021-12-07 10:36:38 +08:00
lxiao336
ba6919baaf
modifications to get the mdtc model torch-scriptable (#14)
* modifying some implmentations of mdtc to get the model torch-scripting through

* modifications to get the mdtc model torch-scriptable

Co-authored-by: lxiao336 <shawl336@163.com>
2021-11-29 11:15:30 +08:00
jingyong hou
edfc6de743 add results of mdtc 2021-11-19 15:31:11 +08:00
Jingyong Hou
674c372c9a formatting 2021-11-11 09:40:05 +08:00
Jingyong Hou
7df9ced666 fixed bug of compute_cmvn_stats.py 2021-11-10 22:40:21 +08:00
Jingyong Hou
4db050eb67 add model mdtc for mobvoi-hotword example 2021-11-10 22:13:46 +08:00
Binbin Zhang
dbebee86fd [examples] support hi xiaowen dataset 2021-11-10 18:57:52 +08:00