pythonにて特定の色の座標を出力する方法

Question

やりたいこと

画像の中の、特定の色の座標を出力したい。

プログラム

ラベリング処理によって２値化画像をラベリングしています。
この続きがどうしても分かりません。
よろしくお願いします。

import cv2
import numpy as np
import matplotlib.pyplot as plt
from scipy import ndimage
from scipy import misc

# 2値化された画像をグレースケール形式で読み込む。
src = cv2.imread('mask.png')

# 連結成分のラベリングを行う。
gray = cv2.cvtColor(src, cv2.COLOR_BGR2GRAY)
num_labels, labels, stats, centroids = cv2.connectedComponentsWithStats(gray)

# ラベリング結果
print('number of labels', num_labels)
print(labels.shape, labels.dtype)  
plt.imshow(labels)
plt.show()

Answer 1

np.where() を使えばよいです。
返り値は Y座標、X座標で別れているので、np.dstack() で結合しています。

import numpy as np

img = np.random.randint(0, 10, (100, 100))

# 値が 5 の座標
indices = np.dstack(np.where(img == 5))
print('輝度値が5の座標一覧:\n', indices)

輝度値が5の座標一覧:
 [[[ 0 36]
  [ 0 37]
  [ 0 44]
  ...
  [99 89]
  [99 93]
  [99 94]]]

Comments

[masukin]

ご回答いつもありがとうございます。
輝度値が５であるとはどのように取得しますか？
すみませんが教えて頂けると助かります。

[tiitoi]

質問内容が「特定の色の座標」ということだったので、特定の色はすでにわかっていて座標を求めたいということだと思ったのですが、そうではないのでしょうか？
質問の場合、画像はラベリング結果なので
np.where(labels == 0) 背景のピクセルの座標
np.where(labels == 1) ラベル1のピクセルの座標
np.where(labels == 2) ラベル2のピクセルの座標
...
となるかと思います。

[masukin]

なるほどです。
すみません。失礼しました。
ご回答ありがとうございます。