83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
import whisper
|
||
import math
|
||
from pydub import AudioSegment # pip install pydub
|
||
|
||
# --------- 参数 ----------
|
||
MODEL_NAME = "turbo"
|
||
DEVICE = "cuda"
|
||
AUDIO_PATH = "audio.mp3"
|
||
SEGMENT_SEC = 10 # 每段 10 秒，越小粒度越细
|
||
OVERLAP_SEC = 0
|
||
LANGUAGE = "zh"
|
||
INITIAL_PROMPT = "这是一段小说。"
|
||
# -------------------------
|
||
|
||
# 1. 把音频切成小段
|
||
audio = AudioSegment.from_file(AUDIO_PATH)
|
||
total_ms = len(audio) # 总时长（毫秒）
|
||
segment_ms = SEGMENT_SEC * 1000
|
||
step_ms = (SEGMENT_SEC - OVERLAP_SEC) * 1000
|
||
|
||
segments = [
|
||
audio[i:i + segment_ms] # pydub 切片
|
||
for i in range(0, total_ms, step_ms)
|
||
]
|
||
|
||
total_chunks = len(segments)
|
||
print(f"音频共 {total_ms / 1000:.1f} 秒，切成 {total_chunks} 段处理")
|
||
|
||
# 2. 加载模型
|
||
model = whisper.load_model(MODEL_NAME, device=DEVICE)
|
||
|
||
# 3. 逐段转录 + 打印进度
|
||
all_segments = []
|
||
|
||
|
||
def percentage_of(total: int, part: int) -> int:
|
||
if total <= 0:
|
||
raise ValueError("第一个参数（总数）必须大于 0")
|
||
if part >= total:
|
||
return 100
|
||
return int(round((part / total) * 100))
|
||
|
||
|
||
for idx, seg in enumerate(segments, 1):
|
||
# 临时保存为 wav，让 Whisper 读取
|
||
tmp_wav = f"_tmp_{idx}.wav"
|
||
seg.export(tmp_wav, format="wav")
|
||
|
||
result = model.transcribe(
|
||
tmp_wav,
|
||
language=LANGUAGE,
|
||
initial_prompt=INITIAL_PROMPT,
|
||
verbose=False,
|
||
word_timestamps=False
|
||
)
|
||
all_segments.append(result)
|
||
|
||
# 计算并打印进度
|
||
progress = idx / total_chunks * 100
|
||
print(str(percentage_of(total_chunks, idx)) + "%")
|
||
|
||
# 删除临时文件
|
||
import os;
|
||
|
||
os.remove(tmp_wav)
|
||
|
||
print("\n转录完成！")
|
||
|
||
|
||
# 4. 合并结果，按时间轴输出
|
||
def seconds_to_min_sec(total_seconds: int) -> str:
|
||
minutes, seconds = divmod(total_seconds, 60)
|
||
return f"{minutes}分{seconds:02d}秒"
|
||
|
||
|
||
current_start = 0
|
||
for chunk_result in all_segments:
|
||
for seg in chunk_result["segments"]:
|
||
start = current_start + seg["start"]
|
||
end = current_start + seg["end"]
|
||
print(f"时间：{seconds_to_min_sec(int(start))} ~ {seconds_to_min_sec(int(end))}；文本：{seg['text'].strip()}")
|
||
current_start += SEGMENT_SEC - OVERLAP_SEC
|