Spaces:
Running
Running
File size: 1,893 Bytes
e735bf3 |
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 51 |
from typing import Tuple
import cv2
import numpy as np
from models import FingerQualityResult
class Visualizer:
@staticmethod
def draw_axis(img, center, vec, length, color, thickness=2):
x0, y0 = center
x1 = int(x0 + length * vec[0])
y1 = int(y0 + length * vec[1])
cv2.arrowedLine(img, (x0, y0), (x1, y1), color, thickness, tipLength=0.2)
@staticmethod
def draw_debug(
img: np.ndarray,
contour: np.ndarray,
bbox: Tuple[int, int, int, int],
angle_deg: float,
result: FingerQualityResult
) -> np.ndarray:
out = img.copy()
x, y, w_box, h_box = bbox
cv2.rectangle(out, (x, y), (x + w_box, y + h_box), (0, 255, 0), 2)
cv2.drawContours(out, [contour], -1, (255, 0, 0), 2)
pts = contour.reshape(-1, 2).astype(np.float64)
mean, eigenvectors, _ = cv2.PCACompute2(pts, mean=np.empty(0))
center = (int(mean[0, 0]), int(mean[0, 1]))
main_vec = eigenvectors[0]
Visualizer.draw_axis(out, center, main_vec, length=80, color=(0, 0, 255), thickness=2)
text_lines = [
f"Blur: {result.blur_score:.1f} ({'OK' if result.blur_pass else 'BAD'})",
f"Illum: {result.illumination_score:.1f} ({'OK' if result.illumination_pass else 'BAD'})",
f"Coverage: {result.coverage_ratio*100:.1f}% ({'OK' if result.coverage_pass else 'BAD'})",
f"Angle: {angle_deg:.1f} deg ({'OK' if result.orientation_pass else 'BAD'})",
f"Quality: {result.quality_score:.2f} ({'PASS' if result.overall_pass else 'FAIL'})",
]
y0 = 25
for line in text_lines:
color = (0, 255, 0) if ("OK" in line or "PASS" in line) else (0, 0, 255)
cv2.putText(out, line, (10, y0), cv2.FONT_HERSHEY_SIMPLEX, 0.6, color, 2, cv2.LINE_AA)
y0 += 22
return out
|