capjamesg commited on
Commit
afa8f00
1 Parent(s): dbe0801

update plotting code, fix data conversion bug

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ def plot_bounding_boxes(image, boxes, box_type):
8
  if not boxes:
9
  return image
10
 
11
- x0, y0, x1, y1 = [int(i.strip()) for i in boxes.split(",")]
12
 
13
  if box_type == "xywh":
14
  x0 = x0 * image.size[0]
@@ -24,7 +24,7 @@ def plot_bounding_boxes(image, boxes, box_type):
24
 
25
  image = np.array(image)
26
 
27
- bounding_box_annotator = sv.BoundingBoxAnnotator()
28
  annotated_image = bounding_box_annotator.annotate(
29
  scene=image, detections=detections
30
  )
 
8
  if not boxes:
9
  return image
10
 
11
+ x0, y0, x1, y1 = [int(float(i.split(".")[0])) for i in boxes.split(",")]
12
 
13
  if box_type == "xywh":
14
  x0 = x0 * image.size[0]
 
24
 
25
  image = np.array(image)
26
 
27
+ bounding_box_annotator = sv.BoundingBoxAnnotator(thickness=3)
28
  annotated_image = bounding_box_annotator.annotate(
29
  scene=image, detections=detections
30
  )