bbox updated
Browse files- 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="
|
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 |
|