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

shane007

+ w( J' u1 Q: r& L7 {7 j9 d
  W/ u" [; q3 k 本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
4 H8 I- R# P& l1 B: y 使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件
' }% e/ q4 E6 o9 k
代码如下
5 w% @, |* a/ w0 Z以下这句用多线程可以增速，否则很慢
; v( a0 ]: E* v5 U( U3 P/ t9 A+ }以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。

4 v0 |1 U7 O1 ?$ X+ G
( r3 i$ O, d9 f: X1 u

1. var builder = factory.CreateBuilder()
   
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

# B/ S/ S7 j- [0 Z: V  a- W- \

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   
3. using System;
   5 d9 \& k: T% S6 }. D; z, L
4. using System.Diagnostics;
   
5. using System.IO;
   8 l; m. L) u& h5 p
6. using System.Threading;
   $ S; z% R+ h. n# b8 u
7. using System.Threading.Tasks;
   % a# q. X! |. a/ U" a7 q
8. using CommandLine;
   
9. using Whisper.net;
   1 a6 k$ T: h8 ~$ b0 @1 G% T, w8 s
10. using Whisper.net.Ggml;
    
11. using Whisper.net.Wave;
    
12. 
    ( _# F( Q# H9 S2 }; V4 o6 I. a
13. await Parser.Default.ParseArguments<Options>(args)
    0 v- L+ k* F# {1 h
14.     .WithParsedAsync(Demo);
    . m: g8 t7 M9 C( E6 o3 ?
15. 
    8 q% D# @& y8 L) q# p& I
16. async Task Demo(Options opt)
    3 @/ {% }# b9 E. z( b. M$ [/ v: n
17. 
    
18. {
    1 C, H' L' @3 V: D# w- O& r+ u
19.     if (!File.Exists(opt.ModelName))
    ' ^! K, `+ w( V$ K
20.     {
    
21.         Console.WriteLine([        DISCUZ_CODE_1        ]quot;Downloading Model {opt.ModelName}");
    % e5 c% H5 R/ e$ e) W' m( z
22.         using var modelStream = await WhisperGgmlDownloader.GetGgmlModelAsync(opt.ModelType);
    ) u( y: A1 I6 u2 ?; _0 X
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    
24.         await modelStream.CopyToAsync(fileWriter);
    
25.     }
    # a5 v( g( }6 k& J) \8 ]
26. 
    + o9 e3 W/ Y5 J6 a+ }8 c; G* n
27.     switch (opt.Command)
    & l0 F2 p0 F) _& S2 g! [6 b
28.     {
    / c3 ~, f' j) j. }+ x( r6 J
29.         case "lang-detect":
    
30.             LanguageIdentification(opt);
    & D% O' N* X# i9 x4 G* {- B+ [- e  P
31.             break;
    
32.         case "transcribe":
    : W, j3 y) V8 W# x- Q( |4 [1 L% b
33.         case "translate":
    
34.             await FullDetection(opt);
    
35.             break;
    * W) w9 r& t: h# g3 ]9 @( u
36.         default:
    
37.             Console.WriteLine("Unknown command");
    0 o' u" P& I5 |6 J! Z7 y
38.             break;
    ; I7 C. Y% V. K; N6 ?
39.     }
    
40. }
    , U! h+ s7 p2 F" e: B) I
41. 
    1 m5 {3 K- c8 B
42. void LanguageIdentification(Options opt)
    : S/ u( n  T& ?1 S
43. {
    
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    ' Y" \: @* P+ p! U
45. 
    
46.     // Same factory can be used by multiple task to create processors.
    
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    
48. 
    5 ?7 ?$ S2 f8 K. q
49. /*    var builder = factory.CreateBuilder()
    
50.        .WithLanguage(opt.Language);*/
    
51.     var builder = factory.CreateBuilder()
    
52.    .WithLanguage("english");
    0 P3 J6 `/ k  ]# S  h: r
53.     using var processor = builder.Build();
    4 Z0 t# C$ G5 e* n/ t& }; @
54. 
    1 }3 `# w4 H5 Z  p( t3 d( k5 d, N
55.     using var fileStream = File.OpenRead(opt.FileName);
    
56. 
    ) p' S. z; B9 q. ?; D7 H: `
57.     var wave = new WaveParser(fileStream);
    4 M4 w3 T$ o6 X3 W2 H( o  m+ |; v" [
58. 
    
59.     var samples = wave.GetAvgSamples();
    7 E2 f9 a# {& F  Y
60. 
    
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    
62.     Console.WriteLine("Language is " + language);
    / r: O+ ~: t  @+ Y
63. }
    
64. 
    
65. async Task FullDetection(Options opt)
    
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    * u0 w& N; z2 B1 n  ]" Q6 U
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    
69. 
    
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    
72.     string languageOption = Console.ReadLine();
    
73.     var builder = factory.CreateBuilder()
    ( _# g4 g/ \  Y5 C& r( w
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    - ~  `; r, [0 F! Y# H" J0 y
75. 
    
76.     if (opt.Command == "translate")
    & R/ w# u* d/ i
77.     {
    
78.         builder.WithTranslate();
    
79.     }
    
80. 
    
81.     WhisperProcessor processor = builder.Build();
    
82.    
    4 b/ V6 r# R0 j3 E5 x" M  ^
83.     Console.WriteLine("请输入wave源文件目录：");
    2 W% U* ]3 g1 F$ R: ~
84.     string sourceDirectory = Console.ReadLine();
    ! M. _4 G' R1 q$ r
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    
87.     string targetDirectory = Console.ReadLine();
    . }( `8 I& R% r! S' j- N
88. 
    
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    
90.     {
    ) m: D( `" s: n/ `
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    9 J( S! f* v# T0 n8 _: H
92.         return;
    
93.     }
    
94. 
    ; r4 J% r1 _* m5 G2 o* q5 z9 c
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    
96. 
    * Z$ y% O! @2 W
97.     Console.WriteLine("处理完成！");
    $ I5 w* O  b% Y' j/ O
98. 
    
99. }
    6 i# Y. i- t$ h+ F% z7 D8 l$ m( _+ \
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     
101.     string sourceDirectory, string targetDirectory)
     
102. {
     2 E$ D! }' f' h3 A3 h
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     
104. 
     
105.     foreach (var sourceFilePath in files)
     ( S4 X2 e  Q3 l, n1 U
106.     {
     
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     * F) Y8 Y. i3 Y$ I0 S
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     " U, P; w+ m) K  J0 e
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     6 S5 v5 N  P' ^9 F7 t
110. 
     , U) [" D* @" g2 {- ~4 d# d
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     6 b( y+ H, Y/ L
112. 
     # K, L2 C0 U( N5 a
113. 
     
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     9 g4 s6 l; a  v. d; {% ]
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     
117. 
     % S& _! z2 _0 ^3 J. E0 k
118.             using var fileStream = File.OpenRead(sourceFilePath);
     
119.             var segmentIndex = 1;
     , |* P$ {: w0 h0 B" `" c
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     4 o9 P- L% {8 _
121.             var startTime = DateTime.Now; // 记录开始时间
     
122. 
     4 |1 }% r: E. g) P6 r0 h
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     . ?% Z/ T$ a/ u: t' G: {0 N. _
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}");
     : _9 Q# A- K9 O
127.                 Console.WriteLine(segment.Text);
     
128.                 Console.WriteLine();
     
129. 
     
130.                 // 将srt内容写入文件
     ! m$ _4 M& ~8 k/ \* }! B
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);
     $ ]9 G0 j$ _# I: ~% \/ k2 o* @0 y/ V
134.                 await writer.WriteLineAsync();
     1 i1 W# R) h- y. `2 U
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     
137. 
     / v: j5 l" x: M$ j# R
138.                 segmentIndex++;
     ) ]4 v8 E! X: I  l9 V
139.             }
     ' r  c; ~- ~3 T, u' ^7 u1 o
140. 
     
141.             var endTime = DateTime.Now; // 记录结束时间
     
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     # e& Z6 N6 ]" l& Z8 c- ^# }
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     
145.         }
     
146.         else {
       A3 x3 ~" j  a& v: Z
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     
148.         }
     3 v( k8 B( g( U% g  U$ x8 a
149.     }
     
150. }
     ( q/ K+ O5 D( A8 x
151. public class Options
     
152. {
     
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     ! j  ?& t6 U* a$ o( J$ x% U: t& B
154.     public string Command { get; set; }
     
155. 
     
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     9 ?; [. E5 u: X2 X% R) D" F
157.     public string FileName { get; set; }
     0 u8 D; T) O0 t* C0 s) m, W9 k
158. 
     
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     
160.     public string Language { get; set; }
     
161. 
     ! d, L$ X0 i/ G
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     
163.     public string ModelName { get; set; }
     $ f) u3 R5 Z' {) Y* T
164. 
     
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
     ; F3 J% f+ l# J( e) M
166.     public GgmlType ModelType { get; set; }
     
167. }
     % j# g; g- U$ }; i0 t% {3 t9 a8 Y

复制代码

* f7 e. Q& K0 ^  Q+ F; G! N) H$ s