justin2341 commited on
Commit
2c1bbf5
·
verified ·
1 Parent(s): e4c24be

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +2 -0
demo.py CHANGED
@@ -21,7 +21,9 @@ def plot_one_box(x, img, color=None, label=None, score=None, line_thickness=3):
21
 
22
  c1 = c2
23
  c2 = c1[0] + t_size[0], c1[1] + t_size[1] + 3
 
24
  cv2.rectangle(img, c1, c2, [0, 0, 255], -1, cv2.LINE_AA) # filled
 
25
  return img
26
 
27
  def alpr(frame):
 
21
 
22
  c1 = c2
23
  c2 = c1[0] + t_size[0], c1[1] + t_size[1] + 3
24
+ pro = f"{score:02d}"
25
  cv2.rectangle(img, c1, c2, [0, 0, 255], -1, cv2.LINE_AA) # filled
26
+ cv2.putText(img, pro, (c1[0], c1[1] + 2), 0, tl / 3, [255, 255, 255], thickness=tf, lineType=cv2.LINE_AA)
27
  return img
28
 
29
  def alpr(frame):