我使用Python,opencv和PIL。
image = cv2.imread('image.jpg')
color = (235, 187, 7)
如果我知道像素颜色,如何获取像素坐标(x,y(?
这是一个数字解决方案。Numpy库尽可能加快操作速度。
- 假设颜色为:
color = (235, 187, 7)
indices = np.where(img == color)
- 我使用 numpy.where(( 方法检索两个数组的元组索引,其中第一个数组包含颜色像素的 x 坐标 (235, 187, 7(,第二个数组包含这些像素的 y 坐标。
现在indices
返回如下内容:
(array([ 81, 81, 81, ..., 304, 304, 304], dtype=int64),
array([317, 317, 317, ..., 520, 520, 520], dtype=int64),
array([0, 1, 2, ..., 0, 1, 2], dtype=int64))
- 然后,我使用 zip(( 方法获取包含这些点的元组列表。
coordinates = zip(indices[0], indices[1])
- 但是,如果您注意到这是具有三个通道的彩色图像,每个坐标将重复三次。我们只需要保留唯一的坐标。这可以使用
set()
方法完成。
unique_coordinates = list(set(list(coordinates)))
尝试类似操作:
color = (235, 187, 7)
im = Image.open('image.gif')
rgb_im = im.convert('RGB')
for x in range(rgb_im.size()[0]):
for y in range(rgb_im.size()[1]):
r, g, b = rgb_im.getpixel((x, y))
if (r,g,b) == colour:
print(f"Found {colour} at {x},{y}!")
但是getpixel可能很慢,所以看看使用像素访问对象。
另请注意,返回的值可能取决于图像类型。 例如,返回单个值时带有pix[1, 1]
,因为 GIF 像素引用 GIF 调色板中的 256 个值之一。
另请参阅这篇SO帖子:GIF和JPEG的Python和PIL像素值不同,此PIL参考页面包含有关convert()
函数的更多信息。
顺便说一下,您的代码可以很好地处理.jpg
图像。
您可以使用以下内容:
import numpy as np
# for color image
color = (75, 75, 75)
pixels = np.argwhere(img == color)
输出(重复相同的坐标三次(颜色数((:
[[ 0 28 0]
[ 0 28 1]
[ 0 28 2]
[ 0 54 0]
[ 0 54 1]
[ 0 54 2]
................]
为避免它,请执行以下操作(对不起代码可读性(:
pixels = pixels[::3][:, [0, 1]]
输出:
[[ 0 28]
[ 0 54]
...........]
对于灰度图像,它看起来更好:
color = (75)
pixels = np.argwhere(img == color)
输出:
[[ 0 28]
[ 0 54]
...........]
import PIL #The reason I use PIL and not opencv is that I find pillow
#(which is imported with 'PIL') a very useful library for image editing.
image = PIL.Image.open('Name_image') #the image is opened and named image
f = image.load() #I'm not sure what the load() operation exactly does, but it
# is necesarry.
color = (235, 187, 7) # the Red Green Blue values that you want to find the
#coordinates of
PixelCoordinates = [] # List in which all pixel coordinates that match
#requirements will be added.
#The lines of code below check for each pixel in the image if the RGB-values
# are equal to (235, 187, 7)
for x in image.size[0]:
for y in image.size[1]:
if f[x,y] == color:
PixelCoordinates.append([x,y])
这是一个仅使用 cv2 库的解决方案
import cv2
blue = int(input("Enter blue value: "))
green = int(input("Enter green value: "))
red = int(input("Enter red value: "))
path = str(input("Enter image path with image extension:"))
img = cv2.imread(path)
img= cv2.resize(img,(150,150))
x,y,z = img.shape
for i in range(x):
for j in range(y):
if img[i,j,0]==blue & img[i,j,1]==green & img[i,j,1]==red:
print("Found color at ",i,j)