nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fix scene_cut inaccuracy bug (#592)

Browse source
This commit is contained in:
xyupeng 2024-07-09 15:59:18 +08:00 committed by GitHub
parent 2a3206819d
commit 61e3790678
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tools/scene_cut/cut.py
View file

@ -35,7 +35,7 @@ def process_single_row(row, args):
if not (timestamp.startswith("[") and timestamp.endswith("]")):
return False
scene_list = eval(timestamp)
scene_list = [(FrameTimecode(s, fps=1), FrameTimecode(t, fps=1)) for s, t in scene_list]
scene_list = [(FrameTimecode(s, fps=100), FrameTimecode(t, fps=100)) for s, t in scene_list]
else:
scene_list = [None]
if args.drop_invalid_timestamps:
@ -104,8 +104,7 @@ def split_video(
duration = t - s
if max_seconds is not None:
fps = s.framerate
max_duration = FrameTimecode(timecode="00:00:00", fps=fps)
max_duration.frame_num = round(fps * max_seconds)
max_duration = FrameTimecode(max_seconds, fps=fps)
duration = min(max_duration, duration)
# save path