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

shane007

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

代码如下
" L+ S% ?# V% w5 O- a& R/ {! `' {以下这句用多线程可以增速，否则很慢
2 S; ]( M  x* e' G' a5 G以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。
/ I, R9 S( f9 x& o( T4 m
: I" e' }, H$ a# Q( D" k
/ G( Q: ], {! H7 W. M

1. var builder = factory.CreateBuilder()
   ' Y& ^* b# h8 G( i' u- {( `0 E
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   ) ~1 S7 F7 p5 ~! t( x
3. using System;
   / O" N+ b7 U3 g& ?, b
4. using System.Diagnostics;
   ' Q: d# a: e0 }0 ?( ?$ a6 c6 n
5. using System.IO;
   ' \& ]) a. l5 R7 f
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using CommandLine;
   2 Z. W& G  P5 T, M
9. using Whisper.net;
   & {+ n; y# Q  n5 I9 Y9 x3 c
10. using Whisper.net.Ggml;
    3 K% X; v# l5 d0 l9 F
11. using Whisper.net.Wave;
    
12. 
    ( @2 @% f) e+ Q/ B; q: I& }+ l6 L1 l
13. await Parser.Default.ParseArguments<Options>(args)
    ( F, A- ?9 o. r! L8 z( F  [
14.     .WithParsedAsync(Demo);
    
15. 
    
16. async Task Demo(Options opt)
    0 n: t( ~4 c0 o. ~. ?0 I
17. 
    
18. {
      `/ x, Z3 |7 G
19.     if (!File.Exists(opt.ModelName))
    
20.     {
    4 X3 v% G# C0 U) E) o. q' A
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);
    % l8 f3 J1 j) _8 K) B$ ^% k
24.         await modelStream.CopyToAsync(fileWriter);
    
25.     }
    
26. 
    
27.     switch (opt.Command)
    
28.     {
    
29.         case "lang-detect":
    
30.             LanguageIdentification(opt);
    ! i8 ^" S- G' t7 |
31.             break;
    * ~! m( N& L6 y' Q
32.         case "transcribe":
    
33.         case "translate":
    : d0 O5 N) p" ]2 H- z; P
34.             await FullDetection(opt);
    % L% L3 D2 E- q) |% X
35.             break;
    
36.         default:
    5 ^+ R5 ?, l6 V3 Y8 D! e, M9 K
37.             Console.WriteLine("Unknown command");
    , K- V8 c9 ^( N$ U
38.             break;
    9 o& q( {; C6 I$ l+ [: V
39.     }
    * T; t$ B* [0 Z% c3 j
40. }
    ' H' R- G7 f4 Z
41. 
    
42. void LanguageIdentification(Options opt)
    
43. {
    
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    
45. 
      t$ i! w. r# ~9 P9 ~
46.     // Same factory can be used by multiple task to create processors.
    
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    0 u6 b$ T+ A7 }5 R3 _
48. 
    3 Q) t% N9 y/ X" @) F1 f: S
49. /*    var builder = factory.CreateBuilder()
    " b0 h- g$ A% H
50.        .WithLanguage(opt.Language);*/
    . U+ x  L# x6 x0 H% T. e/ [
51.     var builder = factory.CreateBuilder()
    9 a6 e, h  l; |) R5 D# r
52.    .WithLanguage("english");
    * t/ N6 u. P) o  u' @
53.     using var processor = builder.Build();
    6 K8 W; P' D; n% _2 Y2 W) P
54. 
    : g9 A* W8 D- y+ R* s
55.     using var fileStream = File.OpenRead(opt.FileName);
    
56. 
    
57.     var wave = new WaveParser(fileStream);
    
58. 
    ; b3 v% L1 I# h% S) Y- m6 T
59.     var samples = wave.GetAvgSamples();
      ^: Y" Z" Y: i
60. 
    
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    5 C) J2 X  k* i5 a3 Q1 T
62.     Console.WriteLine("Language is " + language);
    6 k. l1 l( q3 v* h3 n$ l% |
63. }
    2 ^" [+ s, ]' b% h0 A% ~
64. 
    
65. async Task FullDetection(Options opt)
    * M, j5 I3 @! s9 O8 l9 ]
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    
69. 
    0 Q* v) J0 m0 S
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    
72.     string languageOption = Console.ReadLine();
    
73.     var builder = factory.CreateBuilder()
    0 T4 j1 U. {5 |% ?6 w& {4 [% F* P
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    
75. 
    
76.     if (opt.Command == "translate")
    
77.     {
    
78.         builder.WithTranslate();
    
79.     }
    
80. 
    
81.     WhisperProcessor processor = builder.Build();
    
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    : {2 {/ I$ d% ?. A# V# O
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    . O( ]: N  Z1 {% S5 t
87.     string targetDirectory = Console.ReadLine();
    
88. 
    3 A8 a6 \, Y% w" k) t: G
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    
90.     {
    
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    
92.         return;
    
93.     }
    
94. 
    * D# p# e; w$ A( a( |  H' p" ?( p4 K
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    
96. 
    ! ]" U( |  L. X; G9 r
97.     Console.WriteLine("处理完成！");
    
98. 
    1 e0 s8 ~/ h4 S1 S$ f4 J
99. }
    
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     
101.     string sourceDirectory, string targetDirectory)
     
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     ) ?9 z5 w2 |6 ]# a3 r8 n* n# E, L
104. 
     
105.     foreach (var sourceFilePath in files)
     
106.     {
     
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     4 |' v6 _) s: \& A* m/ E7 t
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     + I) ?1 [: [! o: l8 s& W3 p' H
110. 
     " ~# z+ }$ i2 y1 m- R
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     6 F, z( g+ _8 d8 v% K
112. 
     
113. 
     ; [" m% Z8 S9 h( I4 K) v0 g, ~) u
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     * z' U& U9 o0 [: g
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     ! e' ], W) d* B( w3 Y, {5 [
117. 
     
118.             using var fileStream = File.OpenRead(sourceFilePath);
     
119.             var segmentIndex = 1;
     
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     1 {- O/ n0 b5 ]" m+ R; X
121.             var startTime = DateTime.Now; // 记录开始时间
     
122. 
     : i5 _4 V; j# R5 q# E0 w
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     5 G7 A6 T/ v+ `7 C; W* u
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}");
     
127.                 Console.WriteLine(segment.Text);
     8 G0 O$ ]1 n6 G7 D; O3 I- @' c" d
128.                 Console.WriteLine();
     
129. 
     
130.                 // 将srt内容写入文件
     - _' p' t2 e: v$ F
131.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     5 g- x1 Z; R7 {+ i- o  q
132.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     ) e1 z! o2 W0 q/ D5 d8 @
133.                 await writer.WriteLineAsync(segment.Text);
     
134.                 await writer.WriteLineAsync();
     
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     
137. 
     * Q% z4 e9 y9 @; P9 Q6 u* w/ p4 f( I! O
138.                 segmentIndex++;
     
139.             }
     
140. 
     ) M3 C9 s6 P) f0 c1 v
141.             var endTime = DateTime.Now; // 记录结束时间
     ; b9 R- m# }! d1 _2 g9 B, A
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     . v( T$ W: z2 E3 F4 k" R- h# u
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     
145.         }
     - m7 M( a$ H) e! t; z% C
146.         else {
     , _4 L. ?, t+ N1 S# M
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
       g$ i% D+ `# c1 K4 A
148.         }
     , C! X0 I6 g1 m, c9 X$ p% B* u
149.     }
     ) Z2 ^' q" h. d% m4 _. J. [
150. }
     8 X5 f, Y+ }5 Q
151. public class Options
     
152. {
     
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     # U" f% n* N/ ?6 @/ m; {; F
154.     public string Command { get; set; }
     
155. 
     2 X9 _' Z9 L0 r
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")]
     3 [; T$ L6 Z9 M' n6 i
160.     public string Language { get; set; }
     ; q. M) B0 W' g
161. 
     
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     
163.     public string ModelName { get; set; }
       F( D! x  [; `$ p$ _0 R! E9 C
164. 
     0 w  @) j; W3 q- C3 ^5 |
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
     # r. O( Z) b$ A9 P( h
166.     public GgmlType ModelType { get; set; }
     
167. }
     * n& \- ~2 ~3 D9 S

复制代码

/ {: }7 _: ~3 f! q9 ~7 T& l
( x* t. R4 H2 |7 ]" E% v
4 K4 z$ P, |3 X5 F  n+ ]8 \3 [