Initial scoliosis pipeline (without models)\

2026-01-23 20:35:07 +05:00
commit 83dfbcc238
11 changed files with 1638 additions and 0 deletions
@@ -0,0 +1,19 @@
+import cv2
+import numpy as np
+
+def draw_arc_contours(
+    img_bgr: np.ndarray,
+    vertebrae,
+    arc_infos,
+    colors=((255, 0, 255), (0, 255, 255), (255, 255, 0)),
+):
+    """Визуализация контура дуги деформации"""
+    vis = img_bgr.copy()
+    for i, info in enumerate(arc_infos):
+        pts = info.get("centerline_pts")
+        if pts is None:
+            continue
+        pts_i = np.round(pts).astype(np.int32).reshape(-1, 1, 2)
+        color = colors[i % len(colors)]
+        cv2.polylines(vis, [pts_i], isClosed=False, color=color, thickness=3)
+    return vis