空洞填充Python实现

Python没有MATLAB中的imfill函数，要实现空洞填充必须自己写函数。文件包含代码以及测试图片，Python3的语法，亲测可用

import cv2;
import numpy as np;

# Read image
im_in = cv2.imread（“test.png“ cv2.IMREAD_GRAYSCALE）;

# Threshold.
# Set values equal to or above 220 to 0.
# Set values below 220 to 255.

th im_th = cv2.threshold（im_in 220 255 cv2.THRESH_BINARY_INV）;

# Copy the thresholded image.
im_floodfill = im_th.copy（）

# Mask used to flood filling.
# Notice the size needs to be 2 pixels than the image.
h w = im_th.shape[:2]
mask = np.zeros（（h+2 w+2） np.uint8）

# Floodfill from point （0 0）
cv2.floodFill（im_floodfill mask （00） 255）;

# Invert floodfilled image
im_floodfill_inv = cv2.bitwise_not（im_floodfill）

# Combine the two images to get the foreground.
im_out = im_th | im_floodfill_inv

# Display images.
cv2.imshow（“Thresholded Image“ im_th）
cv2.imshow（“Floodfilled Image“ im_floodfill）
cv2.imshow（“Inverted Floodfilled Image“ im_floodfill_inv）
cv2.imshow（“Foreground“ im_out）
cv2.waitKey（0）

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     文件         944  2018-06-13 10:22  imfill.py
     文件        9443  2018-06-13 10:18  test.png