bowong
/
ComfyUI-CustomNode
6
0
Fork 0
Code Pull Requests Releases Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes/compute_video_point.py
View File

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