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

shane007

* b/ ~3 N) b5 Z& _" s7 Z
# x8 A# o# u- x  H: t& c; c8 \ 本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
4 _2 O  q* {# L$ }4 Q 使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件
* I+ G5 W, Y# g$ z5 Z% e$ d
! ^0 L, \5 \" q, d! ]" P代码如下
以下这句用多线程可以增速，否则很慢
  n3 E" A0 c; c  o8 i4 y1 u  F. f以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。
( [$ v: B3 [6 O. l/ N6 `5 F3 T6 w# N1 F

1. var builder = factory.CreateBuilder()
   9 P% X: j- `) T! [
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

) W# |# O+ O" j1 X: u

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   7 d1 s' l" ]* X9 k' |
2. 
   3 O9 O+ Z) B& N
3. using System;
   1 ~3 Q' u+ ?2 ?  Z3 l
4. using System.Diagnostics;
   
5. using System.IO;
   ! ^2 Y$ o! o+ b5 w0 t7 y. f
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. 
    
13. await Parser.Default.ParseArguments<Options>(args)
    
14.     .WithParsedAsync(Demo);
    0 Y. k3 L+ a% D$ t7 n( A
15. 
    ) B  R; L. |  C. `$ A7 c8 L8 C" m
16. async Task Demo(Options opt)
    9 V, [0 ^  P8 b! X
17. 
    " Q. p' D5 M' v- J
18. {
    7 s+ Y/ J3 ]+ ]+ d2 z
19.     if (!File.Exists(opt.ModelName))
    ' S1 Q4 G3 k7 P. R: l) O
20.     {
    - b& D& G  Z7 t0 Y- z
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);
    
24.         await modelStream.CopyToAsync(fileWriter);
    
25.     }
    7 q" m/ _, ^2 n: ^
26. 
    " v2 m# ]1 }' Z
27.     switch (opt.Command)
    ' @7 W+ F$ x' A8 G6 s
28.     {
    
29.         case "lang-detect":
    
30.             LanguageIdentification(opt);
    
31.             break;
    1 g( B- }3 S% }3 c& X# F/ [; }% p
32.         case "transcribe":
    2 f+ D' U; K! P1 U3 ]4 R
33.         case "translate":
    ' @  n/ V2 E7 L! s$ |3 I
34.             await FullDetection(opt);
    $ J! o% G8 i8 D+ E: ^3 s
35.             break;
    1 p; z- U! h  I) X1 V" w# r; `
36.         default:
    
37.             Console.WriteLine("Unknown command");
    
38.             break;
    
39.     }
    
40. }
    ! w0 L$ i$ D; M  \: @
41. 
    
42. void LanguageIdentification(Options opt)
    
43. {
    $ z$ h9 x. m+ f" N
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    
45. 
    
46.     // Same factory can be used by multiple task to create processors.
    
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    9 _, |. j  M( u7 w
48. 
    2 G3 F  W+ g: {- M
49. /*    var builder = factory.CreateBuilder()
    
50.        .WithLanguage(opt.Language);*/
    
51.     var builder = factory.CreateBuilder()
    - i' w- H- `  c2 _+ X& B
52.    .WithLanguage("english");
    
53.     using var processor = builder.Build();
    , q% L: v: o7 C+ C$ d  q- A  ?( u
54. 
    ; j4 S7 R* N5 y, t- W
55.     using var fileStream = File.OpenRead(opt.FileName);
    2 k1 D& a, c. M. v& U
56. 
    
57.     var wave = new WaveParser(fileStream);
    7 k* O' J+ A  Z# Y5 g
58. 
    : G9 P& S4 T8 v
59.     var samples = wave.GetAvgSamples();
    3 m  J$ z7 e1 u; T
60. 
    
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    . Y, y6 v+ ]4 }5 u/ F
62.     Console.WriteLine("Language is " + language);
    
63. }
    ; }7 E4 m3 H8 m, p
64. 
    
65. async Task FullDetection(Options opt)
    
66. {
    . _- [+ s$ Z$ S4 g7 F+ g# }, s6 {
67.     // Same factory can be used by multiple task to create processors.
    
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    ; C6 J: ~6 p- U7 T( d8 n) k
69. 
    / \& S  H7 K2 R6 n7 }* I* N
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    
72.     string languageOption = Console.ReadLine();
    
73.     var builder = factory.CreateBuilder()
    
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    4 a& q& ^- u! G+ D6 }* V! X* E0 `
75. 
    
76.     if (opt.Command == "translate")
    
77.     {
    3 T9 e# V3 n2 a# U
78.         builder.WithTranslate();
    / Q0 ?( @- j1 t( B
79.     }
    6 ^6 ~! y( h5 W
80. 
    . V. M! T/ V- H+ o( s
81.     WhisperProcessor processor = builder.Build();
    ( r& H# w: I8 C, z. L
82.    
    6 y' b8 u" }" k7 p' O1 Q
83.     Console.WriteLine("请输入wave源文件目录：");
    
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    
87.     string targetDirectory = Console.ReadLine();
      `: o4 ]) [; U# F8 b
88. 
    5 M# l6 `5 N- o3 N3 W" `
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    : z: M+ S) N. `- e3 X" D
90.     {
    
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    , r" u9 g' E/ z& {2 G# _
92.         return;
    ) x/ k! I! V. V: w* ~
93.     }
    
94. 
    0 w. ?! ]3 {/ j$ y8 z9 g3 B
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    
96. 
    7 L5 x- r' s7 w6 Y
97.     Console.WriteLine("处理完成！");
    
98. 
    
99. }
    
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     " P/ @8 t$ E; a! D, ^* d: g
101.     string sourceDirectory, string targetDirectory)
     $ _0 l; a, F0 X) N: ?
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     
104. 
     5 y( Y/ \% w7 p% ^! @
105.     foreach (var sourceFilePath in files)
     
106.     {
     ! @, m* j' N$ E/ S+ H
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     - D4 ?: N" U" u
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     
112. 
     
113. 
     , s# u" f4 ~2 o% z$ ]
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     
117. 
     6 c+ H$ ?7 X! a' r& x
118.             using var fileStream = File.OpenRead(sourceFilePath);
     
119.             var segmentIndex = 1;
     
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     
121.             var startTime = DateTime.Now; // 记录开始时间
     
122. 
     
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     ' h" V2 B! g: g) [
124.             {
     ' `% l8 z& O( n- n0 O
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}");
     * x4 P/ N$ K* B
127.                 Console.WriteLine(segment.Text);
     # h1 l7 h0 a* n$ g# X) T# C3 f
128.                 Console.WriteLine();
     ( y% e' ~; i) s
129. 
     
130.                 // 将srt内容写入文件
     - W/ a* S% t- E6 ~
131.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     & @9 d4 I. j; r
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文件
     
137. 
     
138.                 segmentIndex++;
     & r8 E- s( N% |* C, x3 W
139.             }
     
140. 
     
141.             var endTime = DateTime.Now; // 记录结束时间
     
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     7 [, z; l, s9 y; S3 u* }
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     
145.         }
     3 l4 M8 J2 }+ i) T
146.         else {
     8 k% z3 U8 U9 h$ \
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     
148.         }
     ' ~0 s; V. W8 s& Z; R
149.     }
     : l* _$ {9 W: P6 i2 B
150. }
     
151. public class Options
     
152. {
     ! g# o* Y" p4 Q7 m2 [( r+ `
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     
154.     public string Command { get; set; }
       @" p( {8 s- g4 u+ V& V2 H3 c
155. 
     
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     
157.     public string FileName { get; set; }
     
158. 
       R7 U' B! A- k1 G& i
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     3 g; v( b; ~& ^  W
160.     public string Language { get; set; }
     2 y; z4 U% `3 O. G9 q1 {* U- J5 l
161. 
     ' T# p+ F+ J$ O2 |1 e# V6 K' Y
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     
163.     public string ModelName { get; set; }
     4 S6 \5 J8 N3 Q4 i
164. 
     1 W* k' m) @/ u# p5 N
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
     ( b) f2 N2 L2 C4 H  v
166.     public GgmlType ModelType { get; set; }
     / `1 ?7 j* Z& `0 Q, q' Y% A4 Z
167. }
     : Y: O. E, M7 a* M% N/ p

复制代码

- e# R) O' w/ [: L

6 {( {( f' \9 T9 ]8 b4 n  y- [4 y