binary.py 文件源码

def to_binary_otsu(img, invert=False):
    if img.dtype == np.bool:
        img = np.array(img, 'uint8')
    if img.max() == img.min():
        if img.min() == 1:
            return np.array(img * 255, 'uint8')
        else:
            return np.array(img, 'uint8')
    else:
        t = threshold_otsu(img)
    img[img <= t] = 255 if invert else 0
    img[img > t] = 0 if invert else 255
    return np.array(img, 'uint8')