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

App bug fix.

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -120,14 +120,7 @@ def plot_results(pil_img, prob, boxes):
120
 
121
  cv2.rectangle(img, c1, c2, c, thickness=2, lineType=cv2.LINE_AA)
122
 
123
- cv2.putText(
124
- img,
125
- f'{id2label[cl.item()]}: {p[cl]:0.2f}',
126
- [xmin + 5, ymin + 5],
127
- cv2.FONT_HERSHEY_SIMPLEX,
128
- 0.7,
129
- c,
130
- 2)
131
  # ax.text(xmin, ymin, text, fontsize=10,
132
  # bbox=dict(facecolor=c, alpha=0.5))
133
  return Image.fromarray(img)
 
120
 
121
  cv2.rectangle(img, c1, c2, c, thickness=2, lineType=cv2.LINE_AA)
122
 
123
+ cv2.putText(img, f'{id2label[cl.item()]}: {p[cl]:0.2f}', [int(xmin) + 5, int(ymin) + 5], cv2.FONT_HERSHEY_SIMPLEX, 0.7, c, 2)
 
 
 
 
 
 
 
124
  # ax.text(xmin, ymin, text, fontsize=10,
125
  # bbox=dict(facecolor=c, alpha=0.5))
126
  return Image.fromarray(img)