plane.py 文件源码

def is_coplanar(sample1, sample2):
    """
    To decide if two cluster of pixels are coplanar
    Args:
        sample1,sample2
    Returns:
        True or False
    """
    vec1 = fit_plane(sample1)
    vec2 = fit_plane(sample2)
    mag1 = np.sqrt(vec1.dot(vec1))
    mag2 = np.sqrt(vec2.dot(vec2))
    cosine = (vec1.dot(vec2))/(mag1 * mag2)
    if np.fabs(cosine)> 0.95:
        return True
    else:
        return False