Android demo for speaker diarization (#1423)
This commit is contained in:
@@ -34,7 +34,7 @@ data class OfflineSpeakerDiarizationSegment(
|
||||
|
||||
class OfflineSpeakerDiarization(
|
||||
assetManager: AssetManager? = null,
|
||||
config: OfflineSpeakerDiarizationConfig,
|
||||
val config: OfflineSpeakerDiarizationConfig,
|
||||
) {
|
||||
private var ptr: Long
|
||||
|
||||
@@ -64,7 +64,7 @@ class OfflineSpeakerDiarization(
|
||||
fun process(samples: FloatArray) = process(ptr, samples)
|
||||
|
||||
fun processWithCallback(
|
||||
samples: FloatArray,
|
||||
samples: FloatArray,
|
||||
callback: (numProcessedChunks: Int, numTotalChunks: Int, arg: Long) -> Int,
|
||||
arg: Long = 0,
|
||||
) = processWithCallback(ptr, samples, callback, arg)
|
||||
@@ -84,14 +84,17 @@ class OfflineSpeakerDiarization(
|
||||
|
||||
private external fun getSampleRate(ptr: Long): Int
|
||||
|
||||
private external fun process(ptr: Long, samples: FloatArray): Array<OfflineSpeakerDiarizationSegment>
|
||||
private external fun process(
|
||||
ptr: Long,
|
||||
samples: FloatArray
|
||||
): Array<OfflineSpeakerDiarizationSegment>
|
||||
|
||||
private external fun processWithCallback(
|
||||
ptr: Long,
|
||||
samples: FloatArray,
|
||||
callback: (numProcessedChunks: Int, numTotalChunks: Int, arg: Long) -> Int,
|
||||
arg: Long,
|
||||
): Array<OfflineSpeakerDiarizationSegment>
|
||||
): Array<OfflineSpeakerDiarizationSegment>
|
||||
|
||||
companion object {
|
||||
init {
|
||||
|
||||
@@ -3,13 +3,6 @@ package com.k2fsa.sherpa.onnx
|
||||
import android.content.res.AssetManager
|
||||
import android.util.Log
|
||||
|
||||
data class SpeakerEmbeddingExtractorConfig(
|
||||
val model: String,
|
||||
var numThreads: Int = 1,
|
||||
var debug: Boolean = false,
|
||||
var provider: String = "cpu",
|
||||
)
|
||||
|
||||
class SpeakerEmbeddingExtractor(
|
||||
assetManager: AssetManager? = null,
|
||||
config: SpeakerEmbeddingExtractorConfig,
|
||||
|
||||
@@ -0,0 +1,8 @@
|
||||
package com.k2fsa.sherpa.onnx
|
||||
|
||||
data class SpeakerEmbeddingExtractorConfig(
|
||||
val model: String,
|
||||
var numThreads: Int = 1,
|
||||
var debug: Boolean = false,
|
||||
var provider: String = "cpu",
|
||||
)
|
||||
Reference in New Issue
Block a user