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

shane007

( l! X! ?) u: G( l3 i9 Q2 \
本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
) l6 G& ~8 d& y- n0 R/ M 使用时需要輸入语言，源文件路径和目标文件路径
- H# E9 k5 N+ V1 H1 s; ~最后是输出srt文件
2 q2 z. B6 x+ T$ l3 h/ L
代码如下
以下这句用多线程可以增速，否则很慢
以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。

8 U- {4 r; T: h% M, }/ f3 `
0 k8 n  Y8 o4 Q. L8 |! U

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

复制代码

: b* E$ J2 @; G  M5 }( k

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   6 |" z( P' P9 C  e: Q
3. using System;
   
4. using System.Diagnostics;
   
5. using System.IO;
     k; m3 ?* h/ z. z7 E4 M9 N/ Y
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using CommandLine;
   
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    
11. using Whisper.net.Wave;
    
12. 
    + }; u6 d9 G# D! c, l
13. await Parser.Default.ParseArguments<Options>(args)
    0 p& {0 j4 U& n7 y: e( v) L5 N
14.     .WithParsedAsync(Demo);
    
15. 
    9 g, F6 b. F3 a- S
16. async Task Demo(Options opt)
    2 K3 `! `+ O8 _6 b0 d
17. 
    ) F# V. L8 L1 h4 G$ n9 Q
18. {
    
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);
      P9 q$ a7 f; b( K! T
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    
24.         await modelStream.CopyToAsync(fileWriter);
    + s4 Z* z. v0 z6 x+ i4 L
25.     }
    8 ^8 s5 z& k1 Q% h* I
26. 
    4 Z. H2 @2 T# B5 F" Z
27.     switch (opt.Command)
    ; s! ?$ l- J! v' P7 l  W7 A
28.     {
    * @( A) B: O' H
29.         case "lang-detect":
    
30.             LanguageIdentification(opt);
    
31.             break;
    
32.         case "transcribe":
    ( m- @7 [9 I$ A  j& d
33.         case "translate":
    ( X( ^: [# z; Y7 u' X) g1 r
34.             await FullDetection(opt);
    
35.             break;
    
36.         default:
    
37.             Console.WriteLine("Unknown command");
    + j9 a+ n; }) \# L
38.             break;
    ! B% m1 h% Y, q4 W
39.     }
    
40. }
    
41. 
    
42. void LanguageIdentification(Options opt)
    
43. {
    
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    
45. 
    0 X+ e. X9 I8 M
46.     // Same factory can be used by multiple task to create processors.
    
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    
48. 
    . E# W- c5 k& W# D& n3 E% ]
49. /*    var builder = factory.CreateBuilder()
    ' h! _) j1 d+ N& u' _
50.        .WithLanguage(opt.Language);*/
    
51.     var builder = factory.CreateBuilder()
    3 U1 h, g5 I' g$ h0 G
52.    .WithLanguage("english");
    2 p2 j" @# k. A6 |% p
53.     using var processor = builder.Build();
    
54. 
    - q/ a$ ?) a0 \
55.     using var fileStream = File.OpenRead(opt.FileName);
    3 j6 L2 y6 j6 v9 Z8 `' ^" L
56. 
    ( I) t" G+ ^/ g" N
57.     var wave = new WaveParser(fileStream);
    5 p; {: n& C. i" [8 e6 [4 x; o
58. 
    
59.     var samples = wave.GetAvgSamples();
    # R; ]3 s% Y: f0 O, M% i6 I. b- h
60. 
    ' a. N" O, u9 \5 O
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    
62.     Console.WriteLine("Language is " + language);
    
63. }
    - o+ C4 M  G* t& c6 x3 |  f2 q& Q2 k
64. 
    + I1 A8 v4 o4 E2 J/ n" Y
65. async Task FullDetection(Options opt)
    
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    $ k9 s; c  L$ X) U: K; Z1 X
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    
69. 
    
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    : X6 e) N% c# F. w
72.     string languageOption = Console.ReadLine();
    8 ^; m: K# C. d4 O
73.     var builder = factory.CreateBuilder()
    
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    
75. 
    
76.     if (opt.Command == "translate")
    
77.     {
    
78.         builder.WithTranslate();
    + }. a1 E7 W$ F1 |
79.     }
    * i% ^) L4 z" _8 {
80. 
    
81.     WhisperProcessor processor = builder.Build();
    
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    ( g6 M* Y# W9 n2 G" `
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    7 _) a. V/ }* E0 A, H
86.     Console.WriteLine("请输入目标文件目录：");
    
87.     string targetDirectory = Console.ReadLine();
    * _# G; f/ q- Q! c' x1 r
88. 
    6 n3 I( [- f! b6 v- ?
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    ; ?( `7 ?& a- e. C
90.     {
    
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    3 j3 f) q6 X6 r3 K$ b
92.         return;
    
93.     }
    
94. 
    
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    6 [* B. y* q9 S* e; n
96. 
    ' H- b$ T* f0 ?9 B
97.     Console.WriteLine("处理完成！");
    
98. 
    
99. }
    
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     % I4 ~# e6 |7 \/ h$ s/ `
101.     string sourceDirectory, string targetDirectory)
     - s# E; m  ~" J+ X
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     5 z- c5 \6 I9 e, L9 j4 M. o1 R& c
104. 
     
105.     foreach (var sourceFilePath in files)
     
106.     {
     
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     , ~! o8 F5 r' x: j9 E- J# r
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     
112. 
       C. U. c8 e* h! i& j2 e$ {
113. 
     
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     
117. 
     
118.             using var fileStream = File.OpenRead(sourceFilePath);
     
119.             var segmentIndex = 1;
     1 }5 \, c9 r+ y6 f5 [0 w& F
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     ! u, w4 @* F' J& c% x$ p
121.             var startTime = DateTime.Now; // 记录开始时间
     
122. 
     
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     ) L1 t7 V' ~8 ^$ g
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}");
     2 b0 V4 {' J7 |& C3 Y/ s' q* `7 [
127.                 Console.WriteLine(segment.Text);
     0 q7 ^6 L; I" E" y& o$ m; _  C
128.                 Console.WriteLine();
     
129. 
     
130.                 // 将srt内容写入文件
     : O1 J" S$ W' z2 f5 Q& Q/ ^$ }- c
131.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     * h. L: m% W5 H$ y0 d. J
132.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     3 w# |( V4 u- |$ k
133.                 await writer.WriteLineAsync(segment.Text);
     
134.                 await writer.WriteLineAsync();
     0 k$ n/ f7 b' G0 E/ i
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     
137. 
     3 R+ \/ J0 k0 G* `( k
138.                 segmentIndex++;
     . L( m& d# Y; N6 ^6 z" E3 n% y
139.             }
     + n9 _$ X- I7 Y& S7 K
140. 
     6 h4 U3 L( H( j
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} 分钟");
     
145.         }
     , R8 S0 ]3 u8 x# [' V6 T
146.         else {
     
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     
148.         }
     
149.     }
     
150. }
     7 E; q* g2 ~2 `- m" N
151. public class Options
     1 t; D8 o* i4 u* y/ i
152. {
     
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     ' ]. S8 F$ S8 w
154.     public string Command { get; set; }
     
155. 
     
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     
157.     public string FileName { get; set; }
     
158. 
     , T$ h# c8 I! w" f! W
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     9 a. d1 J  M7 C6 g9 }$ ~# }8 |2 B
160.     public string Language { get; set; }
     . }# A5 y  A0 r; Y
161. 
     1 X6 d4 E6 X* c9 |% Z
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     
163.     public string ModelName { get; set; }
     0 ~' ^- I( O3 s* O
164. 
     
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
     8 ?) p7 b% H% O( I- j9 J
166.     public GgmlType ModelType { get; set; }
     
167. }
     

复制代码