n/futokb
1
0
Fork 0
mirror of https://gitlab.futo.org/keyboard/latinime.git synced 2024-09-28 14:54:30 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update melSpectrogram

Browse Source
This commit is contained in:
Aleksandras Kostarevas 2023-09-27 23:05:27 +03:00
parent f1ff995cc7
commit a88bd5c820
3 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
java/src/org/futo/inputmethod/latin/uix/voiceinput/downloader/DownloadActivity.kt
View File

@ -256,6 +256,8 @@ class DownloadActivity : ComponentActivity() {
private val themeOption: MutableState<ThemeOption?> = mutableStateOf(null)
private fun updateContent() {
// TODO: In some cases seems to cause a crash?
// May be related https://github.com/mozilla-mobile/focus-android/issues/7712
setContent {
themeOption.value?.let { themeOption ->
val themeIdx = useDataStore(key = THEME_KEY.key, default = themeOption.key)

2
voiceinput-shared/src/main/java/org/futo/voiceinput/shared/util/AudioFeatureExtraction.kt
View File

@ -209,7 +209,7 @@ class AudioFeatureExtraction(
* This function converts input audio samples to 1x80x3000 features
*/
fun melSpectrogram(y: DoubleArray): FloatArray {
val paddedWaveform = DoubleArray(min(numSamples, y.size + hopLength)) {
val paddedWaveform = DoubleArray(min(numSamples, y.size + hopLength).coerceAtLeast(nFFT)) {
if (it < y.size) {
y[it]
} else {

8
voiceinput-shared/src/main/java/org/futo/voiceinput/shared/whisper/MelProcessor.kt
View File

@ -12,11 +12,5 @@ private val extractor = AudioFeatureExtraction(
)
fun extractMelSpectrogramForWhisper(samples: DoubleArray): FloatArray {
val paddedSamples = if(samples.size <= 640) {
samples + DoubleArray(640) { 0.0 }
} else {
samples
}
return extractor.melSpectrogram(paddedSamples)
return extractor.melSpectrogram(samples)
}