nickmuchi commited on
Commit
a6f39ee
1 Parent(s): 94577e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,8 +52,8 @@ def visualize_prediction(img, output_dict, threshold=0.5, id2label=None):
52
  ax = plt.gca()
53
  colors = COLORS * 100
54
  for score, (xmin, ymin, xmax, ymax), label, color in zip(scores, boxes, labels, colors):
55
- ax.add_patch(plt.Rectangle((xmin, ymin), xmax - xmin, ymax - ymin, fill=False, color=color, linewidth=15))
56
- ax.text(xmin, ymin, f"{label}: {score:0.2f}", fontsize=45, bbox=dict(facecolor="yellow", alpha=0.5))
57
  plt.axis("off")
58
  return fig2img(plt.gcf())
59
 
 
52
  ax = plt.gca()
53
  colors = COLORS * 100
54
  for score, (xmin, ymin, xmax, ymax), label, color in zip(scores, boxes, labels, colors):
55
+ ax.add_patch(plt.Rectangle((xmin, ymin), xmax - xmin, ymax - ymin, fill=False, color=color, linewidth=10))
56
+ ax.text(xmin, ymin, f"{label}: {score:0.2f}", fontsize=55, bbox=dict(facecolor="yellow", alpha=0.5))
57
  plt.axis("off")
58
  return fig2img(plt.gcf())
59