From 962998a167f8146184d12c058b0f3685e844f16c Mon Sep 17 00:00:00 2001 From: generatedunixname537391475639613 Date: Sun, 21 Dec 2025 01:40:42 -0800 Subject: [PATCH] fbcode/deeplearning/projects/sam3_release/sam3/train/nms_helper.py Differential Revision: D88935213 fbshipit-source-id: b0b9cd57858641f7ce398865caef5eed4ad5d8bb --- sam3/train/nms_helper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sam3/train/nms_helper.py b/sam3/train/nms_helper.py index cd5b6dc..d6378bc 100644 --- a/sam3/train/nms_helper.py +++ b/sam3/train/nms_helper.py @@ -33,7 +33,7 @@ def convert_bbox_format(bbox: list) -> List[float]: # -------------------- Track-level NMS -------------------- def process_track_level_nms(video_groups: Dict, nms_threshold: float) -> Dict: """Apply track-level NMS to all videos""" - for video_id, tracks in video_groups.items(): + for tracks in video_groups.values(): track_detections = [] # Process tracks @@ -76,7 +76,7 @@ def process_track_level_nms(video_groups: Dict, nms_threshold: float) -> Dict: # -------------------- Frame-level NMS -------------------- def process_frame_level_nms(video_groups: Dict, nms_threshold: float) -> Dict: """Apply frame-level NMS to all videos""" - for video_id, tracks in video_groups.items(): + for tracks in video_groups.values(): if not tracks: continue