HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix missing argument (#17253)

Browse source
This commit is contained in:
Laughing 2024-10-30 19:37:56 +08:00 • committed by GitHub
parent 6ffd8841fd
commit e798dbf52e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
ultralytics/models/sam/modules/sam.py
View file

@ -854,6 +854,7 @@ class SAM2Model(torch.nn.Module):
mask_inputs, mask_inputs,
output_dict, output_dict,
num_frames, num_frames,
track_in_reverse,
prev_sam_mask_logits, prev_sam_mask_logits,
): ):
"""Performs a single tracking step, updating object masks and memory features based on current frame inputs.""" """Performs a single tracking step, updating object masks and memory features based on current frame inputs."""