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

shane007

本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
9 K- z5 f" a' I, [( t; t' M* K 使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件
+ a1 x- z8 h, Y& A  ]* q
代码如下
8 y/ m; |) _3 h" d以下这句用多线程可以增速，否则很慢
以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。
1 S. s4 ?* J. _! V$ w

1. var builder = factory.CreateBuilder()
   9 s8 B3 B& d( ~- d/ b- U6 n) Q- M
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

7 P) z# Z) z. c5 Q2 h

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   
3. using System;
   8 _9 }, ?# d- Y( ~1 z
4. using System.Diagnostics;
   
5. using System.IO;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   # l1 V1 E8 |9 `3 Y' [, }
8. using CommandLine;
   
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    
11. using Whisper.net.Wave;
    " R9 ]8 k$ M% C' t" z
12. 
    
13. await Parser.Default.ParseArguments<Options>(args)
    - V+ R$ q% R- z* _5 a
14.     .WithParsedAsync(Demo);
    " z3 ?7 h5 q  s5 d, P
15. 
    & c! Q' @) F1 p9 ]
16. async Task Demo(Options opt)
    
17. 
    
18. {
    % u* K1 F5 a) O/ s+ |( {
19.     if (!File.Exists(opt.ModelName))
    
20.     {
    " h: h. m3 H# F  J4 y4 d6 [: W
21.         Console.WriteLine([        DISCUZ_CODE_1        ]quot;Downloading Model {opt.ModelName}");
    % c" U4 a' t7 N0 d: r
22.         using var modelStream = await WhisperGgmlDownloader.GetGgmlModelAsync(opt.ModelType);
    ' d$ m0 J! c6 u  i% a
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    
24.         await modelStream.CopyToAsync(fileWriter);
    
25.     }
    ! @" R2 \7 |; l. i: b; J6 C
26. 
    
27.     switch (opt.Command)
    / \1 [. a7 g8 F  y$ `
28.     {
    ; S# g: U: K; w3 h' P
29.         case "lang-detect":
    ( O/ ~! t1 s% W9 U& A
30.             LanguageIdentification(opt);
    
31.             break;
    
32.         case "transcribe":
    
33.         case "translate":
    4 A: e+ z8 [/ S8 ~3 y4 c; E1 v
34.             await FullDetection(opt);
    
35.             break;
    3 Z1 r# a: w% V" G& W6 }
36.         default:
    + i( P6 m& s9 V$ @: M7 A5 B7 d% v
37.             Console.WriteLine("Unknown command");
    * ]5 G% x7 O0 e: t& C- f
38.             break;
    
39.     }
    
40. }
    - b4 V4 f- Z& ~/ |# V
41. 
    
42. void LanguageIdentification(Options opt)
    
43. {
    : K8 S8 h7 t# C: r" D
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    
45. 
    
46.     // Same factory can be used by multiple task to create processors.
    : P( Y: c( W* B% o. H" Q& S* ^
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    
48. 
    
49. /*    var builder = factory.CreateBuilder()
    # g9 k% i; y5 O" A% l- W% x. M( a
50.        .WithLanguage(opt.Language);*/
    4 `, p: F7 Q, e5 ~% E
51.     var builder = factory.CreateBuilder()
    6 z3 U4 `3 f, g2 n- s8 z
52.    .WithLanguage("english");
    / z# a  W* O$ |
53.     using var processor = builder.Build();
    + o; n0 q2 H8 y( |3 P# A. i
54. 
    2 T9 F0 I+ w9 s7 ^/ x9 U( g
55.     using var fileStream = File.OpenRead(opt.FileName);
    
56. 
    
57.     var wave = new WaveParser(fileStream);
    
58. 
    
59.     var samples = wave.GetAvgSamples();
    
60. 
    
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    2 w4 j" ~) A2 P# J$ P
62.     Console.WriteLine("Language is " + language);
    , b0 g: H' r1 r+ U1 r
63. }
    
64. 
    # }/ R4 k, \4 o3 S- @
65. async Task FullDetection(Options opt)
    
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    ' P9 F' a% K8 X; B5 Z
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    
69. 
    ! z# Z! E) o, G
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    * c: m/ }" h' g$ D4 j; A4 |1 N8 z
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    2 j. ~  w9 S* x3 ]5 }+ S  Q
72.     string languageOption = Console.ReadLine();
    8 O  t- H  p- _7 `) W0 i, t
73.     var builder = factory.CreateBuilder()
    " J: s" ]' J+ ]4 P# b
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    
75. 
    
76.     if (opt.Command == "translate")
    / O$ U  c1 ]$ _2 |$ ~8 \* m
77.     {
    - U2 \+ Q/ X& l' ]! g
78.         builder.WithTranslate();
    ! d4 b5 q# H4 L  {
79.     }
    % h  J$ J% P) W
80. 
    
81.     WhisperProcessor processor = builder.Build();
    
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    * }$ G* G- x2 f0 w4 f* C* h4 \
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    & B  @% N: I, S! V* h
87.     string targetDirectory = Console.ReadLine();
    
88. 
    
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    & m2 L" G6 Z4 o! p" I" f
90.     {
      J0 a- R2 R+ r; G
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    . z' I% v5 y4 [. ?6 V0 d' k
92.         return;
    ; ~- ?/ H' }( r" R$ F( t  X
93.     }
    . e) b! [5 }2 t2 c
94. 
    
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    
96. 
    $ o# h8 B. R: {) }" O
97.     Console.WriteLine("处理完成！");
    8 T7 c5 _; c5 o  R; b' F# [
98. 
    0 c* F8 E9 J! h) w1 A; E. m
99. }
    
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     
101.     string sourceDirectory, string targetDirectory)
     
102. {
     3 }8 j; ]! d( t# p2 D
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     
104. 
     
105.     foreach (var sourceFilePath in files)
     9 _& v2 u2 b) ]8 C
106.     {
     
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     $ v- P4 p  z9 F% D+ @
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     " X# I. P8 l# ]
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     
112. 
     6 S; x' M/ g/ i# f
113. 
     . I! r* p, {# y7 K% K
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     6 Z6 R3 N5 V- v0 s* o( Y
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     # j- W6 J/ G+ w4 K( O
117. 
     
118.             using var fileStream = File.OpenRead(sourceFilePath);
     
119.             var segmentIndex = 1;
     
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     9 B( E" I: S' l$ S3 G8 o; v
121.             var startTime = DateTime.Now; // 记录开始时间
     
122. 
     
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     
124.             {
     5 z# _: n2 u* l% W* y
125.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
       W, ?8 ]5 R( g  j5 e5 p1 u* \: P) F
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. 
     . O& y( ]- [0 `' Y, }- G& p
130.                 // 将srt内容写入文件
     4 P4 f$ \  e9 e' e+ _1 _& C- h
131.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     1 x4 y9 X* |! v; ]* z0 S- H( |& C. S
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);
     
134.                 await writer.WriteLineAsync();
     , a  J$ R; ]& R# _0 ^0 F+ i6 x" Z
135. 
     8 Z. ]2 L" K0 j" v1 H
136.                 writer.Flush(); // 立即保存srt文件
     ( N1 z& \8 N2 t& i6 _- Z. @2 S/ |
137. 
     
138.                 segmentIndex++;
     
139.             }
     7 L' W7 }+ @! I- P+ j1 i
140. 
     
141.             var endTime = DateTime.Now; // 记录结束时间
     
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     % d% i. V$ R* ]3 O
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     % W8 K$ {5 b+ o8 v+ V4 a
145.         }
     
146.         else {
     
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     9 n* r5 J1 f- I7 \
148.         }
     
149.     }
     : a4 A0 P8 }( B
150. }
     1 R2 D4 X1 i. ?; D6 a
151. public class Options
     
152. {
     ! s. i/ B( X- r
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     # f( L1 F- n( x; ^8 f
154.     public string Command { get; set; }
     
155. 
     3 R  H7 g4 [7 x- N+ f
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     
157.     public string FileName { get; set; }
     ! K8 U0 S1 N* c/ @* j
158. 
     
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
       {+ h0 ?. L% k: u0 h
160.     public string Language { get; set; }
     
161. 
     . p7 V7 q  @  f) o
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     
163.     public string ModelName { get; set; }
     + b+ T% f2 O  [* l! }& c2 Q6 u' S
164. 
     5 G* i0 y' n- m- g6 T
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
     
166.     public GgmlType ModelType { get; set; }
     & Q. x# R! ?$ e% b( @. G
167. }
     

复制代码