Spaces:
Sleeping
Sleeping
File size: 2,053 Bytes
eb0e26a b098c89 7a73535 b098c89 7a73535 b098c89 7a73535 eb0e26a b098c89 7698a6c a30d789 7698a6c b098c89 7a73535 a30d789 7a73535 b098c89 a30d789 7a73535 a30d789 27d8552 7a73535 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
import cv2
import numpy as np
import logging
from typing import List, Dict, Any
# Configure logging
logging.basicConfig(
filename="app.log",
level=logging.INFO,
format="%(asctime)s - %(levelname)s - %(message)s"
)
def overlay_boxes(frame: np.ndarray, detected_items: List[Dict[str, Any]]) -> np.ndarray:
"""
Overlay bounding boxes and labels on the frame for detected items.
Args:
frame: Input frame in BGR format.
detected_items: List of detected items with coordinates and labels.
Returns:
np.ndarray: Frame with overlaid bounding boxes and labels.
"""
if not isinstance(frame, np.ndarray) or frame.size == 0:
logging.error("Invalid input frame provided to overlay_service.")
return frame
if not isinstance(detected_items, list):
logging.error("Detected items must be a list.")
return frame
try:
for item in detected_items:
if "box" not in item or "type" not in item or "severity" not in item:
logging.warning(f"Skipping item without box, type, or severity: {item}")
continue
x_min, y_min, x_max, y_max = map(int, item["box"])
severity = item["severity"]
if item["type"] == "crack":
color = (0, 0, 255) if severity == "Severe" else (0, 255, 255) if severity == "Moderate" else (0, 255, 0) if severity == "Minor" else (255, 165, 0)
label = f"Crack: {severity}"
else: # Hole
color = (255, 0, 0) if severity == "Severe" else (255, 255, 0) if severity == "Moderate" else (0, 128, 0)
label = f"Hole: {severity}"
cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 2)
cv2.putText(frame, label, (x_min, y_min - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
logging.info(f"Overlaid {len(detected_items)} items on frame.")
return frame
except Exception as e:
logging.error(f"Error overlaying boxes: {str(e)}")
return frame |