Mirrors/SubtitleEdit
1
0
Fork 0
mirror of https://github.com/SubtitleEdit/subtitleedit.git synced 2024-11-21 18:52:36 +01:00
Code Issues Projects Releases Wiki Activity

Merge pull request #8900 from ivandrofly/feature/voskdistate

Browse Source 
Feature/voskdistate
This commit is contained in:
Nikolaj Olsson 2024-10-12 18:48:01 +02:00 committed by GitHub
commit ecaa9972f3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 32 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
src/ui/Forms/AudioToText/VoskDictate.cs
View File

@ -69,44 +69,46 @@ namespace Nikse.SubtitleEdit.Forms.AudioToText
_model = new Model(modelFileName); _model = new Model(modelFileName);
} }
var rec = new VoskRecognizer(_model, 16000.0f); using (var rec = new VoskRecognizer(_model, 16000.0f))
rec.SetMaxAlternatives(0);
rec.SetWords(true);
var list = new List<ResultText>();
var buffer = new byte[4096];
using (var source = File.OpenRead(WaveFileName))
{ {
int bytesRead; rec.SetMaxAlternatives(0);
while ((bytesRead = source.Read(buffer, 0, buffer.Length)) > 0) rec.SetWords(true);
var list = new List<ResultText>();
var buffer = new byte[4096];
using (var source = File.OpenRead(WaveFileName))
{ {
if (rec.AcceptWaveform(buffer, bytesRead)) int bytesRead;
while ((bytesRead = source.Read(buffer, 0, buffer.Length)) > 0)
{ {
var res = rec.Result(); if (rec.AcceptWaveform(buffer, bytesRead))
var results = VoskAudioToText.ParseJsonToResult(res); {
list.AddRange(results); var res = rec.Result();
} var results = VoskAudioToText.ParseJsonToResult(res);
else list.AddRange(results);
{ }
var res = rec.PartialResult(); else
{
_ = rec.PartialResult();
}
} }
} }
}
var finalResult = rec.FinalResult(); var finalResult = rec.FinalResult();
var finalResults = VoskAudioToText.ParseJsonToResult(finalResult); var finalResults = VoskAudioToText.ParseJsonToResult(finalResult);
list.AddRange(finalResults); list.AddRange(finalResults);
try try
{ {
File.Delete(WaveFileName); File.Delete(WaveFileName);
WaveFileName = null; WaveFileName = null;
} }
catch catch
{ {
// ignore // ignore
} }
return ResultListToText(list); return ResultListToText(list);
}
} }
private static string ResultListToText(List<ResultText> list) private static string ResultListToText(List<ResultText> list)