冒险解谜游戏中文网 ChinaAVG

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

---

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


本程序是在Whisper.net.Demo基础上修改的,可以批量识别wav文件，
使用时需要輸入语言，源文件路径和目标文件路径
最后是输出srt文件
3 _& D% R9 _5 N$ V
代码如下
以下这句用多线程可以增速，否则很慢
以下这份代码，还只能使用到CPU，速度依旧比较慢，使用CPU的方法另行研究。
! k, ^; Y1 |/ {

1. var builder = factory.CreateBuilder()
   
2. .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);

复制代码

) f" i( f1 o4 w2 T' c0 P

1. // Licensed under the MIT license: https://opensource.org/licenses/MIT
   
2. 
   
3. using System;
   5 u% ~1 P( p% i
4. using System.Diagnostics;
   2 |5 N2 ?$ b0 T( m! t( R
5. using System.IO;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   8 K2 V6 a' W4 C; O
8. using CommandLine;
   
9. using Whisper.net;
   
10. using Whisper.net.Ggml;
    4 d# E6 n/ l6 I: T
11. using Whisper.net.Wave;
    5 [( @9 J/ F2 R) e
12. 
    
13. await Parser.Default.ParseArguments<Options>(args)
    
14.     .WithParsedAsync(Demo);
    # `! _8 ?# p( v$ h8 c
15. 
    
16. async Task Demo(Options opt)
    
17. 
    4 U2 O, F2 i8 q  Y& N3 |3 z( V
18. {
    
19.     if (!File.Exists(opt.ModelName))
    
20.     {
    
21.         Console.WriteLine([        DISCUZ_CODE_1        ]quot;Downloading Model {opt.ModelName}");
    , ~  l; K# O8 d% W1 }! m
22.         using var modelStream = await WhisperGgmlDownloader.GetGgmlModelAsync(opt.ModelType);
    
23.         using var fileWriter = File.OpenWrite(opt.ModelName);
    ' |8 g+ y3 ^. G/ M8 G
24.         await modelStream.CopyToAsync(fileWriter);
    
25.     }
    8 z$ U& t/ _% j
26. 
    8 d8 V# R1 J- _# P  C
27.     switch (opt.Command)
    ) i* H8 h' U/ q+ K9 Q( `3 F0 h
28.     {
    
29.         case "lang-detect":
    . O: ], q2 t- c; M+ R" ~3 U
30.             LanguageIdentification(opt);
    
31.             break;
    9 `- _! |: [  `
32.         case "transcribe":
    4 o" p3 S' F; |/ }
33.         case "translate":
    
34.             await FullDetection(opt);
    2 Z* v/ ?9 Z1 C- }! {
35.             break;
    
36.         default:
    * F' I1 N5 \' \1 E9 s& W! a3 b
37.             Console.WriteLine("Unknown command");
    6 A% I. i5 s1 `. p& {$ k
38.             break;
    
39.     }
    
40. }
    
41. 
    
42. void LanguageIdentification(Options opt)
    4 Z4 x3 W1 X8 x- S
43. {
    ; H; A" e8 [5 _5 R9 A0 D
44.     var bufferedModel = File.ReadAllBytes(opt.ModelName);
    9 y, T# g/ @" v3 [% T# R2 `* A+ ?
45. 
    
46.     // Same factory can be used by multiple task to create processors.
    
47.     using var factory = WhisperFactory.FromBuffer(bufferedModel);
    / U2 v6 ^+ g7 ^7 f% M/ B) E
48. 
    
49. /*    var builder = factory.CreateBuilder()
    
50.        .WithLanguage(opt.Language);*/
    
51.     var builder = factory.CreateBuilder()
    9 c1 H9 s" P+ D. h) P
52.    .WithLanguage("english");
    + K; S  H& f1 i, m" x) I
53.     using var processor = builder.Build();
    - `# q8 d! J4 v2 q
54. 
    ; z4 E5 L  ?- V& A6 {0 s" G& t
55.     using var fileStream = File.OpenRead(opt.FileName);
    
56. 
    
57.     var wave = new WaveParser(fileStream);
    
58. 
    ' s# S+ P+ [- Z& ~8 Y7 B( c) H
59.     var samples = wave.GetAvgSamples();
    ' O7 U4 |( s+ h5 h8 N
60. 
    2 P" f4 r6 Q* ?- u0 P9 I
61.     var language = processor.DetectLanguage(samples, speedUp: true);
    1 n  n: |8 Z* z8 }: u+ Y+ j
62.     Console.WriteLine("Language is " + language);
    
63. }
    
64. 
    0 o7 P1 s) y/ s8 I+ W5 }+ x
65. async Task FullDetection(Options opt)
    / K  T3 M/ e/ J$ j: ?
66. {
    
67.     // Same factory can be used by multiple task to create processors.
    5 r- E; M1 G6 }/ ~3 i7 h: ~
68.     using var factory = WhisperFactory.FromPath(opt.ModelName);
    " O9 X& l; N; f7 x1 c
69. 
    ; y7 @" w( G  F  p2 s
70.     //   var builder = factory.CreateBuilder().WithLanguage(opt.Language);
    $ F( m8 V, o, I. y% m; T$ l
71.     Console.WriteLine("请输入语言选项（例如：english，chinese, japanese等）：");
    % L' n- N: U2 p8 t3 M  K
72.     string languageOption = Console.ReadLine();
      ?/ b! `& n4 f+ S$ g$ Z  y
73.     var builder = factory.CreateBuilder()
    
74.     .WithLanguage(languageOption).WithSpeedUp2x().WithThreads(16);
    ' d; _7 Y' K# |3 }  i
75. 
    
76.     if (opt.Command == "translate")
    
77.     {
    
78.         builder.WithTranslate();
    
79.     }
    9 n) e3 f1 t3 m; g  {; k/ W9 U6 t
80. 
    
81.     WhisperProcessor processor = builder.Build();
    ' }6 r% x% n+ ~$ _' ]; b5 z
82.    
    
83.     Console.WriteLine("请输入wave源文件目录：");
    $ [$ V" `8 o( {, i' l  K, K
84.     string sourceDirectory = Console.ReadLine();
    : K/ O* q7 d8 D) `7 k0 _+ d  g
85. 
    
86.     Console.WriteLine("请输入目标文件目录：");
    
87.     string targetDirectory = Console.ReadLine();
    $ G8 v* M* W% l, y% Q% e
88. 
    ; r5 d) `/ K# p
89.     if (!Directory.Exists(sourceDirectory) || !Directory.Exists(targetDirectory))
    . P2 p5 z+ d0 m+ c1 t! E  P
90.     {
    2 J" f; O( _4 z: q
91.         Console.WriteLine("目录不存在，请检查输入的目录路径。");
    
92.         return;
    ; r: v  N  O6 W$ }
93.     }
    0 f( Z+ |: o- b& N* s
94. 
      i8 V6 q8 g7 H2 N" b; E
95.     await ProcessFilesAsync(processor ,sourceDirectory, targetDirectory);
    
96. 
    
97.     Console.WriteLine("处理完成！");
    
98. 
    
99. }
    ' `7 x- R) x8 E: n
100. static async Task ProcessFilesAsync(WhisperProcessor processor,
     9 T" W2 L& u4 B- @; Z
101.     string sourceDirectory, string targetDirectory)
     
102. {
     
103.     var files = Directory.GetFiles(sourceDirectory, "*.wav", SearchOption.AllDirectories);
     
104. 
     % |$ U. v' u5 O
105.     foreach (var sourceFilePath in files)
     9 N3 m* P* Y& C4 o* y7 P
106.     {
     / y$ @% }' F* U0 S4 T" h
107.         string relativePath = Path.GetRelativePath(sourceDirectory, sourceFilePath);
     
108.         string destinationFilePath = Path.Combine(targetDirectory, relativePath);
     
109.         destinationFilePath = Path.ChangeExtension(destinationFilePath, ".srt");
     & @5 G# e! F3 T+ P, j+ j6 _# G
110. 
     
111.         Directory.CreateDirectory(Path.GetDirectoryName(destinationFilePath));
     - t$ y; e# O& {- h
112. 
     2 h% ]$ h3 ~, i8 ~  S
113. 
     * T* ^  t) a' ?: q4 {! m' L
114.         if (!File.Exists(destinationFilePath))
     
115.         {
     8 m5 ?0 i$ Z2 C. w% p* K
116.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;正在处理文件：{sourceFilePath}");
     8 S2 I. j5 m, X. Z( |+ ~4 t
117. 
     7 F6 v1 a  |) `9 j* F
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; // 记录开始时间
     & Q" h# w9 g9 ~5 j
122. 
     1 |4 ~+ C5 L. L
123.             await foreach (var segment in processor.ProcessAsync(fileStream, CancellationToken.None))
     
124.             {
     6 M$ J* q0 _; Z5 i+ e9 R/ l
125.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
     $ V0 B- `5 m* w, Q
126.                 Console.WriteLine([        DISCUZ_CODE_1        ]quot;{segment.Start:hh\\:mm\\:ss\\,fff} --> {segment.End:hh\\:mm\\:ss\\,fff}");
     3 {  T+ f% ~* \! e9 @- k( T
127.                 Console.WriteLine(segment.Text);
     
128.                 Console.WriteLine();
     * Q. e- R* s! [5 J" F
129. 
     ) Q; G  h3 U) `
130.                 // 将srt内容写入文件
     
131.                 await writer.WriteLineAsync([        DISCUZ_CODE_1        ]quot;{segmentIndex}");
       O9 l* J9 n$ z# q0 s! p1 b& @
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);
     6 D" _% n; p  r- E5 d3 s' W7 K) Q! v
134.                 await writer.WriteLineAsync();
     3 U" ?1 L- r6 a: k3 u) n
135. 
     ! x5 u" s4 F6 k6 @$ O
136.                 writer.Flush(); // 立即保存srt文件
     
137. 
     " C& Z; r) |( y1 Z- A- j& e9 |
138.                 segmentIndex++;
     
139.             }
     
140. 
       ]) O4 b  Z: t7 G  R
141.             var endTime = DateTime.Now; // 记录结束时间
     
142.             var elapsedMinutes = (endTime - startTime).TotalMinutes;
     + @/ h  k1 D9 M0 Y% g
143.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;已生成srt文件：{destinationFilePath}");
     ; P+ K) z; a+ T* y* l$ v- k3 x
144.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;生成耗费时间：{elapsedMinutes} 分钟");
     , ?) _4 y- h9 M5 D2 V: u6 x
145.         }
     / P9 Y# @* b$ h1 c7 D# J5 M
146.         else {
     
147.             Console.WriteLine([        DISCUZ_CODE_1        ]quot;srt文件已经存在：{destinationFilePath}");
     , t. U& \5 W. \- p
148.         }
     
149.     }
     
150. }
     
151. public class Options
     
152. {
     
153.     [Option('t', "command", Required = false, HelpText = "Command to run (lang-detect, transcribe or translate)", Default = "transcribe")]
     + \; Z+ Y4 R. `1 {( {, k
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. 
     
159.     [Option('l', "lang", Required = false, HelpText = "Language", Default = "auto")]
     
160.     public string Language { get; set; }
     
161. 
     
162.     [Option('m', "modelFile", Required = false, HelpText = "Model to use (filename", Default = "ggml-large.bin")]
     8 X- q) N! o: J1 p
163.     public string ModelName { get; set; }
       |" V0 F+ e& `$ k9 a$ _
164. 
     % P2 n: G' N7 R" Y, `0 i
165.     [Option('g', "ggml", Required = false, HelpText = "Ggml Model type to download (if not exists)", Default = GgmlType.Base)]
       }  I0 ^  W9 [! w1 X+ [6 }
166.     public GgmlType ModelType { get; set; }
     
167. }
     

复制代码

; A4 U9 n8 V/ G' `1 q) ?5 a


* s. ^1 K2 ]+ T4 S: w0 C/ u

---

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

---

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

Powered by Discuz! X3.2