eagle0504 commited on
Commit
8357c17
1 Parent(s): f6a717b

bbox updated

Browse files
Files changed (1) hide show
  1. utils/helpers.py +1 -1
utils/helpers.py CHANGED
@@ -224,7 +224,7 @@ def draw_boxes(image: Any, predictions: List[Dict[str, Any]]) -> Any:
224
  box = pred["box"]
225
  xmin, ymin, xmax, ymax = box.values()
226
  # Draw a rectangle over the image using the box's coordinates
227
- draw.rectangle([xmin, ymin, xmax, ymax], outline="red", width=2)
228
  # Annotate the image with label and score at the top-left corner of the bounding box
229
  draw.text((xmin, ymin), f"{label} ({score:.2f})", fill="red", font=font)
230
 
 
224
  box = pred["box"]
225
  xmin, ymin, xmax, ymax = box.values()
226
  # Draw a rectangle over the image using the box's coordinates
227
+ draw.rectangle([xmin, ymin, xmax, ymax], outline="green", width=1)
228
  # Annotate the image with label and score at the top-left corner of the bounding box
229
  draw.text((xmin, ymin), f"{label} ({score:.2f})", fill="red", font=font)
230