Spaces:
Runtime error
Runtime error
sshi
commited on
Commit
·
ff9db0f
1
Parent(s):
0a48b8a
App bug fix.
Browse files
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 =
|
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)
|