孔洞填充的python代码

以下是一个简单的孔洞填充算法的Python代码实现：

import numpy as np

def fill_holes(img):
    # 使用OpenCV的findContours函数寻找图像中的轮廓
    contours, hierarchy = cv2.findContours(img, cv2.RETR_CCOMP, cv2.CHAIN_APPROX_SIMPLE)
    
    # 创建与原始图像大小相同的掩膜
    mask = np.zeros_like(img)
    
    # 对每个轮廓进行处理
    for i in range(len(contours)):
        # 如果轮廓属于外部轮廓，则将其填充为白色
        if hierarchy[0][i][3] == -1:
            cv2.drawContours(mask, contours, i, 255, -1)
        # 否则，将其填充为黑色
        else:
            cv2.drawContours(mask, contours, i, 0, -1)
    
    # 将掩膜与原始图像按位与，以得到填充后的图像
    result = cv2.bitwise_and(img, mask)
    
    return result

这个算法的基本思路是使用OpenCV的findContours函数找到图像中的所有轮廓，并根据轮廓的层次结构确定哪些轮廓是孔洞，然后将孔洞填充为白色或黑色。最后，将填充后的掩膜与原始图像按位与，以得到填充后的图像。