opencv之使用floodfill填充孔洞

import cv2
import numpy as np


def fillHole(src_image):
    src_image_temp = src_image.copy()
    # 创建mask, 必须长宽大于原图的2个像素
    h, w = src_image.shape[:2]
    mask = np.zeros((h + 2, w + 2), np.uint8)
    # 使用漫水填充， 种子点为（0， 0）
    cv2.floodFill(src_image_temp, mask, seedPoint=(0, 0), newVal=255)
    # 填充后的图像取反
    src_image_temp_inv = cv2.bitwise_not(src_image_temp)
    # 原图中填充空洞像素
    dst_image = cv2.bitwise_or(src_image, src_image_temp_inv)

    return dst_image


image = cv2.imread("C:\\Users\\PC\\Desktop\\22.bmp", 0)
res = fillHole(image)
cv2.imshow('im', res)
cv2.waitKey()