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

shane007

本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
1 n6 d& T# i0 X% R: S2 S+ `( q 使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件
6 R6 m5 g9 j! j0 b" a/ o
代码如下
以下这句用多线程可以增速，否则很慢
以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。
: w$ k. R) n+ `( J. c4 C

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

复制代码

, S( n- c& [* V9 z
% X$ b1 j6 S# p) H2 A& @) K

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   
3. using System;
   $ E; n9 p- f4 y) x  h& j
4. using System.Diagnostics;
   
5. using System.IO;
   # @& E% t8 [! A' E' x$ p! T
6. using System.Threading;
   . Q" \5 X! k1 Z3 z
7. using System.Threading.Tasks;
   
8. using CommandLine;
   
9. using Whisper.net;
   6 j8 o* \$ J3 r% ]) K
10. using Whisper.net.Ggml;
    
11. using Whisper.net.Wave;
    6 e1 _/ ]0 x8 ^' c% V8 X" v1 n
12. 
    
13. await Parser.Default.ParseArguments<Options>(args)
    
14.     .WithParsedAsync(Demo);
    
15. 
    / I( w3 m$ r' N7 Q" I5 m
16. async Task Demo(Options opt)
    % n4 k+ r/ I/ M& R$ S" N: {* |
17. 
    ( ?& D3 b* C, Y3 k# O
18. {
      x3 ~" J. c* v! k$ O8 ^
19.     if (!File.Exists(opt.ModelName))
    # {$ f2 t: N/ V" ~& F' Y
20.     {
    6 r/ J  D( o/ R
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);
    ( E3 v1 C4 _( G( z# Z
24.         await modelStream.CopyToAsync(fileWriter);
    8 W1 S  X6 Y' m* q& Q( X
25.     }
    / o9 R# o& `8 \" x# P' e
26. 
    
27.     switch (opt.Command)
    
28.     {
    2 I! Y/ ~  k: m
29.         case "lang-detect":
    3 A, f* d: i0 U3 k2 E8 s8 D3 w
30.             LanguageIdentification(opt);
    6 u0 V1 n8 Z) t9 U  S
31.             break;
    
32.         case "transcribe":
    & V* q7 [& R' F
33.         case "translate":
      K$ Q" R  m! y
34.             await FullDetection(opt);
    , `9 q- p3 y! X& E1 o
35.             break;
    6 v1 s# C+ T. ^2 ^- p5 R' X
36.         default:
    
37.             Console.WriteLine("Unknown command");
    / X3 w% y6 j+ I4 O2 p' ?0 k
38.             break;
    
39.     }
    & j5 M0 t6 H! l4 w
40. }
    " C0 G7 K6 u* b  c
41. 
    9 @8 t; C, [2 J
42. void LanguageIdentification(Options opt)
    6 N: E, r+ s% S# M) V
43. {
    
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    
45. 
    0 B, u; o* h8 t- D' d8 l
46.     // Same factory can be used by multiple task to create processors.
    " m* O* U" f! F4 W0 L
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    
48. 
    
49. /*    var builder = factory.CreateBuilder()
    ; V2 @, C5 F; t
50.        .WithLanguage(opt.Language);*/
    
51.     var builder = factory.CreateBuilder()
    
52.    .WithLanguage("english");
    
53.     using var processor = builder.Build();
    " Q& v2 `+ v- r
54. 
    ' y, y+ Q+ T& i/ d1 N  ^
55.     using var fileStream = File.OpenRead(opt.FileName);
    4 P3 b# Q9 b# V( ^
56. 
    
57.     var wave = new WaveParser(fileStream);
      X) b9 B' F2 ?6 U1 f  @$ z
58. 
    
59.     var samples = wave.GetAvgSamples();
    " F' x: K: U# K8 P! \6 k
60. 
    
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    
62.     Console.WriteLine("Language is " + language);
    ; _6 \; Z- `8 }, |& u$ A
63. }
    ; j" W6 X/ q, Q
64. 
    
65. async Task FullDetection(Options opt)
    5 P% G0 S1 S& j8 {& c( r5 X
66. {
    ; w, ?. O+ Q) ~
67.     // Same factory can be used by multiple task to create processors.
    
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    
69. 
    * z( F8 l, e$ ?; w
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    ) R8 y) m) ^) T. L6 [+ d# w
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    : k: @, M% l' y. m
72.     string languageOption = Console.ReadLine();
    
73.     var builder = factory.CreateBuilder()
    
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    / _$ m7 r. X# j) a! v1 Q- B
75. 
    
76.     if (opt.Command == "translate")
    
77.     {
    ! `- i$ d/ [8 K$ Q
78.         builder.WithTranslate();
    
79.     }
    7 `8 c( C, \: k  m
80. 
    5 o/ m4 B' [+ V; l8 h; ~
81.     WhisperProcessor processor = builder.Build();
    
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    
87.     string targetDirectory = Console.ReadLine();
    9 A8 {# a: q0 }" t+ m9 J
88. 
      j4 s4 s/ |1 ?; T6 X
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    / P+ R! r: n* N! G/ i
90.     {
    
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    
92.         return;
    " g2 j6 F6 U6 u5 e9 z: @& z5 M' [
93.     }
    
94. 
    
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    
96. 
    ! m- L- z. {# ?8 u/ d, d" }
97.     Console.WriteLine("处理完成！");
    % f/ R. H# k( T! P7 p2 g3 k
98. 
    
99. }
    5 {  B3 L& A+ k/ u3 R
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
       a7 t# t* P' u: U5 U' r
101.     string sourceDirectory, string targetDirectory)
     
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     : g- |& g8 u5 Z# K3 u9 [3 S
104. 
     1 z5 i% j! I0 h5 C4 T, s
105.     foreach (var sourceFilePath in files)
     
106.     {
     
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     2 ?! j1 Y3 L: `0 T  z& ~: i
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     ; h' k4 w6 O. T
110. 
     $ J" `% h5 W0 i; ]6 Z" L7 \7 y
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     
112. 
     
113. 
     ; J! b% V5 K% \3 O
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     ! E& a$ i  c+ a2 Z" U/ i
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     & A" o4 i6 V% Z  ?
117. 
     2 S) K( P. D1 U/ b- }) y
118.             using var fileStream = File.OpenRead(sourceFilePath);
     6 a9 J, A# ^7 c, w; L* O5 l7 W
119.             var segmentIndex = 1;
     
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     + N1 k( Y- g2 L/ ^; E: B
121.             var startTime = DateTime.Now; // 记录开始时间
     7 I8 j6 m% o5 N# m) }6 }
122. 
     * s; s$ X; f) B
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     / o1 A3 k/ d( C& {
124.             {
     " e; S5 {, K2 Z1 R) U
125.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     $ o7 i8 D# w2 ?( j8 P* d8 r# u4 p
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}");
     4 t6 c1 Z6 v! g6 W' o* j( C
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();
     
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     1 c4 B5 S7 o+ m/ W
137. 
     
138.                 segmentIndex++;
     ) K7 T8 D% t7 q4 X+ {  t3 R8 x4 F
139.             }
     
140. 
     * E" _% X% k" f$ _% P5 m
141.             var endTime = DateTime.Now; // 记录结束时间
     
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     / n: A/ }5 I) }1 _8 _. M
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     % Y$ C, W- t! o' X" K' V! G
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     
145.         }
     1 f. A1 t% T+ y
146.         else {
     . f1 N6 v7 G" G# B/ q$ s7 o
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     
148.         }
     
149.     }
     
150. }
     ! v! y$ g' B, K" K
151. public class Options
     
152. {
     3 x; j- E' P& O% Z  w
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     6 Y1 O; U* p# N& I
154.     public string Command { get; set; }
     7 q& o' U- Z3 G
155. 
     
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     % X6 \- p% e! u6 u: {& d
157.     public string FileName { get; set; }
     7 V. i' j- s7 w
158. 
     
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     
160.     public string Language { get; set; }
     
161. 
     7 r' M2 N0 [& G: q2 r' ?5 Q
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     
163.     public string ModelName { get; set; }
     ; q. q, G0 }8 _6 E( }- o' Q" t
164. 
     
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
     
166.     public GgmlType ModelType { get; set; }
     
167. }
     ! d2 x8 o# w/ d$ V- R5 H6 v

复制代码

5 h* ?  k# Z& ~; u8 N$ T
2 H5 g7 ~* u/ q! ^+ Q/ S2 Z