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

shane007

本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
6 s  z' s) u+ v; t 使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件
) u0 |2 e& |+ g/ A5 R8 Q" r
9 t4 v1 ?  q7 o0 W代码如下
以下这句用多线程可以增速，否则很慢
以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。

; r; Q! `  H' r" c

1. var builder = factory.CreateBuilder()
   $ G- F9 ^: |4 S& F% h" e; H3 q! L
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

/ y; r" w. l) g$ {- I2 ?3 J! z
/ Z( P: a" x; v

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   9 M% k. o) C/ ]# l4 K
3. using System;
   
4. using System.Diagnostics;
   8 C9 d& t, y0 r4 z) P
5. using System.IO;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   # }! N9 H6 |8 M1 s
8. using CommandLine;
   
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    & y' N- G& x; K
11. using Whisper.net.Wave;
    ' J7 ]& B/ V' O' i' U6 D% z
12. 
    
13. await Parser.Default.ParseArguments<Options>(args)
    ) h. }4 t2 w% }9 q4 ?
14.     .WithParsedAsync(Demo);
    $ Q; X( T) y" h+ B% N! f0 t
15. 
    
16. async Task Demo(Options opt)
    . b0 c( u" R" H" }; h1 P! F
17. 
    
18. {
    1 X& |( w( j, g" x4 {' p" b4 F! J
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);
    
24.         await modelStream.CopyToAsync(fileWriter);
    0 V, u' r0 b9 W4 v0 R, {) M
25.     }
    
26. 
    / U% {$ A4 o6 l
27.     switch (opt.Command)
    
28.     {
    
29.         case "lang-detect":
    1 D" l; X* X8 n( Q6 @! M3 T3 d1 U
30.             LanguageIdentification(opt);
    2 g6 U; f/ n( ~$ h6 @, n
31.             break;
    7 D8 X3 w3 B: w+ D8 T. ^5 d% F
32.         case "transcribe":
    $ I& b) g- O' s# p# n
33.         case "translate":
    
34.             await FullDetection(opt);
    3 |8 R! u* m* ~
35.             break;
    
36.         default:
    
37.             Console.WriteLine("Unknown command");
    
38.             break;
    
39.     }
    
40. }
    7 b: H; Q' |0 d
41. 
    # M+ a; }/ o$ N5 u' o4 s
42. void LanguageIdentification(Options opt)
    
43. {
    $ I) B" J+ b5 v! c8 P
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    : ~- b; |! s7 N. u. ^8 `! ~; F$ i
45. 
    % ]$ a) O1 Y' H6 f
46.     // Same factory can be used by multiple task to create processors.
    
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    . \- w* z: r0 s) z  T* R* D! V
48. 
    
49. /*    var builder = factory.CreateBuilder()
    
50.        .WithLanguage(opt.Language);*/
    / @8 h6 B1 E7 Q7 o6 r! ^
51.     var builder = factory.CreateBuilder()
    8 J7 E0 ^  J4 N7 y, ^6 w
52.    .WithLanguage("english");
    , r' O9 R1 F) j5 i0 l
53.     using var processor = builder.Build();
    
54. 
    8 w  L, u" Z% S3 _4 U
55.     using var fileStream = File.OpenRead(opt.FileName);
    4 Z# o( d* T9 r$ T* b& e
56. 
    
57.     var wave = new WaveParser(fileStream);
    
58. 
    + b. U7 g9 Q8 v. N1 S: j
59.     var samples = wave.GetAvgSamples();
    
60. 
    
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    # z  ^# s* V, ]! ?6 ?4 L
62.     Console.WriteLine("Language is " + language);
    
63. }
    
64. 
    
65. async Task FullDetection(Options opt)
    
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    
69. 
    
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    9 f8 v# A5 v" K  h
72.     string languageOption = Console.ReadLine();
    , m+ D/ L! Y% {6 {7 x7 C
73.     var builder = factory.CreateBuilder()
    % e/ s4 |+ ?) i
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    
75. 
    8 h! s6 K# b/ X8 q0 C" T
76.     if (opt.Command == "translate")
    8 [! r) Q( B) P, b- q( a; ~
77.     {
    
78.         builder.WithTranslate();
    ' T& G: }8 E$ h1 G0 `+ R9 j
79.     }
    , |* u6 o9 Z. O& ~9 b
80. 
    , O4 c8 n2 @0 v! i9 Q( @
81.     WhisperProcessor processor = builder.Build();
    ! X0 X! U) L- f4 `
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    9 `! g1 [( ^- c& g  ]5 e  J
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    # A0 P8 C/ ?5 ]# V! B
86.     Console.WriteLine("请输入目标文件目录：");
    " _5 C: e- T0 d- u
87.     string targetDirectory = Console.ReadLine();
    
88. 
    # _2 [6 u1 ^: \$ U4 x
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    - j8 O- W3 o( _+ p, A
90.     {
    
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    8 h- L2 u3 m' O3 ^+ v
92.         return;
    8 g1 O9 g; v- d" L4 _; A" {
93.     }
    
94. 
    ! t' g: n+ \( X3 z3 v; M0 H/ O
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    % L. G- p, A6 C! E- d: q
96. 
    : {! ], A4 W' Q
97.     Console.WriteLine("处理完成！");
    ( K7 Q+ Q5 ~3 C6 [+ p& d4 U
98. 
    * i2 E. f/ C/ V- R  Q. w
99. }
    7 F& Q9 Q/ a5 c) m
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     
101.     string sourceDirectory, string targetDirectory)
       O, W& C9 _: v' `
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     4 o& g5 j$ x% E! E
104. 
     " ]% n/ F7 l3 b
105.     foreach (var sourceFilePath in files)
     4 B( J' L7 l- s: R1 R
106.     {
     
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     6 _( D# o" q# \8 z& m- I
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     , p$ w' B# q" F' e/ G' }' x
110. 
     3 ~1 a! Y+ @, m* x! Q* g
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     : G* j. P/ L7 ~' k+ ]! F8 I/ G8 m  m
112. 
     
113. 
     9 ^6 f7 |6 ~& _) P5 T4 J% y9 B: p! E: F
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     1 w& g* Z6 K8 e, u
117. 
     
118.             using var fileStream = File.OpenRead(sourceFilePath);
     
119.             var segmentIndex = 1;
     
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     
121.             var startTime = DateTime.Now; // 记录开始时间
     
122. 
     
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     
124.             {
     
125.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     ( L. Z  p$ n1 l- @
126.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     
127.                 Console.WriteLine(segment.Text);
     
128.                 Console.WriteLine();
     
129. 
     
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}");
     # m' K1 u! M% u6 R
133.                 await writer.WriteLineAsync(segment.Text);
     : u4 w! I, G. Y( y
134.                 await writer.WriteLineAsync();
     . k$ b* v! e+ r* q: ]/ |0 e
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     # g( j, ?6 S' ]1 E' V
137. 
     ! e) v- d' r: T& w& i9 p
138.                 segmentIndex++;
     
139.             }
     
140. 
     
141.             var endTime = DateTime.Now; // 记录结束时间
     
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     & \7 ?* K  m0 h+ [2 T% j
145.         }
     * E" I0 |& ?3 K
146.         else {
     
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     ( E+ H* G! Q1 X1 H" E  F
148.         }
     " \2 g! ^  Y3 K7 _
149.     }
     
150. }
     
151. public class Options
     
152. {
     4 G& w2 h7 B! D$ E
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     $ s+ u& B/ L& A8 Y% y5 ]
154.     public string Command { get; set; }
     * G# V3 I1 Y+ ?" q  E; w. h* q
155. 
     
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     4 E1 a8 Z+ O. ~. R. v/ ~* n
157.     public string FileName { get; set; }
     
158. 
     4 g, y. M( k/ G& U0 n! y- Q
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     , m9 w6 c' m# C( Q4 U4 w
160.     public string Language { get; set; }
     2 W' o8 E( K- G# E' T* H
161. 
     8 ^7 i, s' A6 z/ @4 f
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     ) h/ l4 k1 l4 P9 H/ Q2 x
163.     public string ModelName { get; set; }
     2 Q' r, _2 A% i/ ^5 I& J, L. Y5 m% T
164. 
     * S* n% H! R) Q9 |
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
     # L: {! P0 d/ S6 Y# S
166.     public GgmlType ModelType { get; set; }
     2 r/ O- o0 a6 _3 u  {+ {
167. }
     / C( t3 K" S. ~9 s

复制代码

9 `4 C4 N5 Q; A# M! f% }8 f" ?