sshi commited on
Commit
ff9db0f
·
1 Parent(s): 0a48b8a

App bug fix.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -115,7 +115,7 @@ def plot_results(pil_img, prob, boxes):
115
 
116
  for p, (xmin, ymin, xmax, ymax) in zip(prob, boxes.tolist()):
117
  cl = p.argmax()
118
- c = int(colors[cl])
119
  c1, c2 = (int(xmin), int(ymin)), (int(xmax), int(ymax))
120
 
121
  cv2.rectangle(img, c1, c2, c, thickness=2, lineType=cv2.LINE_AA)
 
115
 
116
  for p, (xmin, ymin, xmax, ymax) in zip(prob, boxes.tolist()):
117
  cl = p.argmax()
118
+ c = colors[cl]
119
  c1, c2 = (int(xmin), int(ymin)), (int(xmax), int(ymax))
120
 
121
  cv2.rectangle(img, c1, c2, c, thickness=2, lineType=cv2.LINE_AA)