File size: 778 Bytes
e62c44b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | import cv2
import numpy as np
from core.config import NUM_CLASSES, COLORS
def render_overlay(img, info):
canvas = img.copy()
overlay = np.zeros_like(img)
for cid in range(NUM_CLASSES):
mask = info[cid].get('mask_corrected', info[cid]['mask_original'])
overlay[mask] = COLORS[cid]
canvas = cv2.addWeighted(canvas, 1.0, overlay, 0.4, 0)
# Отрисовка скелетов поверх (упрощенно для API)
for cid in [0, 1]:
skel = info[cid].get('skeleton')
if skel is not None and skel.sum() > 0:
thick = cv2.dilate(skel.astype(np.uint8), np.ones((3, 3), np.uint8))
canvas[thick > 0] = (255, 255, 255) if cid == 0 else (255, 200, 100)
return canvas |