zerohertzLib.vision.util ¶

def _cvt_bgra(img: NDArray[np.uint8]) -> NDArray[np.uint8]:
    """cv2로 읽어온 image를 BGRA 채널로 전환

    Args:
        img: 입력 image (`[H, W, C]`)

    Returns:
        BGRA image (`[H, W, 4]`)
    """
    shape = img.shape
    if len(shape) == 2:
        return cv2.cvtColor(img, cv2.COLOR_GRAY2BGRA)
    if shape[2] == 3:
        return cv2.cvtColor(img, cv2.COLOR_BGR2BGRA)
    return img

def _is_bbox(shape: tuple[int, ...]) -> tuple[bool, bool]:
    """Bbox 여부 검증

    Args:
        shape: Bbox의 `shape`

    Returns:
        복수의 bbox 여부 및 format의 정보
    """
    if len(shape) == 1 and shape[0] == 4:
        # [cx, cy, w, h] or N * [x0, y0, x1, y1]
        multi = False
        poly = False
    elif len(shape) == 2:
        if shape[1] == 4:
            # N * [cx, cy, w, h] or N * [x0, y0, x1, y1]
            multi = True
            poly = False
        elif shape[0] >= 4 and shape[1] == 2:
            # [[x0, y0], [x1, y1], [x2, y2], [x3, y3], ...]
            multi = False
            poly = True
        else:
            raise ValueError(
                "The 'box' must be of shape [4], [N, 4], [4, 2], [N, 4, 2]"
            )
    elif len(shape) == 3 and shape[1] == 4 and shape[2] == 2:
        # N *[[x0, y0], [x1, y1], [x2, y2], [x3, y3]]
        multi = True
        poly = True
    else:
        raise ValueError("The 'box' must be of shape [4], [N, 4], [4, 2], [N, 4, 2]")
    return multi, poly

def is_pts_in_poly(
    poly: NDArray[DTypeLike], pts: list[int | float] | NDArray[DTypeLike]
) -> bool | NDArray[bool]:
    """지점들의 좌표 내 존재 여부 확인 function

    Args:
        poly: 다각형 (`[N, 2]`)
        pts: point (`[2]` or `[N, 2]`)

    Returns:
        입력 `point` 의 다각형 `poly` 내부 존재 여부

    Examples:
        >>> poly = np.array([[10, 10], [20, 10], [30, 40], [20, 60], [10, 20]])
        >>> zz.vision.is_pts_in_poly(poly, [20, 20])
        True
        >>> zz.vision.is_pts_in_poly(poly, [[20, 20], [100, 100]])
        array([ True, False])
        >>> zz.vision.is_pts_in_poly(poly, np.array([20, 20]))
        True
        >>> zz.vision.is_pts_in_poly(poly, np.array([[20, 20], [100, 100]]))
        array([ True, False])
    """
    poly = Path(poly)
    if isinstance(pts, list):
        if isinstance(pts[0], list):
            return poly.contains_points(pts)
        return poly.contains_point(pts)
    if isinstance(pts, np.ndarray):
        shape = pts.shape
        if len(shape) == 1:
            return poly.contains_point(pts)
        if len(shape) == 2:
            return poly.contains_points(pts)
        raise ValueError("The 'pts' must be of shape [2], [N, 2]")
    raise TypeError("The 'pts' must be 'list' or 'np.ndarray'")