Add Go API for speech enhancement GTCRN models (#1991)
This commit is contained in:
43
go-api-examples/speech-enhancement-gtcrn/main.go
Normal file
43
go-api-examples/speech-enhancement-gtcrn/main.go
Normal file
@@ -0,0 +1,43 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
sherpa "github.com/k2-fsa/sherpa-onnx-go/sherpa_onnx"
|
||||
"log"
|
||||
)
|
||||
|
||||
func main() {
|
||||
log.SetFlags(log.LstdFlags | log.Lmicroseconds)
|
||||
|
||||
config := sherpa.OfflineSpeechDenoiserConfig{}
|
||||
|
||||
// Please download the models from
|
||||
// https://github.com/k2-fsa/sherpa-onnx/releases/tag/speech-enhancement-models
|
||||
|
||||
config.Model.Gtcrn.Model = "./gtcrn_simple.onnx"
|
||||
config.Model.NumThreads = 1
|
||||
config.Model.Debug = 1
|
||||
|
||||
sd := sherpa.NewOfflineSpeechDenoiser(&config)
|
||||
defer sherpa.DeleteOfflineSpeechDenoiser(sd)
|
||||
|
||||
wave_filename := "./inp_16k.wav"
|
||||
|
||||
wave := sherpa.ReadWave(wave_filename)
|
||||
if wave == nil {
|
||||
log.Printf("Failed to read %v\n", wave_filename)
|
||||
return
|
||||
}
|
||||
|
||||
log.Println("Started")
|
||||
audio := sd.Run(wave.Samples, wave.SampleRate)
|
||||
log.Println("Done!")
|
||||
|
||||
filename := "./enhanced-16k.wav"
|
||||
ok := audio.Save(filename)
|
||||
if !ok {
|
||||
log.Fatalf("Failed to write", filename)
|
||||
} else {
|
||||
log.Println("Saved to ", filename)
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user