You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
600 B
19 lines
600 B
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
|
|
|