时间戳

首页 > 解决方案 > 在图像中查找某个像素值

python - 在图像中查找某个像素值

问题描述

我想检查每个像素是否为红色。因此，我准备了一张白色图像，在其中打印了一些红线。红线正是这种颜色：(0, 0, 255)。

使用以下代码，在图像中未检测到红色。怎么了？

import cv2
import numpy as np

img = cv2.imread(r"....\red.png")
x,y,z = img.shape
print(x,y,z)

for i in range(x):
    for j in range(y):
        r = np.where((img[i,j,0]==0 & img[i,j,1]==0 & img[i,j,2]==255))
        redarray = np.array(r)
        red = np.size(redarray)

print("r: ", r)
print("redarray: ", redarray)
print("red: ", red)

标签： pythonimagenumpyopencvfor-loop

解决方案

您不需要 for 循环来完成此操作。numpy 的优点之一！

np.where将通过一些工作来做到这一点：

np.where((img == (0, 0, 255)).all(axis=2))

按照您的示例，您将需要创建一个 if 语句，如果像素为红色，则该语句将构建一个像素列表。redarray在您的帖子中将仅反映每个 for 循环迭代的当前像素。你不是在那里建立一个列表。所以这样的事情应该有效：

red_pixels = []
for i in range(x):
    for j in range(y):
        if np.array_equal(img[i, j], [0, 0, 255]):
            red_pixels.append((i, j))

推荐阅读