Clip boxes in tracker (#3950)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
4329269c4e
commit
7137872f8e
1 changed files with 1 additions and 0 deletions
|
|
@ -114,6 +114,7 @@ class Results(SimpleClass):
|
|||
def update(self, boxes=None, masks=None, probs=None):
|
||||
"""Update the boxes, masks, and probs attributes of the Results object."""
|
||||
if boxes is not None:
|
||||
ops.clip_boxes(boxes, self.orig_shape) # clip boxes
|
||||
self.boxes = Boxes(boxes, self.orig_shape)
|
||||
if masks is not None:
|
||||
self.masks = Masks(masks, self.orig_shape)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue