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

shane007

: Q% H/ D( ]+ X! R0 [3 L
本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
使用时需要輸入语言，源文件路径和目标文件路径
: ~1 U) Y2 W) \' Y0 W' p最后是输出srt文件
4 _1 w3 ^2 }# [
" J. N5 X6 \* b, @  \1 T7 e& H6 c代码如下
以下这句用多线程可以增速，否则很慢
以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。

1. var builder = factory.CreateBuilder()
   $ y+ F" n: p' m% C& C% u7 k0 v
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   
3. using System;
   ) B7 ]2 ]) w& }% o
4. using System.Diagnostics;
   $ Z. ^* D, t! m! X' L$ b
5. using System.IO;
   - k; S' ~, i0 v' U; b2 \
6. using System.Threading;
   : q& I; B9 Y! r" Y; l. l% B* h; q
7. using System.Threading.Tasks;
   $ X$ k# L8 j& x0 R2 J
8. using CommandLine;
   5 o7 E; T" e) e: p! b" S) o% k
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    
11. using Whisper.net.Wave;
    
12. 
    ! j9 |% _( f: }8 e) [3 b3 T
13. await Parser.Default.ParseArguments<Options>(args)
    * z" K. S% o  V% l8 V
14.     .WithParsedAsync(Demo);
    
15. 
    
16. async Task Demo(Options opt)
    
17. 
    
18. {
    0 ]6 B, X* s& Q$ Z6 @
19.     if (!File.Exists(opt.ModelName))
    
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);
    " {% P; {2 X4 t; L& O
24.         await modelStream.CopyToAsync(fileWriter);
    
25.     }
    
26. 
    
27.     switch (opt.Command)
    
28.     {
    0 e& g0 f9 G+ b0 Z4 |& w
29.         case "lang-detect":
    0 F$ `1 a# n( }4 S6 \; o* B9 [
30.             LanguageIdentification(opt);
    
31.             break;
    
32.         case "transcribe":
    / s8 b( K4 a- d$ x
33.         case "translate":
    
34.             await FullDetection(opt);
    5 M5 W+ @. r( }+ e6 q
35.             break;
    
36.         default:
    5 b" o0 S/ m; A( ~: n7 U$ r6 x4 }/ C- s
37.             Console.WriteLine("Unknown command");
    6 [$ s% A+ C2 A! M1 F
38.             break;
    + G; ?( j' A! x( o
39.     }
    
40. }
    
41. 
    ! J8 @1 ?' [/ H' }( a  }4 i
42. void LanguageIdentification(Options opt)
    
43. {
    
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    
45. 
    $ s& R: }4 M! U+ n0 Y7 V/ S$ }
46.     // Same factory can be used by multiple task to create processors.
    5 n* C1 i( B9 |6 \; c* Y
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    
48. 
    : s/ r: n" c$ L4 Y! g6 O9 O
49. /*    var builder = factory.CreateBuilder()
    
50.        .WithLanguage(opt.Language);*/
    
51.     var builder = factory.CreateBuilder()
    
52.    .WithLanguage("english");
    " k* y+ ?# e4 @: e: c5 ]
53.     using var processor = builder.Build();
    
54. 
    7 Q# K" X/ H3 @1 U. u8 B" c( f
55.     using var fileStream = File.OpenRead(opt.FileName);
    ' Q, f7 h8 V* g0 U3 e; d* b6 @
56. 
    
57.     var wave = new WaveParser(fileStream);
    
58. 
    
59.     var samples = wave.GetAvgSamples();
    - [! j/ V; q$ ^5 a; y
60. 
    1 g2 G9 K' W/ |7 q7 i) X& [
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    2 X( }( e# p6 d
62.     Console.WriteLine("Language is " + language);
    5 i9 f, a4 I$ |& L' }; Z9 d
63. }
    
64. 
    ; ?3 T2 e2 q0 @! i  }: l3 g/ F8 u
65. async Task FullDetection(Options opt)
    ! f; w: B+ S% ?5 S3 O. G5 A
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    : k- R8 M  x" G3 I6 J
69. 
    ! j4 e0 x7 i. r, ], C" ]
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    
72.     string languageOption = Console.ReadLine();
    6 b9 K" E) B9 ~7 Z
73.     var builder = factory.CreateBuilder()
    # {& R: {3 L7 Q) @7 Y; R! H
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    ; n: K) m0 a/ b0 f5 ~! y( h
75. 
    
76.     if (opt.Command == "translate")
    
77.     {
    0 {- t9 n6 l1 p2 A; c$ l: S0 c
78.         builder.WithTranslate();
    * J8 o$ J9 s7 Z, S
79.     }
    " [1 W( `6 P% N5 I9 D. [
80. 
    " ]+ b" i4 i, o% x6 ]) {& F
81.     WhisperProcessor processor = builder.Build();
    # A9 u, C  c& F/ ?
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    # F! a2 e' D( c2 b# C% i: r; J0 r
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    
87.     string targetDirectory = Console.ReadLine();
    
88. 
    - |4 ?- w8 Y' r4 i4 i
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    
90.     {
    
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    
92.         return;
    
93.     }
    2 x' X: t+ `* i6 p# O; k# a; P
94. 
    - B9 x2 B/ G3 z0 _6 Z4 W
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    # M1 G/ L' @3 H# E$ f
96. 
    : w& Y+ U7 e& x" @( y
97.     Console.WriteLine("处理完成！");
    
98. 
    - U: {3 f+ Z7 j
99. }
    
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     
101.     string sourceDirectory, string targetDirectory)
     # V) ~9 c3 ~' x5 I) ~
102. {
     : \* c! b, d2 \" A$ F, I5 g
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     
104. 
     
105.     foreach (var sourceFilePath in files)
     1 ?) M. p  j7 |1 F6 }# C2 ^8 E
106.     {
     % k) K9 f) r, l$ x% Z$ k
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     9 Y! W0 U+ i" N: A+ M
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     - d+ q$ r5 V7 `
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     
112. 
     1 q2 o: j6 Y. f' z: l
113. 
     
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     6 }( H4 c3 D; c, j# Q1 G( Y
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     5 q: b2 l' G# _; T
117. 
     
118.             using var fileStream = File.OpenRead(sourceFilePath);
     
119.             var segmentIndex = 1;
     - y0 x# F9 \. b$ {5 h! y- c+ M. W
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     
121.             var startTime = DateTime.Now; // 记录开始时间
     5 c, M- t8 B1 S' O- n" }" W
122. 
     
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     2 }' Y: U6 z4 u/ h0 ?
124.             {
     
125.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     ) U/ a# Q1 f  v) I9 g9 z
126.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     
127.                 Console.WriteLine(segment.Text);
     : ]2 _0 Y1 t; }) o4 F( b8 [
128.                 Console.WriteLine();
     
129. 
     4 y/ c1 d9 a; ?/ _+ y( d8 B" `  y
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}");
     1 {) _3 b9 [. x# D$ [6 [% g: x1 c
133.                 await writer.WriteLineAsync(segment.Text);
     ' I+ J, p  G  ~2 Q8 I" O& \. |
134.                 await writer.WriteLineAsync();
     : q, w/ t% N3 n! I8 O/ ]
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     4 c9 r  J: N# a3 C
137. 
     
138.                 segmentIndex++;
     
139.             }
     
140. 
     
141.             var endTime = DateTime.Now; // 记录结束时间
     1 z7 J( m# K9 q, _# N
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     
145.         }
     
146.         else {
     
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     
148.         }
     ; M: l1 W# |( r+ L/ @
149.     }
     
150. }
     
151. public class Options
     
152. {
     
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     
154.     public string Command { get; set; }
     
155. 
     ( _+ a' O) N7 }+ r5 |
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     
157.     public string FileName { get; set; }
     
158. 
     
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     
160.     public string Language { get; set; }
     4 G) [+ S4 f5 ^5 G; y" d; S% k3 |. _
161. 
     # b; A; s& M$ n1 ]) e8 b
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     # v3 G; g( y) c  L
163.     public string ModelName { get; set; }
     
164. 
     
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
     
166.     public GgmlType ModelType { get; set; }
     
167. }
     

复制代码

" q& ?& c9 x. l4 g0 _6 ~  z) j
2 n& j& m4 P: y
8 x1 H7 s; @& i& j/ {$ P- _% _