Question

我堆积了找到当前颜色间隔的像素位置的问题。这很慢：

def inRange(self, tgColor, excColor, jump):
    if tgColor[0] > excColor[0] - jump and tgColor[0] < excColor[0] + jump and tgColor[1] > excColor[1] - jump and tgColor[1] < excColor[1] + jump and tgColor[2] > excColor[2] - jump and tgColor[2] < excColor[2] + jump:
           return True
    return False

 for iy in xrange(self.pilImage.size[1]):
        for ix in xrange(self.pilImage.size[0]):
            if self.inRange(data[ix, iy], self.targetColor, self.jump):

那么，您能否帮助我改进此代码，使其更快地运行。（图像尺寸 - 640 x 480）也许是另一个lib：OpenCV，pygame，PIL？

Answer 1

您的代码可能超级慢

OpenCV带有一个函数cv2.inRange()。您传递最小和最大像素值，并获得一个二进制图像，其中传递像素为白色，失败像素为黑色。

然后，您可以使用numpy.where()查找白色像素的索引。

以下是灰度值的示例。它也可以扩展到彩色图像。 [Link]

示例：

>>> import cv2
>>> import numpy as np
>>> x = np.random.randint(1,10, (5,5))
>>> x
array([[9, 5, 1, 3, 1],
       [7, 7, 2, 1, 7],
       [9, 1, 4, 7, 4],
       [3, 6, 6, 7, 2],
       [3, 4, 2, 3, 1]])
>>> y = cv2.inRange(x,4,8)
>>> y
array([[  0, 255,   0,   0,   0],
       [255, 255,   0,   0, 255],
       [  0,   0, 255, 255, 255],
       [  0, 255, 255, 255,   0],
       [  0, 255,   0,   0,   0]], dtype=uint8)

>>> z = np.transpose(np.where(y>0))
>>> z
array([[0, 1],
       [1, 0],
       [1, 1],
       [1, 4],
       [2, 2],
       [2, 3],
       [2, 4],
       [3, 1],
       [3, 2],
       [3, 3],
       [4, 1]])

使用当前颜色间隔快速查找像素位置

1 个答案: