opencv-python实现PS中的油漆填充功能3 多坐标一次性填充

Posted 2021-08-27 zzg1097940918

直接看效果

在坐标（100,100）、（200,200）、（250,250）这3个地方一键填充

 可以看到这个功能还是有些用处的

不过如果填充的点超过图像的尺寸填到外面就会报错

 cv.floodFill(Img2, mask, (a[i][0], a[i][1]), (127, 127, 127), (100, 100, 100), (50, 50 ,50), cv.FLOODFILL_FIXED_RANGE)  #获得每个坐标点进行填充
cv2.error: OpenCV(4.5.3) C:\\Users\\runneradmin\\AppData\\Local\\Temp\\pip-req-build-u4kjpz2z\\opencv\\modules\\imgproc\\src\\floodfill.cpp:509: error: (-211:One of the arguments' values is out of range) Seed point is outside of image in function 'cv::floodFill'

报这样的错误

另外opencv-python报错也有其他报错原因，比如下载方式的问题，

最好是

pip3 install opencv-contrib-python

如果已经安装了，需要删除的代码

pip uninstall opencv-python

当然，路径问题带中文，或空格的有时也会报错

那么直接上本文的代码

import cv2 as cv
import numpy as np

a = [[100, 100], [200, 200], [250, 250]]
def fill_color_demo(image,a):    #定义一键填充颜色的函数
    Img2 = image.copy()   #对输入的图片进行复制
    h, w = image.shape[:2]     #获得图片的长宽
    mask = np.zeros([h+2, w+2],np.uint8)   #mask必须行和列都加2，且必须为uint8单通道阵列，填充边缘需要多2个像素，否则会报错
    for i in range(len(a)):        #对数组内的坐标点遍历
        cv.floodFill(Img2, mask, (a[i][0], a[i][1]), (127, 127, 127), (100, 100, 100), (50, 50 ,50), cv.FLOODFILL_FIXED_RANGE)  #获得每个坐标点进行填充
    #cv.floodFill，参数1,：表示输入图片，参数2：表示单通道的掩码，参数3：表示泛洪算法的起始点，参数4表示填充的颜色，参数5,6表示当前观察像素点与邻域像素点最大正负差值
    #x坐标是从左到右，y坐标是从上到下
    cv.imshow("result", Img2) #显示结果图片

img = cv.imread('./1.jpg') #读入图片
cv.imshow('input', img)   #显示输入的图片
fill_color_demo(img,a)   #将输入的图片转入定义好的填充颜色函数中
cv.waitKey(0)        #imshow需要输入这个保持窗口开着