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,
|
image=current_image,
|
||||||
point_inputs=backbone_out["point_inputs_per_frame"].get(stage_id, None),
|
point_inputs=backbone_out["point_inputs_per_frame"].get(stage_id, None),
|
||||||
mask_inputs=backbone_out["mask_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,
|
output_dict=output_dict,
|
||||||
num_frames=num_frames,
|
num_frames=num_frames,
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -657,8 +657,6 @@ class Sam3TrackerPredictor(Sam3TrackerBase):
|
|||||||
image=image,
|
image=image,
|
||||||
point_inputs=None,
|
point_inputs=None,
|
||||||
mask_inputs=mask_inputs,
|
mask_inputs=mask_inputs,
|
||||||
gt_masks=None,
|
|
||||||
frames_to_add_correction_pt=[],
|
|
||||||
output_dict={
|
output_dict={
|
||||||
"cond_frame_outputs": {},
|
"cond_frame_outputs": {},
|
||||||
"non_cond_frame_outputs": {},
|
"non_cond_frame_outputs": {},
|
||||||
|
|||||||
Reference in New Issue
Block a user