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

shane007

) Z2 w; k8 E7 z2 Q3 c! b0 c 本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
& c& f6 C5 o* S# N 使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件
' O9 y7 ]1 J) _8 |2 w% s% v( X1 s, j
代码如下
以下这句用多线程可以增速，否则很慢
! H, Y: B- t8 Q# J7 h以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。
! f% s4 Q5 n3 G
. U" m* p! {' l( b0 D
- Q" W0 A+ X" `6 Q2 {

1. var builder = factory.CreateBuilder()
   $ u) V& X+ |+ n3 p' z
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   4 \: l2 R' o( Y7 G3 q
3. using System;
   & z8 K3 R* L: y" Q  v, h6 H" L" ]
4. using System.Diagnostics;
   
5. using System.IO;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using CommandLine;
   1 f, R6 Q# P* ]: p, C. Z- R
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    
11. using Whisper.net.Wave;
    % k( v& N. o. T8 E& M7 X0 |- j  \
12. 
    
13. await Parser.Default.ParseArguments<Options>(args)
    
14.     .WithParsedAsync(Demo);
    ' J0 M2 W' C6 o( C$ @8 y
15. 
    
16. async Task Demo(Options opt)
    / K4 i+ Y6 n3 C/ I
17. 
    
18. {
    
19.     if (!File.Exists(opt.ModelName))
    
20.     {
    2 w5 N/ S! ^+ X9 }. k" m8 R
21.         Console.WriteLine([        DISCUZ_CODE_1        ]quot;Downloading Model {opt.ModelName}");
    5 B$ \3 ]0 `( z/ P2 h, m7 q
22.         using var modelStream = await WhisperGgmlDownloader.GetGgmlModelAsync(opt.ModelType);
    
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    5 B9 {4 X' ?; R9 x5 C
24.         await modelStream.CopyToAsync(fileWriter);
    ; d: ]0 y  n( x2 U, P2 n  i
25.     }
    
26. 
    
27.     switch (opt.Command)
    
28.     {
    
29.         case "lang-detect":
    2 f6 H0 E2 i( J
30.             LanguageIdentification(opt);
    
31.             break;
    
32.         case "transcribe":
    
33.         case "translate":
    2 D3 o7 Z) X- p7 b  n+ z
34.             await FullDetection(opt);
    
35.             break;
    4 ~) ~8 c: Q; g
36.         default:
    , H- Q+ J/ a' {, {3 E6 D, ~
37.             Console.WriteLine("Unknown command");
    6 k; ~, \  S/ E, J9 L
38.             break;
    
39.     }
    
40. }
    
41. 
    
42. void LanguageIdentification(Options opt)
    
43. {
    
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    
45. 
    1 t7 z8 |! c, l  w5 D
46.     // Same factory can be used by multiple task to create processors.
    
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    + l/ X/ a( s) Z/ M* F$ W/ O
48. 
    
49. /*    var builder = factory.CreateBuilder()
    
50.        .WithLanguage(opt.Language);*/
    
51.     var builder = factory.CreateBuilder()
    
52.    .WithLanguage("english");
    
53.     using var processor = builder.Build();
    
54. 
    
55.     using var fileStream = File.OpenRead(opt.FileName);
    8 c9 Z" j; }1 R/ D8 b# j  ?
56. 
    - q6 S4 U: n6 f8 z3 P
57.     var wave = new WaveParser(fileStream);
    
58. 
    
59.     var samples = wave.GetAvgSamples();
    
60. 
    
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    1 g" Q$ W( ^& M, C/ }: i: @# G
62.     Console.WriteLine("Language is " + language);
    & X, L% y0 P- k( k( G; l
63. }
    
64. 
    
65. async Task FullDetection(Options opt)
    4 m& c# p/ M8 g/ ?% k5 d
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    $ o7 e2 _4 C. \: M5 s  ?/ j2 t
69. 
    
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    5 x* [/ `5 H5 J! E8 L" Z
72.     string languageOption = Console.ReadLine();
    
73.     var builder = factory.CreateBuilder()
    
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    - M( l" [: R  d4 P7 G
75. 
    ! \2 D5 P+ F9 r9 }" L5 S
76.     if (opt.Command == "translate")
    
77.     {
    $ Y  ~8 X/ \7 d0 N
78.         builder.WithTranslate();
    
79.     }
    6 k- r: A2 \/ P2 y
80. 
    
81.     WhisperProcessor processor = builder.Build();
    # s7 J1 `  J6 p6 L$ m: F, h
82.    
    1 A0 s7 |$ z$ N- |1 K
83.     Console.WriteLine("请输入wave源文件目录：");
    ) E; e# }' W3 G
84.     string sourceDirectory = Console.ReadLine();
    - P  {; A% A" x. i. Q, f
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    % K0 p+ D( R: C0 y( h. |! r  \
87.     string targetDirectory = Console.ReadLine();
    
88. 
    
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    & ]) _0 H0 v7 U6 G
90.     {
    
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    , o( J$ h& y0 o! S$ D- [) M5 h* w, Q
92.         return;
    
93.     }
    6 Q8 `! d7 u: P" J  m
94. 
    1 I# A" E0 L7 @0 S) r+ ]
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    
96. 
    
97.     Console.WriteLine("处理完成！");
    4 k. s* y& L6 B% z7 U, t
98. 
    
99. }
    ) J/ `, \6 C; P# C' u
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     
101.     string sourceDirectory, string targetDirectory)
     ( j% g/ S0 U5 R
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     
104. 
     
105.     foreach (var sourceFilePath in files)
     
106.     {
     
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     " x4 \. w) d7 e: M
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     + A9 ?1 c! S$ b5 F7 i( H
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     & q# ]2 d1 u7 g7 ^/ w8 q! X% q
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     
112. 
     , r* n/ m9 @+ m7 _$ h
113. 
     9 R6 H; L% I. u; ^( f8 D4 }
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     & S$ z* ^7 |& |4 i; J. ^
117. 
     
118.             using var fileStream = File.OpenRead(sourceFilePath);
     
119.             var segmentIndex = 1;
     6 n" f+ s: ~7 o7 L7 c3 f( s
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     
121.             var startTime = DateTime.Now; // 记录开始时间
     
122. 
     6 |" v  _3 t/ U: i6 r5 _* t
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     9 P" N2 G0 u3 Y- P
124.             {
     
125.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     2 u$ N  P9 C! |. X$ {% }6 P
126.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     $ [$ p  L) a$ `% B
127.                 Console.WriteLine(segment.Text);
     
128.                 Console.WriteLine();
     
129. 
     ( c& h# E$ b9 ]1 @, `+ H8 U
130.                 // 将srt内容写入文件
     
131.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     " a; F5 ?0 Z" E+ @2 E6 B; A( O" I  O
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);
     ' m( Y  y) R+ r2 J# F
134.                 await writer.WriteLineAsync();
     
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     , z. e" l. G3 T8 t2 R
137. 
     # E  h2 e7 V5 Q
138.                 segmentIndex++;
     
139.             }
     
140. 
     ' W( @5 _8 @+ j: f/ w% B
141.             var endTime = DateTime.Now; // 记录结束时间
     . f) p7 S% i* h: L- M- t+ }# s
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
       f0 g+ f! H- y8 `3 C+ i2 s
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     ; m/ Y, }4 c1 ^5 q1 h- o+ Z8 [
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     2 E6 W2 Z) |3 Q6 F
145.         }
     % ?7 |! s# o- [+ h5 f
146.         else {
     $ n  H+ }+ N( {4 g& W8 S9 x
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     4 @0 R- e6 E; u2 D
148.         }
     
149.     }
     3 V0 A. g- Q# J& T" r7 z- h
150. }
     
151. public class Options
     
152. {
     
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     - R( j6 D5 E) R/ X/ S- ^7 I' M
154.     public string Command { get; set; }
     3 ^+ e* b5 V# S7 A: @9 j
155. 
     5 b7 G3 t& w1 c
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     7 Z7 {9 M& c# t: z) J" @$ S3 r# [
157.     public string FileName { get; set; }
     # T2 U# n3 }' \: y
158. 
     
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     
160.     public string Language { get; set; }
     
161. 
       Y7 F7 x$ `# }6 }" E/ `
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     
163.     public string ModelName { get; set; }
     9 {+ n4 q) ]$ \: U
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; }
     ! e0 O. g+ i4 g
167. }
     & d3 j# v, C& e

复制代码

. D+ q  u: c5 E6 a) n; t- W
( z9 C. T0 T  i5 Q# A+ g
! a" I' F3 U1 J" R0 |+ B