图像边缘算法-欧氏距离法

# -*- coding: utf-8 -*-
import cv2
import numpy as np

fn = '10093lv6-0.jpg'
def get_EuclideanDistance(x,y):
    myx = np.array(x)
    myy = np.array(y)
    return np.sqrt(np.sum((myx-myy)*(myx-myy)))

if __name__ == '__main__':
    print('loading %s ...' % fn)
    myimg1 = cv2.imread(fn)
    print ()
    w = myimg1.shape[1]
    h = myimg1.shape[0]
    print (w,h)
    sz1 = w
    sz0 = h
    # 创建空白图像
    myimg2 = np.zeros((sz0, sz1, 3), np.uint8)
    #cv2.imshow('img122', myimg2)
    # 对比产生线条
    #将当前像素与邻接的下部和右部的像素进行比较，如果相似，则将当前像素设置为白色，否则设置为黑色。
    #判定像素是否相似，使用欧氏距离算法，将一个像素的三个色彩分量映射在三维空间中，如果2个像素点的欧氏距离小于某个常数的阈值，就认为它们相似。
    black = np.array([0,0,0])
    white = np.array([255,255,255])
    centercolor = np.array([125,125,125])
    for y in range(0,sz0 - 1):
        for x in range(0,sz1 - 1):
            mydown = myimg1[y+1,x,:]
            myright = myimg1[y,x+1,:]
            myhere = myimg1[y,x,:]
            lmyhere = myhere
            lmyright = myright
            lmydown = mydown
            if get_EuclideanDistance(lmyhere, lmydown) >16 and get_EuclideanDistance(lmyhere,lmyright)>16:
                myimg2[y,x,:] = black
            elif get_EuclideanDistance(lmyhere,lmydown) <=16 and get_EuclideanDistance(lmyhere,lmyright)<=16:
                myimg2[y,x,:] = white
            else:
                myimg1[y,x,:]=centercolor
        print ('.',)
    cv2.namedWindow('img2')
    cv2.imshow('img2',myimg2)
    cv2.waitKey()
    cv2.destrdestroyAllWindows()