floodFill(image, mask, seedPoint, newVal[, loDiff[, upDiff[, flags]]])

src(seed.x, seed.y) - loDiff <= src(x, y) <= src(seed.x, seed.y) +upDiff

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

#讀取原始圖像
img = cv2.imread('windows.png')

#獲取圖像行和列
rows, cols = img.shape[:2]

#目標(biāo)圖像
dst = img.copy()

#mask必須行和列都加2且必須為uint8單通道陣列
#mask多出來(lái)的2可以保證掃描的邊界上的像素都會(huì)被處理
mask = np.zeros([rows+2, cols+2], np.uint8)  

#圖像漫水填充處理
#種子點(diǎn)位置(30,30) 設(shè)置顏色(0,255,255) 連通區(qū)范圍設(shè)定loDiff upDiff
#src(seed.x, seed.y) - loDiff <= src(x, y) <= src(seed.x, seed.y) +upDiff
cv2.floodFill(dst, mask, (30, 30), (0, 255, 255),
              (100, 100, 100), (50, 50, 50),
              cv2.FLOODFILL_FIXED_RANGE)

#顯示圖像
cv2.imshow('src', img)
cv2.imshow('dst', dst)
cv2.waitKey()
cv2.destroyAllWindows()

# coding:utf-8
import cv2
import random
import sys
import numpy as np

#使用說(shuō)明 點(diǎn)擊鼠標(biāo)選擇種子點(diǎn)
help_message = '''USAGE: floodfill.py [<image>]
Click on the image to set seed point
Keys:
  f - toggle floating range
  c - toggle 4/8 connectivity
  ESC - exit
'''
 
if __name__ == '__main__':

    #輸出提示文本
    print(help_message)

    #讀取原始圖像
    img = cv2.imread('scenery.png')

    #獲取圖像高和寬
    h, w = img.shape[:2]

    #設(shè)置掩碼 長(zhǎng)和寬都比輸入圖像多兩個(gè)像素點(diǎn) 
    mask = np.zeros((h+2, w+2), np.uint8)

    #設(shè)置種子節(jié)點(diǎn)和4鄰接
    seed_pt = None
    fixed_range = True
    connectivity = 4 

    #圖像漫水填充分割更新函數(shù)
    def update(dummy=None):
        if seed_pt is None:
            cv2.imshow('floodfill', img)
            return
        
        #建立圖像副本并漫水填充
        flooded = img.copy()
        mask[:] = 0 #掩碼初始為全0
        lo = cv2.getTrackbarPos('lo', 'floodfill') #像素鄰域負(fù)差最大值
        hi = cv2.getTrackbarPos('hi', 'floodfill') #像素鄰域正差最大值
        print('lo=', lo, 'hi=', hi)

        #低位比特包含連通值 4 (缺省) 或 8
        flags = connectivity
        
        #考慮當(dāng)前像素與種子像素之間的差（高比特也可以為0）
        if fixed_range:
            flags |= cv2.FLOODFILL_FIXED_RANGE
            
        #以白色進(jìn)行漫水填充
        cv2.floodFill(flooded, mask, seed_pt,
                      (random.randint(0,255), random.randint(0,255),
                       random.randint(0,255)), (lo,)*3, (hi,)*3, flags)

        #選定基準(zhǔn)點(diǎn)用紅色圓點(diǎn)標(biāo)出
        cv2.circle(flooded, seed_pt, 2, (0, 0, 255), -1)
        print("send_pt=", seed_pt)

        #顯示圖像
        cv2.imshow('floodfill', flooded)

    #鼠標(biāo)響應(yīng)函數(shù)
    def onmouse(event, x, y, flags, param):
        global seed_pt #基準(zhǔn)點(diǎn)

        #鼠標(biāo)左鍵響應(yīng)選擇漫水填充基準(zhǔn)點(diǎn)
        if flags & cv2.EVENT_FLAG_LBUTTON:
            seed_pt = x, y
            update()

    #執(zhí)行圖像漫水填充分割更新操作
    update()
    
    #鼠標(biāo)更新操作
    cv2.setMouseCallback('floodfill', onmouse)

    #設(shè)置進(jìn)度條
    cv2.createTrackbar('lo', 'floodfill', 20, 255, update)
    cv2.createTrackbar('hi', 'floodfill', 20, 255, update)

    #按鍵響應(yīng)操作
    while True:
        ch = 0xFF & cv2.waitKey()
        #退出
        if ch == 27:
            break
        #選定時(shí)flags的高位比特位0
        #鄰域的選定為當(dāng)前像素與相鄰像素的差, 聯(lián)通區(qū)域會(huì)很大
        if ch == ord('f'):
            fixed_range = not fixed_range 
            print('using %s range' % ('floating', 'fixed')[fixed_range])
            update()
        #選擇4方向或則8方向種子擴(kuò)散
        if ch == ord('c'):
            connectivity = 12-connectivity 
            print('connectivity =', connectivity)
            update()
    cv2.destroyAllWindows()