maxiw commited on
Commit
bbaff07
1 Parent(s): e58d5af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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, xmax, ymin, ymax = box
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, xmax, ymin, ymax = box
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)