Update demo.py

demo.py

@@ -19,9 +19,10 @@ def plot_one_box(x, img, color=None, label=None, score=None, line_thickness=3):
         cv2.rectangle(img, c1, c2, color, -1, cv2.LINE_AA)  # filled
         cv2.putText(img, label, (c1[0], c1[1] - 2), 0, tl / 3, [0, 0, 0], thickness=tf, lineType=cv2.LINE_AA)
 
+        pro = f"{score:.3f}"
+        t_size = cv2.getTextSize(pro, 0, fontScale=tl / 3, thickness=tf)[0]
         c1 = c2
         c2 = c1[0] + t_size[0], c1[1] + t_size[1] + 3
-        pro = f"{score:.2f}"
         cv2.rectangle(img, c1, c2, [0, 255, 255], -1, cv2.LINE_AA)  # filled
         cv2.putText(img, pro, (c1[0], c2[1] - 2), 0, tl / 3, [0, 0, 0], thickness=tf, lineType=cv2.LINE_AA)
     return img