Remove extra args in track_step
Reviewed By: jayleicn Differential Revision: D87886578 Privacy Context Container: L1256182 fbshipit-source-id: 99d47aac7ca76ba8b321716b69d1306581152ac9
This commit is contained in:
committed by
meta-codesync[bot]
parent
2d1cbaeac7
commit
757bbb0206
@@ -900,8 +900,6 @@ class Sam3TrackerBase(torch.nn.Module):
|
||||
image=current_image,
|
||||
point_inputs=backbone_out["point_inputs_per_frame"].get(stage_id, None),
|
||||
mask_inputs=backbone_out["mask_inputs_per_frame"].get(stage_id, None),
|
||||
gt_masks=backbone_out["gt_masks_per_frame"].get(stage_id, None),
|
||||
frames_to_add_correction_pt=frames_to_add_correction_pt,
|
||||
output_dict=output_dict,
|
||||
num_frames=num_frames,
|
||||
)
|
||||
|
||||
@@ -657,8 +657,6 @@ class Sam3TrackerPredictor(Sam3TrackerBase):
|
||||
image=image,
|
||||
point_inputs=None,
|
||||
mask_inputs=mask_inputs,
|
||||
gt_masks=None,
|
||||
frames_to_add_correction_pt=[],
|
||||
output_dict={
|
||||
"cond_frame_outputs": {},
|
||||
"non_cond_frame_outputs": {},
|
||||
|
||||
Reference in New Issue
Block a user