如何使用python获取某些颜色（RGB）的像素坐标[x，y]列表？

Question

我正在尝试进行室内导航，我需要机器人可以自动导航的室内地图。 我正在考虑使用每个地方（某些部分）具有不同颜色的图像，我想知道获取某些颜色坐标的方法。 这样我就可以使用该坐标为特定颜色区域指定位置。 如何使用python获取具有特定颜色的图像的像素坐标列表？ 在这种情况下，我知道颜色的 RGB 代码！ 我目前正在使用 pycharm

Answer 1

假设您的图片具有形状 (Y, X, 3)。 Y是图像的高度， X是宽度。 最后一个维度是 3 个 RGB 颜色通道。

您可以在图像上使用此代码来获取具有特定 RGB 集的像素坐标列表：

def pixels(image, rgb_set):
  set = list()

  for j in range(0, image.shape[0]):
    for i in range(0, image.shape[1]):
      if image[j][i] == rgb_set:
        set.append([j, i])

  return set