Skip to content

Commit 586eb04

Browse files
feat(put_rectangle): change rectangle color on white. round probs values
1 parent feebcd1 commit 586eb04

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

functional.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,8 @@ def get_frame(h):
5656

5757
def put_rectangle(img, boxes, scores):
5858
for (x1, y1, x2, y2), score in zip(boxes.astype(int), scores):
59-
cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 0), 2)
60-
cv2.putText(img, str(score), (x1, y1),
59+
cv2.rectangle(img, (x1, y1), (x2, y2), (255, 255, 255), 2)
60+
cv2.putText(img, str(round(score, 3)), (x1, y1),
6161
cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 0, 0), 2, cv2.LINE_AA)
6262
return img
6363

0 commit comments

Comments
 (0)