main.go
2.6 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
package main
import (
sherpa "github.com/k2-fsa/sherpa-onnx-go/sherpa_onnx"
"log"
)
func main() {
log.SetFlags(log.LstdFlags | log.Lmicroseconds)
config := sherpa.KeywordSpotterConfig{}
// Please download the models from
// https://github.com/k2-fsa/sherpa-onnx/releases/tag/kws-models
config.ModelConfig.Transducer.Encoder = "./sherpa-onnx-kws-zipformer-wenetspeech-3.3M-2024-01-01/encoder-epoch-12-avg-2-chunk-16-left-64.onnx"
config.ModelConfig.Transducer.Decoder = "./sherpa-onnx-kws-zipformer-wenetspeech-3.3M-2024-01-01/decoder-epoch-12-avg-2-chunk-16-left-64.onnx"
config.ModelConfig.Transducer.Joiner = "./sherpa-onnx-kws-zipformer-wenetspeech-3.3M-2024-01-01/joiner-epoch-12-avg-2-chunk-16-left-64.onnx"
config.ModelConfig.Tokens = "./sherpa-onnx-kws-zipformer-wenetspeech-3.3M-2024-01-01/tokens.txt"
config.KeywordsFile = "./sherpa-onnx-kws-zipformer-wenetspeech-3.3M-2024-01-01/test_wavs/test_keywords.txt"
config.ModelConfig.NumThreads = 1
config.ModelConfig.Debug = 1
spotter := sherpa.NewKeywordSpotter(&config)
defer sherpa.DeleteKeywordSpotter(spotter)
wave_filename := "./sherpa-onnx-kws-zipformer-wenetspeech-3.3M-2024-01-01/test_wavs/3.wav"
wave := sherpa.ReadWave(wave_filename)
if wave == nil {
log.Printf("Failed to read %v\n", wave_filename)
return
}
log.Println("----------Use pre-defined keywords----------")
stream := sherpa.NewKeywordStream(spotter)
defer sherpa.DeleteOnlineStream(stream)
stream.AcceptWaveform(wave.SampleRate, wave.Samples)
for spotter.IsReady(stream) {
spotter.Decode(stream)
result := spotter.GetResult(stream)
if result.Keyword != "" {
// You have to reset the stream right after detecting a keyword
spotter.Reset(stream)
log.Printf("Detected %v\n", result.Keyword)
}
}
log.Println("----------Use pre-defined keywords + add a new keyword----------")
stream2 := sherpa.NewKeywordStreamWithKeywords(spotter, "y ǎn y uán @演员")
defer sherpa.DeleteOnlineStream(stream2)
stream2.AcceptWaveform(wave.SampleRate, wave.Samples)
for spotter.IsReady(stream2) {
spotter.Decode(stream2)
result := spotter.GetResult(stream2)
if result.Keyword != "" {
log.Printf("Detected %v\n", result.Keyword)
}
}
log.Println("----------Use pre-defined keywords + add 2 new keywords----------")
stream3 := sherpa.NewKeywordStreamWithKeywords(spotter, "y ǎn y uán @演员/zh ī m íng @知名")
defer sherpa.DeleteOnlineStream(stream3)
stream3.AcceptWaveform(wave.SampleRate, wave.Samples)
for spotter.IsReady(stream3) {
spotter.Decode(stream3)
result := spotter.GetResult(stream3)
if result.Keyword != "" {
log.Printf("Detected %v\n", result.Keyword)
}
}
}