opencv 如何用相邻的颜色填充图像上的对象？

mkshixfv 于 2023-05-23 发布在其他

关注(0)|答案(2)|浏览(120)

目前，我正面临着用相邻的颜色给粉红色的盒子着色的麻烦，这样图像看起来会更真实的。我的形象是这样的：

到目前为止，我使用CV2包并实现了这一点：

我的代码：

up = np.array([151,157,255])
pink_mask = cv2.inRange(img, up, up)
cnts, _ = cv2.findContours(pink_mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
for c in cnts:
    color = tuple(map(int, img[0, 0]))
    cv2.fillPoly(img, pts=[c], color=color)

在这里，我用图像上的第一个像素填充，因为我不知道如何用相邻的颜色填充它。

opencv

来源：https://stackoverflow.com/questions/76278959/how-to-fill-objects-on-image-with-adjacent-colors

2条答案

按热度按时间

nhn9ugyo1#

我们可以扩大遮罩，并使用cv2.inPaint：

import numpy as np
import cv2

img = cv2.imread('input.png')

up = np.array([151,157,255])
pink_mask = cv2.inRange(img, up, up)
#cnts, _ = cv2.findContours(pink_mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# for c in cnts:
#     color = tuple(map(int, img[0, 0]))
#     cv2.fillPoly(img, pts=[c], color=color)
pink_mask = cv2.dilate(pink_mask, np.ones((3, 3), np.uint8))  # Dilate the mask

img = cv2.inpaint(img, pink_mask, 5, cv2.INPAINT_TELEA)

cv2.imwrite('output.png', img)

输出：