Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,6 @@ def visualize_prediction(pil_img, output_dict, threshold=0.5, id2label=None):
|
|
48 |
wpercent = (basewidth/float(pil_img.size[0]))
|
49 |
hsize = int((float(pil_img.size[1])*float(wpercent)))
|
50 |
img = pil_img.resize((basewidth,hsize), Image.ANTIALIAS)
|
51 |
-
p
|
52 |
ax = plt.gca()
|
53 |
colors = COLORS * 100
|
54 |
for score, (xmin, ymin, xmax, ymax), label, color in zip(scores, boxes, labels, colors):
|
|
|
48 |
wpercent = (basewidth/float(pil_img.size[0]))
|
49 |
hsize = int((float(pil_img.size[1])*float(wpercent)))
|
50 |
img = pil_img.resize((basewidth,hsize), Image.ANTIALIAS)
|
|
|
51 |
ax = plt.gca()
|
52 |
colors = COLORS * 100
|
53 |
for score, (xmin, ymin, xmax, ymax), label, color in zip(scores, boxes, labels, colors):
|