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

Optimize seg_bbox calculations (#19056)

Browse source
This commit is contained in:
Muhammad Rizwan Munawar 2025-02-05 03:38:10 +05:00 • committed by GitHub
parent 5bca9341e8
commit c5ac5548ae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
ultralytics/utils/plotting.py
View file

@ -801,9 +801,8 @@ class Annotator:
return return
cv2.polylines(self.im, [np.int32([mask])], isClosed=True, color=mask_color, thickness=2) cv2.polylines(self.im, [np.int32([mask])], isClosed=True, color=mask_color, thickness=2)
text_size, _ = cv2.getTextSize(label, 0, self.sf, self.tf)
if label: if label:
text_size, _ = cv2.getTextSize(label, 0, self.sf, self.tf)
cv2.rectangle( cv2.rectangle(
self.im, self.im,
(int(mask[0][0]) - text_size[0] // 2 - 10, int(mask[0][1]) - text_size[1] - 10), (int(mask[0][0]) - text_size[0] // 2 - 10, int(mask[0][1]) - text_size[1] - 10),