Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def image_to_base64(image):
|
|
30 |
def draw_bounding_boxes(image, bounding_boxes, outline_color="red", line_width=2):
|
31 |
draw = ImageDraw.Draw(image)
|
32 |
for box in bounding_boxes:
|
33 |
-
xmin,
|
34 |
draw.rectangle([xmin, ymin, xmax, ymax], outline=outline_color, width=line_width)
|
35 |
return image
|
36 |
|
@@ -40,11 +40,11 @@ def rescale_bounding_boxes(bounding_boxes, original_width, original_height, scal
|
|
40 |
y_scale = original_height / scaled_height
|
41 |
rescaled_boxes = []
|
42 |
for box in bounding_boxes:
|
43 |
-
xmin,
|
44 |
rescaled_box = [
|
45 |
xmin * x_scale,
|
46 |
-
xmax * x_scale,
|
47 |
ymin * y_scale,
|
|
|
48 |
ymax * y_scale
|
49 |
]
|
50 |
rescaled_boxes.append(rescaled_box)
|
|
|
30 |
def draw_bounding_boxes(image, bounding_boxes, outline_color="red", line_width=2):
|
31 |
draw = ImageDraw.Draw(image)
|
32 |
for box in bounding_boxes:
|
33 |
+
xmin, ymin, xmax, ymax = box
|
34 |
draw.rectangle([xmin, ymin, xmax, ymax], outline=outline_color, width=line_width)
|
35 |
return image
|
36 |
|
|
|
40 |
y_scale = original_height / scaled_height
|
41 |
rescaled_boxes = []
|
42 |
for box in bounding_boxes:
|
43 |
+
xmin, ymin, xmax, ymax = box
|
44 |
rescaled_box = [
|
45 |
xmin * x_scale,
|
|
|
46 |
ymin * y_scale,
|
47 |
+
xmax * x_scale,
|
48 |
ymax * y_scale
|
49 |
]
|
50 |
rescaled_boxes.append(rescaled_box)
|