В основном, расширяйте mask
, используемый для рисования. Кроме того, увеличение радиуса inpaint даст немного лучшие результаты.
Это было бы моим предложением:
import cv2
from matplotlib import pyplot as plt
# Read image
img = cv2.imread('ic.png', cv2.IMREAD_GRAYSCALE)
# Binary threshold image
mask = cv2.threshold(img, 120, 255, cv2.THRESH_BINARY)[1]
# Remove small noise
inp_mask = cv2.morphologyEx(mask,
cv2.MORPH_OPEN,
cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5)))
# Dilate mask
inp_mask = cv2.dilate(inp_mask,
cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (15, 15)))
# Inpaint
dst = cv2.inpaint(img, inp_mask, 15, cv2.INPAINT_NS)
# Show results
plt.figure(1, figsize=(10, 10))
plt.subplot(2, 2, 1), plt.imshow(img, cmap='gray'), plt.title('Original image')
plt.subplot(2, 2, 2), plt.imshow(mask, cmap='gray'), plt.title('Thresholded image')
plt.subplot(2, 2, 3), plt.imshow(inp_mask, cmap='gray'), plt.title('Inpaint mask')
plt.subplot(2, 2, 4), plt.imshow(dst, cmap='gray'), plt.title('Inpainted image')
plt.tight_layout()
plt.show()
И это было бы выходные данные:
Надеюсь, это поможет!
----------------------------------------
System information
----------------------------------------
Platform: Windows-10-10.0.16299-SP0
Python: 3.8.1
Matplotlib: 3.2.0rc3
OpenCV: 4.2.0
----------------------------------------