Spaces:
Runtime error
Runtime error
Update services/overlay_service.py
Browse files
services/overlay_service.py
CHANGED
@@ -2,9 +2,6 @@ import cv2
|
|
2 |
|
3 |
def overlay_boxes(frame, boxes):
|
4 |
for box in boxes:
|
5 |
-
x_min, y_min, x_max, y_max
|
6 |
-
|
7 |
-
cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 2)
|
8 |
-
text = f"{label}: {score:.2f}"
|
9 |
-
cv2.putText(frame, text, (x_min, y_min - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 1)
|
10 |
return frame
|
|
|
2 |
|
3 |
def overlay_boxes(frame, boxes):
|
4 |
for box in boxes:
|
5 |
+
x_min, y_min, x_max, y_max = map(int, box)
|
6 |
+
cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), (0, 0, 255), 2)
|
|
|
|
|
|
|
7 |
return frame
|