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

shane007

8 K' P. \) k, u9 |& P8 ?
5 F8 h) J* b/ |1 o! J" e 本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
; y( a6 u$ @  n5 b$ [( B$ v 使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件

, v$ k  R) v- _- J代码如下
以下这句用多线程可以增速，否则很慢
2 |1 S4 I$ @0 `0 C以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。

% V2 ^: `, g; I. q6 n+ P
$ n/ x! k" Q3 l% V* P! q- }

1. var builder = factory.CreateBuilder()
   / \$ I% s% R0 B& i3 q+ ?8 Z
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   1 j2 g$ C6 E# P+ `2 r1 G
2. 
   4 l0 U1 K+ d7 M0 n( I. P* N+ k+ r
3. using System;
   
4. using System.Diagnostics;
   
5. using System.IO;
   9 _7 J6 c  L+ w7 q
6. using System.Threading;
   
7. using System.Threading.Tasks;
   ( g6 }5 d0 e& K
8. using CommandLine;
   ) e% ]9 }/ F5 x
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    
11. using Whisper.net.Wave;
    9 R: {4 u. O) b) ?% V0 o# v
12. 
    ) _. T$ ]% N! E- a6 m$ k  o
13. await Parser.Default.ParseArguments<Options>(args)
    
14.     .WithParsedAsync(Demo);
    / T2 |* q: |: c6 N0 ~6 b" Y$ W  ^& a
15. 
    # L4 e+ f/ ]% t/ m2 ~' @( j: |( {
16. async Task Demo(Options opt)
    ' H# b. g) ^$ Y, F8 Y
17. 
    
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);
    # W3 i" |2 O$ h
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    9 ^( Y6 m* y2 Z& Z" i7 B" [
24.         await modelStream.CopyToAsync(fileWriter);
      a4 ]( z2 N3 I1 v
25.     }
    ( @) W! S2 g- I# e+ F! ~, C$ C5 W! s  h
26. 
    
27.     switch (opt.Command)
    
28.     {
    8 [% u/ M: W5 D( D  a% o# s2 Y' W
29.         case "lang-detect":
    6 w: d% O/ r3 @% W$ C$ \5 j
30.             LanguageIdentification(opt);
    
31.             break;
      I# m; M1 j5 I# k
32.         case "transcribe":
    2 }2 \2 }5 G1 X6 k* Q
33.         case "translate":
    : C6 o$ ~" P: ]2 P
34.             await FullDetection(opt);
    $ M/ A9 F, T- G, r
35.             break;
    
36.         default:
    
37.             Console.WriteLine("Unknown command");
    $ q! R; z2 w2 C/ v/ ~; t% n
38.             break;
    
39.     }
    ) M2 t7 x4 O8 M2 J- w  z
40. }
    
41. 
    
42. void LanguageIdentification(Options opt)
    
43. {
    6 i' V& r) _0 E9 \& U
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    " X! I( ?! E9 W& U( n6 V
45. 
      C: c% l" h1 A( o: H* j
46.     // Same factory can be used by multiple task to create processors.
    6 }$ A1 e4 Q$ e8 F0 U; T
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    ' Q+ Z* Q# d, A$ _! M; J; E& ~
48. 
    / Z, g7 J, d$ o0 u
49. /*    var builder = factory.CreateBuilder()
    - L* u/ ^- `9 H' R( g) P$ t" l
50.        .WithLanguage(opt.Language);*/
    
51.     var builder = factory.CreateBuilder()
    / y4 o+ n; r4 H9 T- U
52.    .WithLanguage("english");
    
53.     using var processor = builder.Build();
    ! s0 e( p  S9 g- T
54. 
    
55.     using var fileStream = File.OpenRead(opt.FileName);
    1 w6 U" q/ k8 m1 f$ V
56. 
    8 P; Z  i9 R2 g) a  Z
57.     var wave = new WaveParser(fileStream);
    ! y, n* b( p5 ?0 x9 @6 O! A1 ]
58. 
    3 H" d& p  V+ W  j& n
59.     var samples = wave.GetAvgSamples();
    
60. 
    2 N) U, s: l* Y7 o- b
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    , o* v4 \) S& F7 |. v; F6 q- ^
62.     Console.WriteLine("Language is " + language);
    
63. }
    * Y2 ~' s! \4 g7 v
64. 
    ! O0 N" \- P0 S
65. async Task FullDetection(Options opt)
    
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    * y8 U8 @' s& i' E' C6 a
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    # S9 W/ z* B- ?) a
69. 
    , P' ^' ~) A, C% Y6 Y% ~8 O. u$ k
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    
72.     string languageOption = Console.ReadLine();
    . X- d3 |0 G+ r3 ^- `
73.     var builder = factory.CreateBuilder()
    & D5 N/ g( V5 X: b7 [% L, |
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    
75. 
    
76.     if (opt.Command == "translate")
    6 Y% t' ^& ^' [! y6 g! _
77.     {
    9 N' M% I+ U' x4 t% t
78.         builder.WithTranslate();
    1 b. I8 A: N  j/ T8 ~
79.     }
    
80. 
    
81.     WhisperProcessor processor = builder.Build();
    2 K3 p$ e  J4 A! ^
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    
84.     string sourceDirectory = Console.ReadLine();
    ( J) x& Q5 M, {- W
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    ) z$ b) F, a( [* H7 g1 N. m" t. J
87.     string targetDirectory = Console.ReadLine();
    
88. 
    
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    $ V7 p! v$ |/ D# r+ D/ v7 c
90.     {
    
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    
92.         return;
    
93.     }
    
94. 
    2 g3 g7 N  |0 m" A& a
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    / ^6 ^  ?3 {- X/ I4 i& O
96. 
    
97.     Console.WriteLine("处理完成！");
    ; C8 _  L" u" y; v; ]6 Y
98. 
    
99. }
    
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     
101.     string sourceDirectory, string targetDirectory)
     
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     6 S! {# R( {/ U
104. 
     , y8 m7 C: H0 J. g2 |
105.     foreach (var sourceFilePath in files)
     
106.     {
     ; |& o5 d: Y2 M
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     * F/ ^  a3 f) c% _2 y, y
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     - d1 G: p, v& S" l+ G  n5 P7 R
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     1 x! e0 q* E$ A! [
112. 
     , B( ^5 G# s* R/ d- q! C: G3 Y
113. 
     
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     . K- B% k1 X5 M" I/ w  e# j3 h
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     " t: n. w# N1 C! H
117. 
     
118.             using var fileStream = File.OpenRead(sourceFilePath);
     ' }8 Y8 q8 w) I( N2 p" d9 ?
119.             var segmentIndex = 1;
     
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     
121.             var startTime = DateTime.Now; // 记录开始时间
     . C- A0 _( R  p' a5 t/ V
122. 
     
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     : d6 \  b- u  M0 u/ ?& G+ t0 M
124.             {
     - j9 L" s- x* v, V0 \  h1 w
125.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     6 m# A( j3 ^( ^' @' c& a
126.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     ) V$ o% E( r- i5 d! w8 V
127.                 Console.WriteLine(segment.Text);
     
128.                 Console.WriteLine();
     0 M0 L7 g) O8 U
129. 
     1 y* ]- ^, s: _7 D# [
130.                 // 将srt内容写入文件
     ) y1 P* Q& D+ R1 U
131.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     7 f; @- r& W1 d' ]8 W+ q
132.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     & a0 k8 _; l6 V9 N. k, f- o6 g
133.                 await writer.WriteLineAsync(segment.Text);
     
134.                 await writer.WriteLineAsync();
     
135. 
     ' ~  j  `  P' o5 d( i
136.                 writer.Flush(); // 立即保存srt文件
     ( W% \8 Y1 ]3 J. G' L7 @% M
137. 
     ) Q2 b; q; X- B( }9 K, q& O8 V6 S
138.                 segmentIndex++;
     4 l0 X" |# j$ G( ?# A3 ^
139.             }
     
140. 
     ; ^! e5 X5 C( M0 v1 r' S- P
141.             var endTime = DateTime.Now; // 记录结束时间
     
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     * a9 S" G5 z, t" F. S0 H9 T5 c
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     - w% c, `- _9 ^. r4 ?
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     % K, I# f9 I3 p
145.         }
     
146.         else {
     
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     ! u9 i, |, X# h) q9 O  r, U
148.         }
     - H& E# T/ [6 T7 c8 Q/ _" ^; Z- C
149.     }
     
150. }
     
151. public class Options
     9 X+ }* V: Y, v/ @2 e/ c* \
152. {
     
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     
154.     public string Command { get; set; }
     
155. 
     
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     $ r, E: B: t  B& f, x1 w
157.     public string FileName { get; set; }
     * u% t* ~, f8 c9 N, d2 B
158. 
     0 s+ n4 B) I) O7 S' p2 D
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     
160.     public string Language { get; set; }
     
161. 
     2 o# s+ _  V' q
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)]
     . O! C% e4 C. H! @* J5 H8 }/ O. D
166.     public GgmlType ModelType { get; set; }
     
167. }
     , {$ `6 ]: s- b3 o5 F1 i

复制代码