FIX 增加视频截取长度保证话说完

This commit is contained in:
康宇佳 2025-03-04 17:56:16 +08:00
parent cafa224374
commit 810d903244
1 changed files with 3 additions and 2 deletions

View File

@ -10,7 +10,8 @@ def validate_time_format(time_str):
def get_duration_wave(audio):
waveform, sample_rate = audio["waveform"], audio["sample_rate"]
return waveform.shape[2] / sample_rate
# 防止话说不完
return ceil(waveform.shape[2] / sample_rate) + 1
class VideoStartPointDurationCompute:
@ -40,5 +41,5 @@ class VideoStartPointDurationCompute:
start_sec = (start_dt - datetime(1900, 1, 1)).total_seconds()
start_point = int(start_sec * fps)
print("audio duration %.2f s"%get_duration_wave(audio))
duration = ceil(get_duration_wave(audio) * fps)
duration = get_duration_wave(audio) * fps
return (start_point, duration,)