heimoshuiyu/faster-whisper
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

Keep segment timestamps aligned with words timestamps after VAD (#119)

Browse Source
This commit is contained in:
Guillaume Klein
2023-04-06 11:54:40 +02:00
committed by GitHub
parent 051b3350e5
commit e9a082dcf2
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
faster_whisper/transcribe.py
View File

@@ -776,14 +776,18 @@ def restore_speech_timestamps(
end=ts_map.get_original_time(word.end, chunk_index), end=ts_map.get_original_time(word.end, chunk_index),
) )
words.append(word) words.append(word)
else:
words = segment.words
segment = segment._replace( segment = segment._replace(
start=ts_map.get_original_time(segment.start), start=words[0].start,
end=ts_map.get_original_time(segment.end), end=words[-1].end,
words=words, words=words,
) )
else:
segment = segment._replace(
start=ts_map.get_original_time(segment.start),
end=ts_map.get_original_time(segment.end),
)
yield segment yield segment