【汉化工具系列 #2】指定wave格式转换为srt格式字幕（CPU版本）

shane007

本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件

代码如下
以下这句用多线程可以增速，否则很慢
1 r, ^( c' @( y" l( p$ _以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。

3 s1 A$ A5 i0 @/ N* _# |0 f& T

1. var builder = factory.CreateBuilder()
   * w/ q9 r! ]3 v% E; @0 x& r/ ~" {) l
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

5 H( f, h' q5 C2 n. A* `" l0 ^
$ s) g/ P, M( j5 ^( F6 @( n. X

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   
3. using System;
   
4. using System.Diagnostics;
   " B, R8 V9 j! t
5. using System.IO;
   - F5 L' h- _0 u7 I1 q% z% y& q; l
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using CommandLine;
   
9. using Whisper.net;
   9 [  r, a5 G. b7 i2 |
10. using Whisper.net.Ggml;
    
11. using Whisper.net.Wave;
    / W# [7 Z. f! W2 l7 j7 ]4 u
12. 
    , e) M2 y) [# g' S1 w$ ?) B
13. await Parser.Default.ParseArguments<Options>(args)
    ' N1 Z+ `+ A" W
14.     .WithParsedAsync(Demo);
    - T; E+ o; K( y$ H; e, _8 q/ e  E
15. 
    
16. async Task Demo(Options opt)
    " q" z$ X- V0 A( X4 {: o( g/ ?
17. 
    
18. {
    
19.     if (!File.Exists(opt.ModelName))
    / G  ^" Y/ v. z0 T
20.     {
    
21.         Console.WriteLine([        DISCUZ_CODE_1        ]quot;Downloading Model {opt.ModelName}");
    
22.         using var modelStream = await WhisperGgmlDownloader.GetGgmlModelAsync(opt.ModelType);
    
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    : D' {9 [6 Y6 Q' s: P( D, U4 I' N
24.         await modelStream.CopyToAsync(fileWriter);
    % j# h3 |3 ^7 r8 S! T; h+ L  c
25.     }
    
26. 
    - B# |  x8 C# G5 F; \: T
27.     switch (opt.Command)
    ' X) U9 w  ^0 h5 F, O7 w/ q
28.     {
    
29.         case "lang-detect":
    
30.             LanguageIdentification(opt);
    
31.             break;
    
32.         case "transcribe":
      I( q8 n  n+ ~* i
33.         case "translate":
    ) n9 @5 n2 v3 u
34.             await FullDetection(opt);
    , |3 P2 c+ L5 J/ t2 [- |: C* Q6 t
35.             break;
    : R$ Z) x0 |1 G4 L- N
36.         default:
    6 z2 |2 N- }, O0 @$ \
37.             Console.WriteLine("Unknown command");
    . X* O/ j3 a' x; k9 w
38.             break;
    3 K2 M0 k" O, Q: E/ W
39.     }
    + M" m. ]) A, O1 y% U  g0 E
40. }
    ; ]/ F  D! {+ r; f6 y& U, B" |+ S8 R
41. 
    0 O6 m% h% q+ N, P* a9 |- Q
42. void LanguageIdentification(Options opt)
    
43. {
    * O: K- ]( O/ i0 @0 {, ^
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    % x  }/ |+ w$ `' \7 {- p( l
45. 
    0 c; v! y' @0 M/ J1 E0 a0 P
46.     // Same factory can be used by multiple task to create processors.
    
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    
48. 
    
49. /*    var builder = factory.CreateBuilder()
    ) ^0 {3 h$ D) E: B4 n2 ~
50.        .WithLanguage(opt.Language);*/
    
51.     var builder = factory.CreateBuilder()
    ; k3 U" n' z# e. _( O
52.    .WithLanguage("english");
    / D6 v- b" y# L4 f) D5 `; t- j" n
53.     using var processor = builder.Build();
    
54. 
    6 }8 d, V- L/ A. _. F
55.     using var fileStream = File.OpenRead(opt.FileName);
    
56. 
    
57.     var wave = new WaveParser(fileStream);
    : q/ @0 Z9 w1 J( [% c0 `) ^
58. 
    # L2 a; L. _* d4 r
59.     var samples = wave.GetAvgSamples();
    
60. 
    ( \; [4 V3 a, h9 N; e. q# W3 i& j
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    
62.     Console.WriteLine("Language is " + language);
    
63. }
    
64. 
    ; t, m" [; o' d* E' e5 N
65. async Task FullDetection(Options opt)
    6 W4 g+ f# o1 _6 d' w: N2 [5 A2 }
66. {
    " V* y4 s) }- i& s7 C) f/ E
67.     // Same factory can be used by multiple task to create processors.
    
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    
69. 
    : n) s  t% D4 P/ R6 \& r5 b
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    
72.     string languageOption = Console.ReadLine();
    2 R3 b" I* u! R1 j5 {: Y
73.     var builder = factory.CreateBuilder()
    
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    1 v7 @$ F8 c. h1 A# `
75. 
    
76.     if (opt.Command == "translate")
    
77.     {
    + S; S; e. U' s# O8 r; s
78.         builder.WithTranslate();
    
79.     }
    5 g$ ^0 }0 X- F+ v# p' e8 Q3 @9 s
80. 
    " i  B( q) V2 {, Q' `8 o1 H) H
81.     WhisperProcessor processor = builder.Build();
    ( s; S2 M, x9 y- `- v& }
82.    
    & @- B2 L! S: T; N
83.     Console.WriteLine("请输入wave源文件目录：");
    4 ^" }% c, `, p( g
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    
87.     string targetDirectory = Console.ReadLine();
    
88. 
    
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    & b0 x/ f- [( ~. \2 k
90.     {
    + S7 w8 w  l8 d+ |+ b
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    , p7 Z0 H6 d& a* b0 Q  z
92.         return;
    
93.     }
    
94. 
    
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    " u* T2 k( N2 j! b' O9 V, a4 {: v
96. 
    1 a8 B$ z- u8 f6 P& t0 Z
97.     Console.WriteLine("处理完成！");
    
98. 
    
99. }
    9 E* V2 i+ W: G  C8 U) a
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     
101.     string sourceDirectory, string targetDirectory)
     
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     # t. _* y7 O7 R1 R4 ~2 h& ?
104. 
     3 B' l2 T& K8 O6 n* o
105.     foreach (var sourceFilePath in files)
     & f" I2 S) S% `
106.     {
     
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     . b' o3 C4 l& D7 \! R! |  x
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     5 h. @9 N- m6 v6 S
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     2 ?" U( a$ N1 i$ U0 a
112. 
     
113. 
     
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     
117. 
     # \* y: @/ d/ O7 s7 e* x& n; k
118.             using var fileStream = File.OpenRead(sourceFilePath);
     % W! m# r5 m2 ~9 u$ K
119.             var segmentIndex = 1;
     : |* ]1 n! N* p1 S9 g
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     " t! T  Q5 s/ b- Q0 {
121.             var startTime = DateTime.Now; // 记录开始时间
     
122. 
     ! S/ L& q: H6 r" n
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     
124.             {
     ) H1 U- R6 @: ~1 s) f* c
125.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     3 `, {( H% G" {3 @5 h, b
126.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     7 W  X0 O* d" a
127.                 Console.WriteLine(segment.Text);
     % J( ]9 W1 Q' T, x
128.                 Console.WriteLine();
     
129. 
     ) P: H. k$ n8 {9 ]) P2 P6 S1 b8 D
130.                 // 将srt内容写入文件
     
131.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     
132.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     
133.                 await writer.WriteLineAsync(segment.Text);
     $ g- E5 U, d. x3 p
134.                 await writer.WriteLineAsync();
     
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     6 y2 T9 T) z: Q+ G; b
137. 
     ! Z, y$ y1 T) ~- v6 z, Y1 y5 J
138.                 segmentIndex++;
     
139.             }
     
140. 
     9 e5 r; l, |) R$ Y' d  e7 v
141.             var endTime = DateTime.Now; // 记录结束时间
     % x4 N8 n% J1 o
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     9 l$ Y$ V2 w4 E: U2 {' |' g
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     ; w8 G; p; |. n, N
145.         }
     " B8 ^3 B$ p- }- F% y
146.         else {
     ' v! j% n2 I) U/ B# E# L# T/ p# g
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     3 x4 p/ n. w) N+ e
148.         }
     
149.     }
     $ `/ A1 Q6 l$ E+ Q2 x" `
150. }
     ! |/ T5 P  G" g$ N  Z
151. public class Options
     
152. {
     
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     6 F/ B# d2 w# ^  x; h' m- b
154.     public string Command { get; set; }
     
155. 
     + \4 D/ X' m" S6 \# k+ a/ `+ T
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     8 @; B. K  {( Q' q' H
157.     public string FileName { get; set; }
     , `# f; f) ]; Y3 O
158. 
     3 u* E) w3 E' _
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     7 I4 A& U4 k% T
160.     public string Language { get; set; }
     
161. 
     ; q& q6 V% M, X7 z
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     4 z+ o8 v: ?- X' D
163.     public string ModelName { get; set; }
     
164. 
     5 x# e2 f, Z: |( f# V) }- u5 |
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
     5 ^" e$ T, C& m2 Z% u
166.     public GgmlType ModelType { get; set; }
     
167. }
     

复制代码

! W1 G- u+ X( Z# I4 \$ o. k
* y/ k9 Y" o8 C$ m6 T  }

7 S. T' A7 O( a+ V

星之韶华

学习学习一下