середа, 21 квітня 2021 р.

Детектор кутів Харіса


def HCorners(img,k=0.04,dilatation=False,Threshold=0.01):
    if img.ndim>=3:
        img_gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
        copy=img.copy()
    else:
        img_gray=img
        copy=cv2.cvtColor(img,cv2.COLOR_GRAY2RGB)
        
    img_gray = np.float32(img_gray)
    dst = cv2.cornerHarris(img_gray,2,3,k)

    if dilatation:
        dst = cv2.dilate(dst,None)
    copy[dst> Threshold*dst.max()]=[0,0,255]
    return copy
    
 

Немає коментарів:

Дописати коментар