Python-OpenCV实现二值图像孔洞填充

代码如下：

import cv2

import numpy as np

def FillHole(mask):

    contours, hierarchy = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

    len_contour = len(contours)

    contour_list = []

    for i in range(len_contour):

        drawing = np.zeros_like(mask, np.uint8)  # create a black image

        img_contour = cv2.drawContours(drawing, contours, i, (255, 255, 255), -1)

        contour_list.append(img_contour)

    out = sum(contour_list)

    return out

if __name__ == '__main__':

    mask_in = cv2.imread('C:\\Users\\admin\\Desktop\\mask_in.tif', 0)

    mask_out = FillHole(mask_in)

    cv2.imwrite('C:\\Users\\admin\\Desktop\\mask_out.tif', mask_out)

效果如下：

     

孔洞填充前                       孔洞填充后