Program.cs 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. using AliFsmnVadSharp;
  2. using AliFsmnVadSharp.Model;
  3. using NAudio.Wave;
  4. internal static class Program
  5. {
  6. [STAThread]
  7. private static void Main()
  8. {
  9. string applicationBase = AppDomain.CurrentDomain.BaseDirectory;
  10. string modelFilePath = applicationBase + "./speech_fsmn_vad_zh-cn-16k-common-pytorch/model.onnx";
  11. string configFilePath = applicationBase + "./speech_fsmn_vad_zh-cn-16k-common-pytorch/vad.yaml";
  12. string mvnFilePath = applicationBase + "./speech_fsmn_vad_zh-cn-16k-common-pytorch/vad.mvn";
  13. int batchSize = 2;
  14. TimeSpan start_time0 = new TimeSpan(DateTime.Now.Ticks);
  15. AliFsmnVad aliFsmnVad = new AliFsmnVad(modelFilePath, configFilePath, mvnFilePath, batchSize);
  16. TimeSpan end_time0 = new TimeSpan(DateTime.Now.Ticks);
  17. double elapsed_milliseconds0 = end_time0.TotalMilliseconds - start_time0.TotalMilliseconds;
  18. Console.WriteLine("load model and init config elapsed_milliseconds:{0}", elapsed_milliseconds0.ToString());
  19. List<float[]> samples = new List<float[]>();
  20. TimeSpan total_duration = new TimeSpan(0L);
  21. for (int i = 0; i < 2; i++)
  22. {
  23. string wavFilePath = string.Format(applicationBase + "./speech_fsmn_vad_zh-cn-16k-common-pytorch/example/{0}.wav", i.ToString());//vad_example
  24. if (!File.Exists(wavFilePath))
  25. {
  26. continue;
  27. }
  28. AudioFileReader _audioFileReader = new AudioFileReader(wavFilePath);
  29. byte[] datas = new byte[_audioFileReader.Length];
  30. _audioFileReader.Read(datas, 0, datas.Length);
  31. TimeSpan duration = _audioFileReader.TotalTime;
  32. float[] wavdata = new float[datas.Length / 4];
  33. Buffer.BlockCopy(datas, 0, wavdata, 0, datas.Length);
  34. float[] sample = wavdata.Select((float x) => x * 32768f).ToArray();
  35. samples.Add(wavdata);
  36. total_duration += duration;
  37. }
  38. TimeSpan start_time = new TimeSpan(DateTime.Now.Ticks);
  39. //SegmentEntity[] segments_duration = aliFsmnVad.GetSegments(samples);
  40. SegmentEntity[] segments_duration = aliFsmnVad.GetSegmentsByStep(samples);
  41. TimeSpan end_time = new TimeSpan(DateTime.Now.Ticks);
  42. Console.WriteLine("vad infer result:");
  43. foreach (SegmentEntity segment in segments_duration)
  44. {
  45. Console.Write("[");
  46. foreach (var x in segment.Segment)
  47. {
  48. Console.Write("[" + string.Join(",", x.ToArray()) + "]");
  49. }
  50. Console.Write("]\r\n");
  51. }
  52. double elapsed_milliseconds = end_time.TotalMilliseconds - start_time.TotalMilliseconds;
  53. double rtf = elapsed_milliseconds / total_duration.TotalMilliseconds;
  54. Console.WriteLine("elapsed_milliseconds:{0}", elapsed_milliseconds.ToString());
  55. Console.WriteLine("total_duration:{0}", total_duration.TotalMilliseconds.ToString());
  56. Console.WriteLine("rtf:{1}", "0".ToString(), rtf.ToString());
  57. Console.WriteLine("------------------------");
  58. }
  59. }