From 47b5578d15c5b80a07ad9bbd5c8750e6de0f49f3 Mon Sep 17 00:00:00 2001 From: Ronghang Hu Date: Wed, 7 Aug 2024 18:53:29 +0000 Subject: [PATCH] nits --- sam2/sam2_video_predictor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sam2/sam2_video_predictor.py b/sam2/sam2_video_predictor.py index 932166237..b5a6bdf4b 100644 --- a/sam2/sam2_video_predictor.py +++ b/sam2/sam2_video_predictor.py @@ -590,7 +590,7 @@ def propagate_in_video_preflight(self, inference_state): storage_key = "cond_frame_outputs" if is_cond else "non_cond_frame_outputs" # Find all the frames that contain temporary outputs for any objects # (these should be the frames that have just received clicks for mask inputs - # via `add_new_points` or `add_new_mask`) + # via `add_new_points_or_box` or `add_new_mask`) temp_frame_inds = set() for obj_temp_output_dict in temp_output_dict_per_obj.values(): temp_frame_inds.update(obj_temp_output_dict[storage_key].keys())