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

Question

我正在尝试制作一个程序，它接收一张图像并查看整个图像以找到一种颜色，比如说蓝色，并给出图像中具有该颜色的那个点的坐标。

Answer 1

您可以使用 Numpy 非常简单地做到这一点，它是 Python 中大多数图像处理库的基础，例如 OpenCV、skimage 或 Wand .在这里，我将使用 OpenCV 来完成，但您同样可以使用上述任何一种或 PIL/Pillow。

使用这张右边有一条蓝线的图片:

#!/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
Y, X = np.where(np.all(im==blue,axis=2))

print(X,Y)

或者，如果您希望将它们压缩到一个数组中:

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

---

如果你更喜欢使用 PIL/Pillow，它会像这样:

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
Y, X = np.where(np.all(im==blue,axis=2))

print(X,Y)