守望者--AIR技术交流

标题: 声纹识别引擎 govpr [打印本页]
作者: 破晓    时间: 2016-11-9 13:42
标题: 声纹识别引擎 govpr

简介

govpr是golang 实现的基于 GMM-UBM 说话人识别引擎(声纹识别),可用于语音验证,身份识别的场景. 目前暂时仅支持汉语数字的语音,语音格式为wav格式(比特率16000,16bits,单声道)

安装

?
1
go get github.com/liuxp0827/govpr

示例

如下是一个简单的示例. 可跳转至 example 查看详细的例子,示例中的语音为纯数字8位数字.语音验证后得到一个得分,可设置阈值来判断验证语音是否为注册训练者本人.

?
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
82
83
84
85
86
87
88
89
package main
import (
    "github.com/liuxp0827/govpr"
    "github.com/liuxp0827/govpr/log"
    "github.com/liuxp0827/govpr/waveIO"
    "io/ioutil"
)
type engine struct {
    vprEngine *govpr.VPREngine
}
func NewEngine(sampleRate, delSilRange int, ubmFile, userModelFile string) *engine {
    return &engine{
        vprEngine: govpr.NewVPREngine(sampleRate, delSilRange, ubmFile, userModelFile),
    }
}
func (this *engine) DestroyEngine() {
    this.vprEngine = nil
}
func (this *engine) TrainSpeech(buffers [][]byte) error {
    var err error
    count := len(buffers)
    for i := 0; i < count; i++ {
        err = this.vprEngine.AddTrainBuffer(buffers)
        if err != nil {
            log.Error(err)
            return err
        }
    }
    defer this.vprEngine.ClearTrainBuffer()
    defer this.vprEngine.ClearAllBuffer()
    err = this.vprEngine.TrainModel()
    if err != nil {
        log.Error(err)
        return err
    }
    return nil
}
func (this *engine) RecSpeech(buffer []byte) error {
    err := this.vprEngine.AddVerifyBuffer(buffer)
    defer this.vprEngine.ClearVerifyBuffer()
    if err != nil {
        log.Error(err)
        return err
    }
    err = this.vprEngine.VerifyModel()
    if err != nil {
        log.Error(err)
        return err
    }
    Score := this.vprEngine.GetScore()
    log.Infof("vpr score: %f", Score)
    return nil
}
func main() {
    log.SetLevel(log.LevelDebug)
    vprEngine := NewEngine(16000, 50, "../ubm/ubm", "model/test.dat")
    trainlist := []string{
        "wav/train/01_32468975.wav",
        "wav/train/02_58769423.wav",
        "wav/train/03_59682734.wav",
        "wav/train/04_64958273.wav",
        "wav/train/05_65432978.wav",
    }
    trainBuffer := make([][]byte, 0)
    for _, file := range trainlist {
        buf, err := loadWaveData(file)
        if err != nil {
            log.Error(err)
            return
        }
        trainBuffer = append(trainBuffer, buf)
    }
    verifyBuffer, err := waveIO.WaveLoad("wav/verify/34986527.wav")
    if err != nil {
        log.Error(err)
        return
    }
    vprEngine.TrainSpeech(trainBuffer)
    vprEngine.RecSpeech(verifyBuffer)
}
func loadWaveData(file string) ([]byte, error) {
    data, err := ioutil.ReadFile(file)
    if err != nil {
        return nil, err
    }
    // remove .wav header info 44 bits
    data = data[44:]
    return data, nil
}

https://www.oschina.net/img/github_logo.gif") left center no-repeat;">liuxp0827 / govpr

Watch3 Star47 Fork7

Voiceprint Recognition by golang — More...

master分支代码最近更新：2016-11-07



来源：https://www.oschina.net/p/govpr?from=20161106

下载zip





欢迎光临 守望者--AIR技术交流 (http://www.airmyth.com/)