Function getCount

tools/ImageCropper/colormatcher.py:38–68

获取匹配成功的数量 Args: img: 图片 lower: 颜色下限值 upper: 颜色上限值 connected: bool 是否是相连的点才会被计数 method: int = -1 颜色匹配方式。即 cv::ColorConversionCodes。可选，默认 4 (RGB)。 常用值：4 (RGB, 3 通道), 40 (HSV, 3

(img, lower, upper, connected: bool, method: int = -1)

    return ret

def getCount(img, lower, upper, connected: bool, method: int = -1) -> int:
    '''
    获取匹配成功的数量
    
    Args:
        img:
            图片
        lower:
            颜色下限值
        upper:
            颜色上限值
        connected: bool
            是否是相连的点才会被计数
        method: int = -1
            颜色匹配方式。即 cv::ColorConversionCodes。可选，默认 4 (RGB)。
            常用值：4 (RGB, 3 通道), 40 (HSV, 3 通道), 6 (GRAY, 1 通道)。
            默认不进行转换。
    Return:
        int
    '''
    # https://github.com/MaaXYZ/MaaFramework/blob/main/source/MaaFramework/Vision/ColorMatcher.cpp
    # ColorMatcher::color_match
    if method >= 0:
        img = cv2.cvtColor(img, method)
    bin = cv2.inRange(img, np.array(lower, img.dtype), np.array(upper, img.dtype))
    # __show(img)
    # __show(bin)
    if connected:
        return __count_non_zero_with_connected(bin)
    else:
        return cv2.countNonZero(bin)

def __count_non_zero_with_connected(bin):
    number, labels, stats, centroids = cv2.connectedComponentsWithStats(bin, connectivity=8, ltype=cv2.CV_16U)