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

shane007

本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
" E# v: I/ N  p. o2 D( G! T 使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件
2 ~3 C8 G* X7 N% L" O
" [( ?! c: \+ j) m* J( _- m代码如下
$ t; C: Y8 B5 B$ N5 F) o以下这句用多线程可以增速，否则很慢
1 x, A- r, B: w" V4 R3 e: m. g  J以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。
. ^' P  l4 @+ S! l  p0 ?
5 _* R% }8 i% {0 u0 S8 Q5 V" A

1. var builder = factory.CreateBuilder()
   + V- D/ W. Z% q4 z+ T6 x% X( J
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

& B7 Y( g6 ?1 y' ~; y  ?( g7 b
/ j& E4 d" P+ x5 S0 m

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   
3. using System;
   ( r* T) N: d+ [1 l( N9 f/ I+ e
4. using System.Diagnostics;
   1 R! y+ W' V8 T5 ^$ n
5. using System.IO;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   . \% O  p: z8 Z# g% V7 f
8. using CommandLine;
   
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    7 E: G' I1 p3 J
11. using Whisper.net.Wave;
    
12. 
    # n' b4 R& F" a) W9 R& E4 B/ A
13. await Parser.Default.ParseArguments<Options>(args)
    
14.     .WithParsedAsync(Demo);
    / S8 i. A* Y6 k0 D8 z
15. 
    % p6 F! ^7 {0 K! g
16. async Task Demo(Options opt)
    
17. 
    
18. {
    
19.     if (!File.Exists(opt.ModelName))
    - }- B! v) [* K7 ^* U+ {+ Z
20.     {
    ; Z7 R' t! ], t" K
21.         Console.WriteLine([        DISCUZ_CODE_1        ]quot;Downloading Model {opt.ModelName}");
    
22.         using var modelStream = await WhisperGgmlDownloader.GetGgmlModelAsync(opt.ModelType);
    ) g9 K+ X* J: e# A. |
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    
24.         await modelStream.CopyToAsync(fileWriter);
    - f9 I% r# Q) u3 n& m7 \
25.     }
    - n7 M! l" K+ n5 f; z% n
26. 
    
27.     switch (opt.Command)
    # v4 z+ r+ z: @' ]  F
28.     {
    
29.         case "lang-detect":
    
30.             LanguageIdentification(opt);
    
31.             break;
    
32.         case "transcribe":
    ! D( Y, v- }+ I8 v) K) s- D
33.         case "translate":
    
34.             await FullDetection(opt);
    
35.             break;
    
36.         default:
    8 ?# o. V# m3 J: E2 d" h; u
37.             Console.WriteLine("Unknown command");
    + ~3 V2 H5 [: U% S; X, Q
38.             break;
    $ x) J5 n( G" s- J! O$ r
39.     }
    . k! ^) D; Z# _, ?0 u& N0 M0 D
40. }
    3 U9 C8 Z+ J& n4 I# C
41. 
    0 z- z% [1 P4 G) a0 y
42. void LanguageIdentification(Options opt)
    + v( L' B/ U5 B6 ^2 w# e- I; ~
43. {
    # m9 m6 f& S% V" b0 ^/ y2 S# \2 o
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    4 Q7 f; O% J- E. r& O, N
45. 
    5 f" \0 m# f( y8 k6 ~8 h( V( u6 i
46.     // Same factory can be used by multiple task to create processors.
    
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    
48. 
    
49. /*    var builder = factory.CreateBuilder()
    8 c" M, _" I9 Q7 J* `* ]" R
50.        .WithLanguage(opt.Language);*/
    ; x1 d1 V$ w% B5 Q' L3 ?) x
51.     var builder = factory.CreateBuilder()
    
52.    .WithLanguage("english");
    ( C$ ]% V; `/ d0 ]" C& P: `
53.     using var processor = builder.Build();
    9 k1 L# W3 l* r6 M' K% k
54. 
    # k( `& R- b9 Y* g: K# k
55.     using var fileStream = File.OpenRead(opt.FileName);
    ' a3 b) a  j3 \/ [; x
56. 
    
57.     var wave = new WaveParser(fileStream);
    / d0 L- ^% Z  _! O; I5 O$ t5 Z
58. 
    1 |8 ^0 \# l# v5 W4 l$ m8 d7 F+ z
59.     var samples = wave.GetAvgSamples();
    
60. 
    " N7 D0 t( G1 ]! ^/ z. T6 E
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    
62.     Console.WriteLine("Language is " + language);
    , t+ Z+ q  f" h( E- e% @
63. }
    
64. 
    
65. async Task FullDetection(Options opt)
    % b" a1 k) d* S* a* I7 x
66. {
    / X, d9 W& y: z' a2 \# }
67.     // Same factory can be used by multiple task to create processors.
    5 D7 i5 t8 U. U8 J8 w9 W4 l$ K
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    % U6 J  i4 \; q7 Z
69. 
    
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    / Q& w5 _: V: f9 C) h4 H/ b
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    / C* f* K* w2 U8 k9 @  Y
72.     string languageOption = Console.ReadLine();
    
73.     var builder = factory.CreateBuilder()
    : `0 Z% ^: G6 Y( W) |
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    
75. 
    ( d2 f7 p# x% M+ Q
76.     if (opt.Command == "translate")
    
77.     {
    
78.         builder.WithTranslate();
    5 V0 e" _" x6 u2 H
79.     }
    2 m$ u4 N( c7 n7 r$ b( o4 Y& v
80. 
    7 i. g' o+ N9 A$ ~
81.     WhisperProcessor processor = builder.Build();
    
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    
84.     string sourceDirectory = Console.ReadLine();
    9 ~9 a  c6 j7 N* \" U+ d
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    5 Q* Q3 U1 q" k7 N3 H
87.     string targetDirectory = Console.ReadLine();
    
88. 
    5 [- Z1 o7 v) {( Q- w; z" h$ S, V
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    
90.     {
    7 t1 [0 v% `) ?* J4 ?. U, h
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    
92.         return;
    * L) w: J% q7 y2 M8 ]# _
93.     }
      d2 a  B) ?2 e
94. 
    
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    
96. 
    
97.     Console.WriteLine("处理完成！");
    0 {' u( r+ D, k' h1 O- ^
98. 
    ! O/ m7 Y4 n) E1 P8 A8 d1 u7 H
99. }
    ! j6 _/ n- ?/ H" C
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     , y# K# X3 k6 q2 C6 N0 E
101.     string sourceDirectory, string targetDirectory)
     3 G  j- b0 ?; e
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     
104. 
     % u: |, E" ^7 ]; K+ c
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");
       t1 l; r. f/ X1 u
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     
112. 
     
113. 
     2 A- E8 l& r5 |: I- O
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     ! e( B+ V8 I0 B3 s
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     
117. 
     ' _: J4 ]2 m/ L1 y
118.             using var fileStream = File.OpenRead(sourceFilePath);
     
119.             var segmentIndex = 1;
     
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     " x. I( _* B9 A. o& o
121.             var startTime = DateTime.Now; // 记录开始时间
     
122. 
     " r! J  I! d0 h) B" C* ^* y0 j
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     + ^3 W3 X: P- k( B5 h& q3 f8 M
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}");
     , ?) Q. }, o& L. n. D2 F
127.                 Console.WriteLine(segment.Text);
     
128.                 Console.WriteLine();
     
129. 
     9 M# ~( y/ d* ~8 f: L6 t& t
130.                 // 将srt内容写入文件
       s' N2 l( Z/ S) a
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}");
     + i7 k: q1 i1 k% f4 [: K
133.                 await writer.WriteLineAsync(segment.Text);
     - B5 d  U6 G- \. t! L% [+ w
134.                 await writer.WriteLineAsync();
     
135. 
     
136.                 writer.Flush(); // 立即保存srt文件
     & p, f$ d2 }& U7 c( T- ?4 s5 w& d
137. 
     2 T+ x( o  w, \
138.                 segmentIndex++;
       S0 v; @& n' x. Z* d
139.             }
     
140. 
     9 b7 w8 r5 D: ~9 k! t
141.             var endTime = DateTime.Now; // 记录结束时间
     ; K6 g* o- v& I
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     ' I1 d: o2 S! N5 y( o
145.         }
     * _" M3 [1 d' n3 Q% G: Q
146.         else {
     / `7 t  o; v1 T+ O
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     8 P& Q+ U2 e  m
148.         }
     
149.     }
     % N# i, X$ T/ M1 s
150. }
     
151. public class Options
     ' j8 z9 J7 }1 ]3 N
152. {
     + h* }/ U9 m7 m8 I9 Z+ H
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")]
     
157.     public string FileName { get; set; }
     
158. 
     % s$ A, I' e) g5 }2 m
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     
160.     public string Language { get; set; }
     ; H8 S0 x$ H6 W2 Q1 m7 f
161. 
     
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     ) Z# s% m+ X% }5 d9 @
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)]
     3 l8 K+ o" m( S! @8 X
166.     public GgmlType ModelType { get; set; }
     # n- c8 x% h9 l: U- i8 C3 V* Q
167. }
     

复制代码

; K' ^0 B7 h9 x$ e4 ?. O# L2 Q
2 q! k: c+ t5 s