xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #1829 from roboflow/fix/process_video 

fix: `UnboundLocalError` in `process_video`
This commit is contained in:
Piotr Skalski 2025-04-25 14:55:29 +02:00 committed by GitHub
parent 49d5505efa 00df1cbb04
commit 21fc9afbdf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
supervision/utils/video.py
View File

@ -234,13 +234,11 @@ def process_video(
source_path=source_path, end=max_frames
)
with VideoSink(target_path=target_path, video_info=source_video_info) as sink:
if show_progress:
# Calculate total_frames only when needed for tqdm
total_frames = (
min(source_video_info.total_frames, max_frames)
if max_frames is not None
else source_video_info.total_frames
)
total_frames = (
min(source_video_info.total_frames, max_frames)
if max_frames is not None
else source_video_info.total_frames
)
for index, frame in enumerate(
tqdm(
video_frames_generator,