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

shane007

本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
使用时需要輸入语言，源文件路径和目标文件路径
3 k! i  {$ ~& N/ B% u% o最后是输出srt文件

代码如下
以下这句用多线程可以增速，否则很慢
以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。
7 T3 x! v3 t) U: |6 D9 o3 E
2 W* Y4 @0 P/ Y

1. var builder = factory.CreateBuilder()
   ! v% U* C+ y" f* |
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

! j1 z# @+ E: ^: Y, y$ V% x

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   # ^6 `# `. ], F  c7 w, ]
3. using System;
   
4. using System.Diagnostics;
   
5. using System.IO;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using CommandLine;
   
9. using Whisper.net;
   8 j, ~! x% \/ e- r% v7 \
10. using Whisper.net.Ggml;
    1 O* S/ T% J$ G* ^3 K3 l
11. using Whisper.net.Wave;
    6 h* o/ T" ~. a% r/ Y0 x
12. 
    
13. await Parser.Default.ParseArguments<Options>(args)
    : p5 W$ f  s9 x5 \' R* v$ i
14.     .WithParsedAsync(Demo);
    
15. 
    
16. async Task Demo(Options opt)
    % E( q5 b/ ^5 ~% w/ n, _
17. 
    7 L: X; H4 |; A9 e5 V4 b. o
18. {
    
19.     if (!File.Exists(opt.ModelName))
    * D: F( w) }0 y4 V* m; e
20.     {
    7 y7 W3 r2 z; a7 y5 J, Z/ F2 L. X
21.         Console.WriteLine([        DISCUZ_CODE_1        ]quot;Downloading Model {opt.ModelName}");
    
22.         using var modelStream = await WhisperGgmlDownloader.GetGgmlModelAsync(opt.ModelType);
    8 A; U" y1 w2 `+ d" o+ H
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    , H! y) n9 x% q% b- |
24.         await modelStream.CopyToAsync(fileWriter);
    
25.     }
    
26. 
    
27.     switch (opt.Command)
    
28.     {
    
29.         case "lang-detect":
    
30.             LanguageIdentification(opt);
    
31.             break;
    
32.         case "transcribe":
    
33.         case "translate":
    
34.             await FullDetection(opt);
    
35.             break;
    
36.         default:
    
37.             Console.WriteLine("Unknown command");
    ; d+ _' S4 H' }: T7 M! w$ k
38.             break;
    8 w$ G' w2 i/ {& f$ S
39.     }
    
40. }
    & J# V3 X6 Q+ r4 T
41. 
    7 V! T/ |- Q) e8 i' \8 [4 Y! B, V8 Z
42. void LanguageIdentification(Options opt)
    5 [0 b# p: P2 m! E
43. {
    
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    
45. 
    " C2 ?' C, `: Y! H; Z5 P/ k/ }  W
46.     // Same factory can be used by multiple task to create processors.
    9 `' Q$ p$ O& D+ }
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    1 z) M9 K- a! N, A" n& p1 v0 F
48. 
    2 M% n1 u8 }6 A; k& c3 V
49. /*    var builder = factory.CreateBuilder()
    
50.        .WithLanguage(opt.Language);*/
    # k$ d0 L" Z* d  p. [
51.     var builder = factory.CreateBuilder()
    
52.    .WithLanguage("english");
    
53.     using var processor = builder.Build();
    
54. 
    
55.     using var fileStream = File.OpenRead(opt.FileName);
    3 h# r: t0 I5 ?! T6 q1 V
56. 
    ; y4 g3 q( Q/ _& V
57.     var wave = new WaveParser(fileStream);
    8 @- A% s- @9 M0 y* Y
58. 
    $ _* J9 P' H8 [7 }  H9 I
59.     var samples = wave.GetAvgSamples();
    
60. 
    + |& q8 `/ m4 R- S
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    + z- c* |2 \8 A- s$ B9 J# h9 ]$ v
62.     Console.WriteLine("Language is " + language);
    $ E+ [1 w" C/ a* Q
63. }
    1 e) v# a, N# \4 t- S% c1 K
64. 
    
65. async Task FullDetection(Options opt)
    * w' f' A% P+ q; f- |; t
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    : y  ~% w; k0 M; W1 K$ d3 e: p
69. 
    
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    3 C* r; ?( Q4 `$ f, g6 a
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
      }) o+ S" I& T- b
72.     string languageOption = Console.ReadLine();
    
73.     var builder = factory.CreateBuilder()
    
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    + l7 c( U/ ~% b2 C
75. 
    
76.     if (opt.Command == "translate")
    . m; f4 A( W% t/ ]" M! q. T/ f
77.     {
    ' d: ^8 O# O( Z; L) v( }9 e
78.         builder.WithTranslate();
    + L& }$ k3 b4 s# W2 {  D
79.     }
    
80. 
    / _" w! h* `! ^
81.     WhisperProcessor processor = builder.Build();
    
82.    
    / s/ J8 m7 e) `# v1 q4 P
83.     Console.WriteLine("请输入wave源文件目录：");
    9 |8 I  d' `/ \! ]; @9 E
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    5 h3 h6 K6 p" I3 p' e# @) w( @
86.     Console.WriteLine("请输入目标文件目录：");
    
87.     string targetDirectory = Console.ReadLine();
    
88. 
    ; F$ \$ E/ @- @) H# x: W# Y0 ?
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    
90.     {
    ' \: Q1 p( _- r8 G
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    
92.         return;
    ! y0 Q) j% j5 ?
93.     }
    4 R) W5 i. M/ L/ X
94. 
    
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    ! {7 N: U4 i: D- S- z
96. 
    
97.     Console.WriteLine("处理完成！");
    2 `% P3 y- F7 V/ B
98. 
    3 [9 @& u4 ?8 h( c4 R  k7 J
99. }
    
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     8 M8 Y8 S0 f2 h
101.     string sourceDirectory, string targetDirectory)
     4 s7 P" T! K, J5 n( n
102. {
     ' g+ P1 o" N! ~8 b& u/ G
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     ' X0 ~. b: [2 h* m  k8 B
104. 
     8 \, z* `8 C: Q% A
105.     foreach (var sourceFilePath in files)
     # E; p6 v) n1 }. H2 M" ^
106.     {
     
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     & x5 _5 L1 x* E' U- M; G
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     
112. 
     
113. 
     
114.         if (!File.Exists(destinationFilePath))
     ! N& B5 Y9 S5 f8 f  C( {
115.         {
     
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     
117. 
     6 w: i" E- k% D. c5 T
118.             using var fileStream = File.OpenRead(sourceFilePath);
     0 ]! T& R+ o& Z; K# M# |
119.             var segmentIndex = 1;
     
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     ; f& ]# e# T$ z+ _6 D  `; r, o
121.             var startTime = DateTime.Now; // 记录开始时间
     
122. 
     1 b, a: j2 A" l2 ~" \
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     
124.             {
     
125.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     
126.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     6 }2 o9 T' x4 t* k; q1 A3 u
127.                 Console.WriteLine(segment.Text);
     
128.                 Console.WriteLine();
     
129. 
     
130.                 // 将srt内容写入文件
     - j; q  C. S+ t9 ~
131.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     6 U1 y/ [3 A/ O9 k  Q" u
132.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     - F- D7 e  z* q9 h
133.                 await writer.WriteLineAsync(segment.Text);
     1 j: ?" E1 @' V3 q0 R3 @& q
134.                 await writer.WriteLineAsync();
     
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     : u/ s0 N0 z2 P$ H4 y
137. 
     
138.                 segmentIndex++;
     7 T4 p5 W) p6 H+ m4 B
139.             }
     # w/ C3 w6 D" L+ W
140. 
     
141.             var endTime = DateTime.Now; // 记录结束时间
     
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     6 e% J8 C. A9 q  D0 z
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     6 c$ h( @! O) u
145.         }
     
146.         else {
     
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     
148.         }
     
149.     }
     
150. }
     ; l" V$ \3 {% n& O
151. public class Options
     6 Z" p& z& K. X9 v  r  b9 v3 W
152. {
     1 \2 i0 U% L; G! i) R6 W+ L& {5 J. [
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     , z( t, Y: s& P: W; l2 ^, p
154.     public string Command { get; set; }
     
155. 
     
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     - ]! t4 R5 V; m# U) e/ C
157.     public string FileName { get; set; }
     
158. 
     
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     
160.     public string Language { get; set; }
     " `5 S! h8 m  `: h, O2 C: \% l
161. 
     : {  Y7 v# E& Y' g: W" k0 ^
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
       q# `4 a& f* }& v6 }; I
163.     public string ModelName { get; set; }
     
164. 
     " f" ^, V- c* m# E' @
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
     5 Y+ A4 }& s3 _" `9 H3 f# z- j
166.     public GgmlType ModelType { get; set; }
     
167. }
     

复制代码

; \7 t* @& m( P/ v  ?/ ?: c* e" o
0 L9 M6 n2 S2 r  y: p# N
2 _0 i0 ?' e4 b! t
- z2 e9 J/ C' e. c

星之韶华

学习学习一下