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

shane007

- e) {9 l8 |- Q0 C& a$ G
, Y8 R+ r- U" I5 {2 h2 Z% _5 ] 本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
使用时需要輸入语言，源文件路径和目标文件路径
' @' D% W' F0 |6 ?2 k  |最后是输出srt文件
0 y! d! s& E- l! Z( ?% j
# |& O1 F( a) o+ i+ F. j代码如下
6 z) J# v7 Y3 w2 \2 z/ B% ~以下这句用多线程可以增速，否则很慢
4 D& m' n8 L; R- T以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。

) l& n( y0 q5 r& C
. u$ z. }8 x  T% o6 m% Y8 ]" o

1. var builder = factory.CreateBuilder()
   / k" S6 v3 P" X; Y: u! @5 o
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

6 J) x5 f- M, I6 ?+ b
" @2 s9 \7 K, ~4 J$ E% g

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   ) r* X$ g/ M5 x/ V
2. 
   3 ^# s0 S8 b: w. G* C
3. using System;
   ! G3 L% U4 ?+ w: T# J' U& @; }2 L
4. using System.Diagnostics;
   / Z+ M- _: [2 E' H* x4 w% g
5. using System.IO;
   - U& k* e+ n# h* Q" X! [
6. using System.Threading;
   
7. using System.Threading.Tasks;
   0 \+ y$ }" k3 x; `% g- u
8. using CommandLine;
   ; m7 ^9 D* U6 ~* |9 L
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    
11. using Whisper.net.Wave;
    7 q5 i! S3 o* f- ?7 p# a' t
12. 
    ) e3 {$ a6 ]: Z- b; `; [4 m
13. await Parser.Default.ParseArguments<Options>(args)
    & w2 O- d6 @, y. X5 t
14.     .WithParsedAsync(Demo);
    " n$ [- B/ {7 X3 U1 R/ Y9 p
15. 
    
16. async Task Demo(Options opt)
    
17. 
    6 ?1 }4 Y4 w2 \7 p
18. {
      E" e& L! N9 }  p/ C
19.     if (!File.Exists(opt.ModelName))
    2 U% Z; ^  z/ k# I+ I% D: x
20.     {
    
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);
    $ i' p7 q% b( y4 u+ Q
24.         await modelStream.CopyToAsync(fileWriter);
    " F2 C/ w( I1 }! \1 o: D7 E
25.     }
    
26. 
    1 `7 v* c9 N. E0 |8 D
27.     switch (opt.Command)
    : _8 X% U7 h5 Z. J9 _& k
28.     {
    
29.         case "lang-detect":
    ! Z6 R' k. _9 T" i7 ^$ y- D
30.             LanguageIdentification(opt);
    / I+ d  \) K5 y
31.             break;
    ( Q! Z* n# u8 V0 ~2 ~! ~
32.         case "transcribe":
    ; j  Q! X+ O" F7 N9 |/ ^% f- R
33.         case "translate":
    
34.             await FullDetection(opt);
    
35.             break;
    
36.         default:
    . U2 O6 o/ k8 G% H* h
37.             Console.WriteLine("Unknown command");
    2 O+ u6 J  K# R. M
38.             break;
    + n# F1 e6 Q- R% v# r, W4 l! n2 K* z
39.     }
    " _/ r) m* d8 c% N+ o' k
40. }
    ( T* d0 o( g/ o. v' O
41. 
    3 E# y4 Z  t& }% b9 n* I. P' C
42. void LanguageIdentification(Options opt)
    , K3 d; \6 C$ P2 D
43. {
    
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    * G7 {4 }, g" e
45. 
    
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()
    ) Q; m. w9 I" s/ o6 R1 L
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);
    , J, O! L2 ?# |6 R0 H( V3 {0 [
56. 
    $ r% p/ d. q; `/ Q
57.     var wave = new WaveParser(fileStream);
    
58. 
    
59.     var samples = wave.GetAvgSamples();
    7 q5 M4 E1 M' }5 K
60. 
    $ V* G; f# k- f+ h/ m# o: L. K
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    $ V4 `9 k1 V/ W
62.     Console.WriteLine("Language is " + language);
    
63. }
    % L$ ?$ p: d+ s7 D0 }5 S
64. 
    
65. async Task FullDetection(Options opt)
    ' ?8 d7 a3 G+ a  K9 u9 }- j
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    % H% C$ K/ H4 t" J% f1 [# @
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    
69. 
    7 r9 u7 r& e2 F0 e' g! F
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    3 P" v6 b) V; r! T- F$ a8 E' N
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    - X- o5 @6 u) `- S% M5 x/ Y7 v
72.     string languageOption = Console.ReadLine();
    
73.     var builder = factory.CreateBuilder()
    
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    
75. 
    
76.     if (opt.Command == "translate")
    3 V+ \# B# }- K5 M+ X
77.     {
    
78.         builder.WithTranslate();
    
79.     }
    
80. 
    
81.     WhisperProcessor processor = builder.Build();
    ! [' U) v) ]* N5 _% y
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    & o" l0 N6 S) f/ U
84.     string sourceDirectory = Console.ReadLine();
    
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    * S8 b9 l$ T' f
87.     string targetDirectory = Console.ReadLine();
    
88. 
    - d4 _3 y6 \& y6 ^7 ~% Z8 h# D* l
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    
90.     {
    
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    
92.         return;
    
93.     }
    # D& S! h% V3 V9 O' V7 l
94. 
    3 }' v5 }+ p& N4 L. d
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    7 h; J% B# H7 H' ?
96. 
    
97.     Console.WriteLine("处理完成！");
    
98. 
    
99. }
    : l& V& x$ W# M( w; S6 X* z
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     3 D4 ~  m$ {1 c# Q* V; V
101.     string sourceDirectory, string targetDirectory)
     * o/ j. D: G3 |& P+ o3 r
102. {
     & m+ T& ]0 b  L# I
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     0 y1 `( J- ?2 H4 E+ _) [
104. 
     
105.     foreach (var sourceFilePath in files)
     
106.     {
     - E8 R7 d9 q& \' D
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     - ], o. R" a: W5 Z3 I' U* }
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     : }6 s# j$ ]4 Q' F/ I
110. 
     $ `! d# a8 @; i
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     + d; f7 Y, `7 Q0 ?4 _/ [
112. 
     9 ]9 ?' E% ^/ f
113. 
     
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     + w- f# H- ^" r, M: F. p3 F1 U, g; z! }
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     ' E: D+ I$ r# M& w9 \. i
117. 
     
118.             using var fileStream = File.OpenRead(sourceFilePath);
     / V  ^$ p" g5 |! s1 l# `
119.             var segmentIndex = 1;
     , i9 [5 [& R3 [0 D  s
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     
121.             var startTime = DateTime.Now; // 记录开始时间
     & b& j) l% Y2 Q  ], S4 c! p
122. 
     - F! ?) K- F9 k, |& q1 T
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}");
     
127.                 Console.WriteLine(segment.Text);
     6 P1 O/ ~/ o7 d% c
128.                 Console.WriteLine();
     
129. 
     
130.                 // 将srt内容写入文件
     
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);
     $ x5 l0 F; w# q0 V& R' r( i
134.                 await writer.WriteLineAsync();
     
135. 
     9 m3 P9 F# t$ f; e, z
136.                 writer.Flush(); // 立即保存srt文件
     2 s( p" P* H  L- D2 q2 I; Y: ?
137. 
     . S6 X. z4 |* ?# T7 \. m( W
138.                 segmentIndex++;
     $ q" x- x( e  ?. `
139.             }
     
140. 
     
141.             var endTime = DateTime.Now; // 记录结束时间
       I8 u9 ?6 F. b& o! c" @$ \
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     * E8 P! R3 s# W/ q
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     
145.         }
     " I# _! ~6 L/ g5 _: p- @: c) e3 k$ t
146.         else {
     ( o' g2 R0 v" @6 j
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
       S9 c- k4 I  k8 Q
148.         }
     4 F7 x+ \# R# h! @; _7 }" w" [
149.     }
     2 t6 n4 q+ [1 A) ~7 }) g* c3 w' ~1 T
150. }
     & I; ?" W$ J: s! k# t+ h% i
151. public class Options
     
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")]
       W9 S8 t/ n5 L1 l0 |) U: l; K
157.     public string FileName { get; set; }
     + e, q8 A8 d/ A  x9 r/ r# }
158. 
     
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     
160.     public string Language { get; set; }
     $ }5 g# {  l5 R
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)]
     
166.     public GgmlType ModelType { get; set; }
     
167. }
     ! B  ~) Z0 V- B0 `* P

复制代码

. W" i$ }5 O4 r; e9 y& G

( \: c2 [: b" G0 ]