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

shane007

1 b6 ~0 M5 L( D- {
9 B' @; H: `& P1 s9 ?+ ] 本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
使用时需要輸入语言，源文件路径和目标文件路径
1 ?8 T- v) [1 b6 e6 e最后是输出srt文件

0 O9 K, I9 y5 s3 t! L! y+ \1 g代码如下
以下这句用多线程可以增速，否则很慢
以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。

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

复制代码

; r4 o* ^  B1 J" _' K/ A! n1 L
, R4 _) L- F$ l- R. ~* M, O: y

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   * p, M7 [. @8 n5 m
2. 
   
3. using System;
   
4. using System.Diagnostics;
   + Z! E2 y4 }3 {$ ]$ v$ ~2 O4 k9 a
5. using System.IO;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   ' q. R8 V' T. |" [
8. using CommandLine;
   
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    7 Y/ M5 W  w  n2 O# b9 l: s$ x
11. using Whisper.net.Wave;
    
12. 
    
13. await Parser.Default.ParseArguments<Options>(args)
    
14.     .WithParsedAsync(Demo);
    
15. 
    
16. async Task Demo(Options opt)
    
17. 
    1 V/ B% u! f" w: [% C8 Q& w, s
18. {
    
19.     if (!File.Exists(opt.ModelName))
    
20.     {
    " s6 L* p+ l9 [1 o- C
21.         Console.WriteLine([        DISCUZ_CODE_1        ]quot;Downloading Model {opt.ModelName}");
    / u/ {% q0 x& E' M' F
22.         using var modelStream = await WhisperGgmlDownloader.GetGgmlModelAsync(opt.ModelType);
    
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    " [" s+ U/ p3 G5 U2 B' B9 F
24.         await modelStream.CopyToAsync(fileWriter);
    
25.     }
    
26. 
    3 Q5 i! @3 _/ j# f; A
27.     switch (opt.Command)
    
28.     {
    . o9 r$ r1 k) g% k1 e+ @
29.         case "lang-detect":
    
30.             LanguageIdentification(opt);
    
31.             break;
    " o3 U  X# ]; [" X5 J7 L: J
32.         case "transcribe":
    & o2 ^2 r; `0 o0 C; w  Z% z- C
33.         case "translate":
    . N3 z8 e4 l4 a. d/ l
34.             await FullDetection(opt);
    
35.             break;
    
36.         default:
    
37.             Console.WriteLine("Unknown command");
    6 a4 r+ \% `; r7 y+ }7 e* r7 ]# H
38.             break;
    : ?* W, f6 t) g! E
39.     }
    
40. }
    * a" P# f; {6 V  {) N/ V+ G: ?
41. 
    
42. void LanguageIdentification(Options opt)
    
43. {
    # ?8 A. w# d$ A* L
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    ) _7 G$ K. j; p3 z7 m9 |
45. 
    , o+ N/ X6 X* a3 ]6 D
46.     // Same factory can be used by multiple task to create processors.
    
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    " \% B# h, O8 d  x2 \
48. 
    ; i. R& X$ l2 X2 ^6 D
49. /*    var builder = factory.CreateBuilder()
    
50.        .WithLanguage(opt.Language);*/
      x! T' |5 i8 |9 M9 s
51.     var builder = factory.CreateBuilder()
    
52.    .WithLanguage("english");
    4 ~( }, J; \5 g9 K9 B( N
53.     using var processor = builder.Build();
    ( u7 d' h" {$ f8 W2 D* P) y
54. 
    
55.     using var fileStream = File.OpenRead(opt.FileName);
    
56. 
    
57.     var wave = new WaveParser(fileStream);
      M. I( C& g7 m* O5 y. j0 O
58. 
    
59.     var samples = wave.GetAvgSamples();
    
60. 
    1 d" e8 x2 k1 D% f
61.     var language = processor.DetectLanguage(samples, speedUp: true);
      A* i+ |  U5 X1 F, i* i% v0 D/ z
62.     Console.WriteLine("Language is " + language);
    
63. }
    3 b1 k" k$ k6 W* ^
64. 
    
65. async Task FullDetection(Options opt)
    0 U1 l# S9 ?1 e7 r- y  F
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    9 Q. ~" y) P/ U; _6 g7 \% F
69. 
    
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    + Q4 f+ y# b( @# T6 ^. _& X+ }
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    : W7 x8 a- n& T. n
72.     string languageOption = Console.ReadLine();
    
73.     var builder = factory.CreateBuilder()
    / W, l# p3 k1 k5 p* F) T1 i
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    
75. 
    + f/ I0 r) C( {. J4 V  \5 Y) l
76.     if (opt.Command == "translate")
    
77.     {
    & W; O7 j0 \) [
78.         builder.WithTranslate();
    
79.     }
    
80. 
    ) R# {$ v* O3 J. Y$ U& y
81.     WhisperProcessor processor = builder.Build();
    ! s) o6 H7 l" C6 Z  @% }
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    
87.     string targetDirectory = Console.ReadLine();
    ; B! \" o+ t; `7 y
88. 
    
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    0 T& [3 d0 x, M& z" ~- H3 H0 y3 w1 a
90.     {
    6 }' Q7 |: w4 q  q  N
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    
92.         return;
    , I* D- d: N  l! p# J
93.     }
    4 S! _! N+ J$ Z+ H
94. 
    
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    
96. 
    ! o5 h3 X, X8 q
97.     Console.WriteLine("处理完成！");
    ! p! G" @" i' d/ @, y  c
98. 
    
99. }
    
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     
101.     string sourceDirectory, string targetDirectory)
     
102. {
     - @% E, f3 c, [
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     7 T7 J- z" C/ T; x2 v9 i4 m, k: \
104. 
     
105.     foreach (var sourceFilePath in files)
     4 Z0 E& L6 i" O( L! [
106.     {
     1 {1 Z5 A5 w9 Y* ?
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     6 @+ m' y6 k; E, _6 ~. k
112. 
     . B# Q& P5 `9 Y+ `
113. 
     
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     + L" _8 d# k+ e$ @9 B5 n
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     1 j0 m2 M6 g# N+ N1 v
117. 
     
118.             using var fileStream = File.OpenRead(sourceFilePath);
     & R2 A" C! C* ^% P, c; I8 g9 H
119.             var segmentIndex = 1;
     ( ?7 l5 }6 i- U: U" Z- d
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     
121.             var startTime = DateTime.Now; // 记录开始时间
     8 ^. U2 ~8 y- b: H( C3 n4 [
122. 
     1 E7 L4 U& S! F: n7 T3 r" B4 o2 u
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     
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}");
     3 `# V- A& X. o5 _
127.                 Console.WriteLine(segment.Text);
     - V" v& D9 |& @. h, Z' Q7 y
128.                 Console.WriteLine();
     
129. 
     
130.                 // 将srt内容写入文件
     5 A1 n/ @- j+ Q2 ^' D
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);
     
134.                 await writer.WriteLineAsync();
     ; j  h5 F( g, J
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     
137. 
       `1 F, W" E/ n' f! K& F8 M
138.                 segmentIndex++;
     ) h' X( j" L9 F* ~
139.             }
     
140. 
     * U0 Z0 }! ^: ~! [% h) [
141.             var endTime = DateTime.Now; // 记录结束时间
     
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     # z+ O  q7 b, g2 [+ e5 B; I
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     . _1 ?% B+ m& t3 `* h
145.         }
     4 Y# W8 l( s8 b
146.         else {
     
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     
148.         }
     - X  o) ?3 l* c4 `* d2 q
149.     }
     - t1 l% ~2 i) v/ `& S7 `  N
150. }
     ; Q( t& `7 r. M* J# t
151. public class Options
     
152. {
     . w5 X/ b: H6 n
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     
154.     public string Command { get; set; }
     1 [! _; O% N$ P' X, B/ w7 r3 G
155. 
     
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     0 b/ k- F$ S# R
157.     public string FileName { get; set; }
     
158. 
     
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     6 z) t+ M7 P0 K( V' j8 I8 ~
160.     public string Language { get; set; }
     " ~$ Y0 W3 ]; c2 p% U
161. 
     
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     
163.     public string ModelName { get; set; }
     
164. 
     
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
       w. R# k* `9 T4 }2 T1 K
166.     public GgmlType ModelType { get; set; }
     7 [" O4 E( t: {$ h
167. }
     , W- U. `# x$ Q7 Q3 }+ ?9 P

复制代码

+ u( U( E, j6 N1 I, x' A- l