sshi commited on
Commit
5eaf2b4
·
1 Parent(s): ff116e7

App bug fix.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ def plot_results(pil_img, prob, boxes):
106
  for p, (xmin, ymin, xmax, ymax) in zip(prob, boxes.tolist()):
107
  cl = p.argmax()
108
  c = colors[cl]
109
- c1, c2 = (xmin, ymin), (xmax, ymax)
110
 
111
  cv2.rectangle(img, c1, c2, c, thickness=2, lineType=cv2.LINE_AA)
112
  # cv2.text(
 
106
  for p, (xmin, ymin, xmax, ymax) in zip(prob, boxes.tolist()):
107
  cl = p.argmax()
108
  c = colors[cl]
109
+ c1, c2 = (int(xmin), int(ymin)), (int(xmax), int(ymax))
110
 
111
  cv2.rectangle(img, c1, c2, c, thickness=2, lineType=cv2.LINE_AA)
112
  # cv2.text(