diff --git a/nodes/compute_video_point.py b/nodes/compute_video_point.py index 5bd9d3a..d4aa267 100644 --- a/nodes/compute_video_point.py +++ b/nodes/compute_video_point.py @@ -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,)