Spaces:
Sleeping
Sleeping
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
|