冒险解谜游戏中文网 ChinaAVG

标题: 【汉化工具系列 #2】指定wave格式转换为srt格式字幕（CPU版本） [打印本页]

---

作者: shane007    时间: 2023-9-4 11:03
标题: 【汉化工具系列 #2】指定wave格式转换为srt格式字幕（CPU版本）

2 m9 n5 s% B5 x
9 B. J# n% ?' h/ T% R6 z, r  B! M8 b 本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件

代码如下
. z: u9 y, G7 c% j以下这句用多线程可以增速，否则很慢
* i4 v: S0 Z7 S9 N) Y9 P- h以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。

1. var builder = factory.CreateBuilder()
   : v) R* c" T4 ]0 o
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   2 E- ~& U7 D( X1 T% d' j9 K0 x# E
3. using System;
     ]7 h; m/ E4 b( D5 ~
4. using System.Diagnostics;
   / ~9 O" i/ D- h  Y1 d
5. using System.IO;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   - ~1 }; J# ^; T  I& k+ F- B
8. using CommandLine;
   ; w% n% U7 d8 ~# o( Y* e
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    
11. using Whisper.net.Wave;
    9 p  l! \  ~. _
12. 
    
13. await Parser.Default.ParseArguments<Options>(args)
    5 P+ C# @9 ~* t- w' i
14.     .WithParsedAsync(Demo);
    
15. 
    
16. async Task Demo(Options opt)
    0 K( B6 L; P4 l6 H; }
17. 
    , O6 f# o3 z9 T* Q
18. {
    * F, g" J+ K% h3 G
19.     if (!File.Exists(opt.ModelName))
      m* f+ a& K9 g2 L2 ^3 l# j1 i
20.     {
    2 ~. ~" C" N1 V8 X
21.         Console.WriteLine([        DISCUZ_CODE_1        ]quot;Downloading Model {opt.ModelName}");
    
22.         using var modelStream = await WhisperGgmlDownloader.GetGgmlModelAsync(opt.ModelType);
    ) Q$ s# z. I* x3 G) z, g
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    
24.         await modelStream.CopyToAsync(fileWriter);
    
25.     }
    
26. 
    
27.     switch (opt.Command)
    
28.     {
    : {" ?9 m2 F$ [, K3 x# m( p
29.         case "lang-detect":
    
30.             LanguageIdentification(opt);
    
31.             break;
    
32.         case "transcribe":
    - ^3 O4 T" Y- X* N& v( W- z, }1 n0 n
33.         case "translate":
    
34.             await FullDetection(opt);
    
35.             break;
    8 B" t2 l# X% h- A0 W0 V
36.         default:
    
37.             Console.WriteLine("Unknown command");
      W" T8 P7 A, y& k
38.             break;
    9 p1 C( O7 H6 R6 v
39.     }
    # r! x& `% D. [+ Z3 G3 ~; @4 _
40. }
    2 T' F$ c% E, y, o0 R+ J
41. 
    8 ~* v% x! r8 {" Q/ Z
42. void LanguageIdentification(Options opt)
    
43. {
    6 [6 n/ c" L2 s+ e& ]1 y# [
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    
45. 
    
46.     // Same factory can be used by multiple task to create processors.
    + Q+ d, c6 w$ h" J
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    # [! W+ N5 f4 a. I
48. 
    3 n: d5 q9 a" @4 z5 J" V
49. /*    var builder = factory.CreateBuilder()
    
50.        .WithLanguage(opt.Language);*/
    , l$ m  |7 [* g
51.     var builder = factory.CreateBuilder()
    
52.    .WithLanguage("english");
    3 |8 S1 c2 T  Y5 s, x1 h: e
53.     using var processor = builder.Build();
    / q* a+ Q* t4 _' b7 I
54. 
    3 E' k) |% a4 T7 n8 W- K- G
55.     using var fileStream = File.OpenRead(opt.FileName);
    0 s$ J& N3 l; o/ N& C
56. 
      X7 J& c6 y! ?) m) ^0 t
57.     var wave = new WaveParser(fileStream);
    
58. 
    # t, D% T& V/ S8 Y( _2 `
59.     var samples = wave.GetAvgSamples();
    
60. 
    
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    
62.     Console.WriteLine("Language is " + language);
    
63. }
    * |) M5 Z* o; i" ?
64. 
    ' t3 e0 N; h# l/ ?1 e
65. async Task FullDetection(Options opt)
    : A: ?) \7 A4 G3 N6 R3 X  Q
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    0 w6 b$ Y5 j* w# n+ O
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    
69. 
    
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    / y; u8 |: `1 Z/ }
72.     string languageOption = Console.ReadLine();
    
73.     var builder = factory.CreateBuilder()
    ( I: P" n: p8 F& Y) ]6 T
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    % Z: P4 G: w+ h/ H' s# I( `* X1 b
75. 
    / T8 B( u' Z6 c# B4 d6 k
76.     if (opt.Command == "translate")
    , Z1 H: z5 B* r5 ^! h
77.     {
    
78.         builder.WithTranslate();
    
79.     }
    " F. E& c  r' `' E1 |, ~
80. 
    
81.     WhisperProcessor processor = builder.Build();
    1 W$ v6 C# z! J& |  m6 b) e
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    
84.     string sourceDirectory = Console.ReadLine();
    " v5 [1 E8 N, Y* ?; b
85. 
    0 [0 N$ l6 I% B# ^% N' H* `
86.     Console.WriteLine("请输入目标文件目录：");
    
87.     string targetDirectory = Console.ReadLine();
    4 \# Z. i$ P+ H# K5 e
88. 
    
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    ) @& e8 g% m0 X( U9 L
90.     {
    
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    , U. K: _# Z6 d
92.         return;
    4 g& q% W1 d+ z1 Y$ M( L0 p
93.     }
    7 K4 {( m7 f8 q* T; r0 b( J- N) v
94. 
    
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    . u8 j0 d9 U( k; ~
96. 
    
97.     Console.WriteLine("处理完成！");
    / b  P7 p$ d' N6 j' M+ T' ^% j
98. 
    
99. }
    4 o9 V  n: i+ o/ e* }2 g* U
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     9 [' r3 c6 ?# P2 I! k) }
101.     string sourceDirectory, string targetDirectory)
     
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     + m0 |3 p3 U5 s2 `# X, l( @
104. 
     
105.     foreach (var sourceFilePath in files)
     
106.     {
     6 e: {( u% ^2 `: [7 e) x
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     % Q# ^2 O0 _9 Y) ]3 I
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     
112. 
     
113. 
     * u: W) x9 v9 e7 d( \4 h
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     
117. 
       x9 d- x; ]. u2 F: z5 Q# W
118.             using var fileStream = File.OpenRead(sourceFilePath);
     
119.             var segmentIndex = 1;
     - g8 \4 T) V0 F- z
120.             using var writer = new StreamWriter(destinationFilePath); // 创建用于写入srt文件的StreamWriter
     ' t2 s' B5 c8 W* }
121.             var startTime = DateTime.Now; // 记录开始时间
     ' A3 h. }6 D* @: s. A! k
122. 
     
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     & `4 X; A; g9 m, D( B+ g& i
124.             {
     
125.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     6 S# L+ _- U$ a4 |; I, ]5 {1 w
126.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     
127.                 Console.WriteLine(segment.Text);
     
128.                 Console.WriteLine();
     % A) J2 {) [) R8 i3 u6 h, B
129. 
     $ U, X( J- F; T2 c3 S
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}");
     # f1 e$ C  T# i8 V
133.                 await writer.WriteLineAsync(segment.Text);
     
134.                 await writer.WriteLineAsync();
     
135. 
     ( K5 b# f' f" B5 q3 L, x# a1 H
136.                 writer.Flush(); // 立即保存srt文件
     
137. 
     
138.                 segmentIndex++;
     2 j. s; k  ]: T: K2 v5 r4 e7 A/ U
139.             }
     1 _+ m- r: r" x3 {6 i! t: Z& b
140. 
     
141.             var endTime = DateTime.Now; // 记录结束时间
     
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     & T2 o) K) M7 u# T
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     : N( I0 _  D' [$ ^% i" U/ G5 `
145.         }
     . M& l1 I+ J" V9 _
146.         else {
     
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     ) v/ b1 W& H7 N, t% ]1 Y! N% |4 s( ]
148.         }
     - q/ Q+ c* Y6 F+ y
149.     }
     ( u2 y  f( [8 P. q, M
150. }
     . M5 n+ ^% K: z+ K9 E& U, `
151. public class Options
     
152. {
     % x. `4 s- i0 e0 q, c
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
       y% _( A$ M7 ?! E( d* ]. S) d
154.     public string Command { get; set; }
     ( J% @; U7 k) ~, N7 A$ G
155. 
     3 |4 {- r) f# h; ~( N/ G
156.     [Option('f', "file", Required = false, HelpText = "File to process", Default = "test.mp3")]
     7 w8 u" k: \6 h; M  e' N9 M' R
157.     public string FileName { get; set; }
     ' x+ e3 S1 y6 h. `, R8 d6 n
158. 
     9 f( [( Y; J1 R8 l; y" F; N$ f
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     
160.     public string Language { get; set; }
     - k5 H- I9 t! R" j
161. 
     
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     6 f1 U2 C( ^: e0 y: Z
163.     public string ModelName { get; set; }
     ! g, C1 b' i$ [1 W
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. }
     

复制代码

. J% P" Z2 S1 b$ |2 l* o7 H
  z: ]) Q  {6 l; g

---

作者: 星之韶华    时间: 2025-4-4 01:06
学习学习一下

---

欢迎光临 冒险解谜游戏中文网 ChinaAVG (https://chinaavg.com/)

Powered by Discuz! X3.2