Skip to content

Commit 246ee7d

Browse files
nathanmdanBreakthrough
authored andcommitted
fix: update progress bar for frame skips
1 parent cc2d5de commit 246ee7d

1 file changed

Lines changed: 2 additions & 7 deletions

File tree

scenedetect/scene_manager.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -582,9 +582,6 @@ def detect_scenes(
582582
if next_frame is not None:
583583
frame_im = next_frame
584584
assert frame_im is not None
585-
if prev_position is None:
586-
prev_position = self._start_pos
587-
assert prev_position is not None
588585
new_cuts = self._process_frame(position, frame_im, callback)
589586
if progress_bar is not None:
590587
if new_cuts:
@@ -593,10 +590,8 @@ def detect_scenes(
593590
)
594591
# Increment progress bar by delta of position.frame_num instead of 1
595592
# to handle VFR video where frame count is an approximation.
596-
# First frame increments by 1 as base case.
597-
delta_pos = position.frame_num - prev_position.frame_num
598-
is_first_frame = position.frame_num == 0
599-
progress_bar.update((1 if is_first_frame else delta_pos) + frame_skip)
593+
delta = 1 if prev_position is None else position.frame_num - prev_position.frame_num
594+
progress_bar.update(delta)
600595
prev_position = position
601596
finally:
602597
if progress_bar is not None:

0 commit comments

Comments
 (0)