42 lines
1001 B
Kotlin
42 lines
1001 B
Kotlin
package com.k2fsa.sherpa.onnx
|
|
// Please download test files in this script from
|
|
// https://github.com/k2-fsa/sherpa-onnx/releases/tag/speech-enhancement-models
|
|
|
|
fun main() {
|
|
test()
|
|
}
|
|
|
|
fun test() {
|
|
val denoiser = createOfflineSpeechDenoiser()
|
|
|
|
val waveFilename = "./inp_16k.wav";
|
|
|
|
val objArray = WaveReader.readWaveFromFile(
|
|
filename = waveFilename,
|
|
)
|
|
val samples: FloatArray = objArray[0] as FloatArray
|
|
val sampleRate: Int = objArray[1] as Int
|
|
|
|
val denoised = denoiser.run(samples, sampleRate);
|
|
denoised.save(filename="./enhanced-16k.wav")
|
|
println("saved to ./enhanced-16k.wav")
|
|
}
|
|
|
|
fun createOfflineSpeechDenoiser(): OfflineSpeechDenoiser {
|
|
val config = OfflineSpeechDenoiserConfig(
|
|
model = OfflineSpeechDenoiserModelConfig(
|
|
gtcrn = OfflineSpeechDenoiserGtcrnModelConfig(
|
|
model = "./gtcrn_simple.onnx"
|
|
),
|
|
provider = "cpu",
|
|
numThreads = 1,
|
|
),
|
|
)
|
|
|
|
println(config)
|
|
|
|
return OfflineSpeechDenoiser(config = config)
|
|
}
|
|
|
|
|