44 lines
937 B
Go
44 lines
937 B
Go
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)
|
|
}
|
|
|
|
}
|