alikia2x/sparkastML
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

update: idk

Browse Source
This commit is contained in:
alikia2x (寒寒) 2024-11-16 21:51:19 +08:00
parent c4ca9c7d4f
commit fdff155673
Signed by: alikia2x
GPG Key ID: 56209E0CCD8420C6
1 changed files with 2 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
forced-alignment/split.py
View File

@ -171,12 +171,6 @@ def process_line(line_idx, start_time, total_lines):
break break
item["end"] = words[idx + 1]["start"] item["end"] = words[idx + 1]["start"]
idx+=1 idx+=1
if line_idx == total_lines:
words = words[1:]
elif line_idx == 1:
words = words[:-1]
else:
words = words[1:-1]
result = [] result = []
for word in words: for word in words:
result.append((word["word"], timestamp(word["start"]), timestamp(word["end"]))) result.append((word["word"], timestamp(word["start"]), timestamp(word["end"])))
@ -217,21 +211,18 @@ def parse_lrc(lrc_file, audio_len):
return lrc_data return lrc_data
lrc_data = parse_lrc("./data/谷雨_raw.lrc", duration) lrc_data = parse_lrc("./data/谷雨.lrc", duration)
i=0 i=0
for line_num in tqdm(lines_to_process): for line_num in tqdm(lines_to_process):
start_time = lrc_data[i][1] start_time = lrc_data[i][1]
result = process_line(line_num, start_time, len(lines_to_process))
end_time = lrc_data[i][2] end_time = lrc_data[i][2]
if timestamp_inverse(result[-1][2]) > end_time: result = process_line(line_num, start_time, len(lines_to_process))
end_time = timestamp_inverse(result[-1][2])
ttml_generator.add_lyrics( ttml_generator.add_lyrics(
begin=timestamp(start_time), end=timestamp(end_time), agent="v1", itunes_key=f"L{i+1}", begin=timestamp(start_time), end=timestamp(end_time), agent="v1", itunes_key=f"L{i+1}",
words=result words=result
) )
i+=1 i+=1
ttml_generator.save("output.ttml")
# 保存文件 # 保存文件
ttml_generator.save("output.ttml") ttml_generator.save("output.ttml")