在检测到指定颜色的图像中查找坐标

Question

我正试图制作一个程序，它接收一幅图像，并在整个图像中寻找一种颜色，比如说蓝色，并给出图像中具有该颜色的点的坐标

Answer 1

您可以非常简单地使用Numpy来实现这一点，Numpy是Python中大多数图像处理库的基础，例如OpenCV，或skimage，或Wand。在这里，我将使用OpenCVOpenCV，但您同样可以使用上述任何一种或PIL/枕头

使用右侧有一条蓝线的图像：

#!/usr/bin/env python3

import cv2
import numpy as np

# Load image
im = cv2.imread('image.png')

# Define the blue colour we want to find - remember OpenCV uses BGR ordering
blue = [255,0,0]

# Get X and Y coordinates of all blue pixels
X,Y = np.where(np.all(im==blue,axis=2))

print(X,Y)

输出

[ 0  2  4  6  8 10 12 14 16 18] [80 81 82 83 84 85 86 87 88 89]

或者，如果希望将它们压缩到单个数组中：

zipped = np.column_stack((X,Y))

array([[ 0, 80],
       [ 2, 81],
       [ 4, 82],
       [ 6, 83],
       [ 8, 84],
       [10, 85],
       [12, 86],
       [14, 87],
       [16, 88],
       [18, 89]])

---

如果您喜欢使用PIL/枕头，它将如下所示：

from PIL import Image
import numpy as np

# Load image, ensure not palettised, and make into Numpy array
pim = Image.open('image.png').convert('RGB')
im  = np.array(pim)

# Define the blue colour we want to find - PIL uses RGB ordering
blue = [0,0,255]

# Get X and Y coordinates of all blue pixels
X,Y = np.where(np.all(im==blue,axis=2))

print(X,Y)