使用鼠标交互实现区域生长算法

案例© Fu Xianjun. All Rights Reserved. 

一:导包

import cv2 
import numpy as np

二:定义Point类
class Point(object):
    def __init__(self,x,y):
        self.x=x
        self.y=y
    def getX(self):
        return self.x
    def getY(self):
        return self.y

三:计算像素偏差
def getGrayDiff(img,currentPoint,tmpPoint):
    return abs(int(img[currentPoint.x,currentPoint.y]) -int(img[tmpPoint.x,tmpPoint.y]))

四:设置八领域

def selectConnects(p):
    if p==8:
        connects = [Point(-1, -1), Point(0, -1), Point(1, -1), Point(1, 0), Point(1, 1),Point(0, 1), Point(-1, 1), Point(-1, 0)]
    else:
        connects = [Point(0, -1), Point(1, 0),Point(0, 1),Point(-1, 0)]
    return connects

五:定义生长函数