Как удалить небольшие островковые пятна в масках растровых изображений с помощью python?

Question

Я новичок в обработке изображений и работаю над проектом, где мне нужно отделить изображения листа от фона как часть проекта.Мне удалось создать маску с использованием OpenCV, но есть некоторые исправления, которые следует удалить.

Вы можете увидеть изображение ниже, чтобы получить четкое представление о том, что я имел в виду:

Answer 1

Попробуйте:

import cv2
import numpy as np 

input = cv2.imread('source.png')
gray = cv2.cvtColor(input, cv2.COLOR_BGR2GRAY)

#find all contours
img = cv2.pyrDown(gray)
_, threshed = cv2.threshold(gray, 128, 255, cv2.THRESH_BINARY)
contours,_ = cv2.findContours(threshed, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

#find maximum contour and draw   
cmax = max(contours, key = cv2.contourArea) 
epsilon = 0.002 * cv2.arcLength(cmax, True)
approx = cv2.approxPolyDP(cmax, epsilon, True)
cv2.drawContours(input, [approx], -1, (0, 255, 0), 3)

cv2.imshow("Contour", input)

width, height = gray.shape

#fill maximum contour and draw   
img = np.zeros( [width, height, 3],dtype=np.uint8 )
cv2.fillPoly(img, pts =[cmax], color=(255,255,255))

cv2.imshow("Filtered", img)

cv2.waitKey(0)
cv2.destroyAllWindows()

Алгоритм :

1. FindContours метод.Вы можете настроить параметр epsilon.
2. Возьмите максимальное число - это ваш лист.
3. Заполните это максимальное число и нарисуйте его.Теперь вы можете сгладить его или выполнить любую другую морфологическую операцию.

Результаты :

Answer 2

На этой странице документации используйте открытие для ваших маленьких точек и Морфологический градиент , чтобы заполнить середину вашего изображения.