Python中的平滑二進制分割掩碼圖像

Question

我想平滑我的二值圖像分割掩碼的輪廓，以改進我的卷積神經網絡分割算法的輸出。

我試過PIL：

# Smooth the outline
from PIL import Image, ImageFilter
pil_img = Image.fromarray(orignal_mask)
smoothed_image = pil_img.filter(ImageFilter.SMOOTH)
smoothed_image = smoothed_image.filter(ImageFilter.SMOOTH_MORE)
smoothed_image_arr = np.array(smoothed_image)
smoothed_image_arr[smoothed_image_arr>=127] = 255
smoothed_image_arr[smoothed_image_arr<127] = 0
cv2.imwrite(data_dir + 'test_masks/smooth_{}'.format(jpg_filename), smoothed_image_arr)

閾值處理后，生成的圖像看起來與原始蒙版基本相同。 我做錯了什么嗎？ 有沒有其他方法可以平滑二進制掩碼？ 我對任何 Python 解決方案持開放態度。

原始鋸齒狀二進制圖像蒙版

Answer 1

您可以嘗試使用 OpenCV 應用形態學操作，開口可以稍微平滑邊緣：

cv2.morphologyEx(mask, cv2.MORPH_OPEN,cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3)))