２つの値を元に画像上にプロットする python opencv

Question

この画像にプログラムに書かれているラベリング処理を２回した時に求めた
x座標とy座標の値を赤い点として表示させたいです

赤い点と書いてありますが
わかりやすくばつ印などでも良いです

！！！！！！下記のビックリマークのところをどう直せば良いか！！！！！！！！
教えてください

import cv2
import numpy as np
from matplotlib import pyplot as plt



src= cv2.imread("kuro1.png")



#ユニフォーム抽出
dst1 = cv2.inRange(src, (3,3,3), (10,8,10))
dst2 = cv2.blur(dst1, ksize=(5,5))
ret,thresh1 = cv2.threshold(dst2,127,255,cv2.THRESH_BINARY)

# ラベリング処理（x座標）
label = cv2.connectedComponentsWithStats(thresh1)

stats = label[2]
area = stats[:, cv2.CC_STAT_WIDTH] * stats[:, cv2.CC_STAT_HEIGHT]
top2_idx = area.argsort()[-3:-1]  # 2番目、3番めに面積が大きいラベル (1番目は背景なので除く)

# オブジェクト情報を項目別に抽出
data = label[2]
center = label[3]

# ラベリング結果書き出し用に二値画像をカラー変換
color_src = cv2.cvtColor(thresh1, cv2.COLOR_GRAY2BGR)

# オブジェクト情報を利用してラベリング結果を表示
for i in top2_idx:
    # 各オブジェクトの外接矩形を赤枠で表示
    x0 = data[i][0]
    y0 = data[i][1]
    x1 = data[i][0] + data[i][2]
    y1 = data[i][1] + data[i][3]
    cv2.rectangle(color_src, (x0, y0), (x1, y1), (0, 0, 255))




    # 各オブジェクトの重心座標をに黄文字で表示
    cv2.putText(color_src, "X: " + str(int(center[i][0])), (x1 - 10, y1 + 15), cv2.FONT_HERSHEY_PLAIN, 1, (0, 255, 255))
 

cv2.imshow('x',color_src)

  

















#「背景」となる画像の取り込み（グレースケール）
img_src01 = cv2.imread("zero1.png", 0)
 
#「差分」をもった画像の取り込み（グレースケール）
img_src02 = cv2.imread("kuro1.png", 0)
 
#「背景差分」計算用オブジェクトの作成　
bgObj = cv2.bgsegm.createBackgroundSubtractorMOG()
 
#差分となっている「前景領域」に対してマスクをかける
fgmask = bgObj.apply(img_src01)
fgmask = bgObj.apply(img_src02)

# ラベリング処理(y座標)
label2 = cv2.connectedComponentsWithStats(fgmask)

### 追加した部分1
stats2 = label2[2]
area2 = stats2[:, cv2.CC_STAT_WIDTH] * stats2[:, cv2.CC_STAT_HEIGHT]
top2_idx2 = area2.argsort()[-3:-1]  # 2番目、3番めに面積が大きいラベル (1番目は背景なので除く)

# オブジェクト情報を項目別に抽出
data2 = label2[2]
center2 = label2[3]

# ラベリング結果書き出し用に二値画像をカラー変換
color_src2 = cv2.cvtColor(fgmask, cv2.COLOR_GRAY2BGR)

# オブジェクト情報を利用してラベリング結果を表示
for i in top2_idx2:
    # 各オブジェクトの外接矩形を赤枠で表示
    x0 = data2[i][0]
    y0 = data2[i][1]
    x1 = data2[i][0] + data2[i][2]
    y1 = data2[i][1] + data2[i][3]
    cv2.rectangle(color_src2, (x0, y0), (x1, y1), (0, 0, 255))

    # 各オブジェクトの重心座標をに黄文字で表示
    cv2.putText(color_src2, "Y: " + str(int(y1)), (x1 - 10, y1 + 30), cv2.FONT_HERSHEY_PLAIN, 1, (0, 255, 255))



cv2.imshow('y',color_src2)



この下からコートの画像に値を点で表したい！！！！！！！！！！！！！！！


midori=cv2.imread("coat2.png")




！！！！！！！！！！！新たにここを追加！！！！！！！！！！
この座標部分をどう直したら良いかがわかりません！！！！！！！！！！！
color_src3 = cv2.cvtColor(coat, cv2.COLOR_GRAY2BGR)
coat3=cv2.circle(color_src3,(int(center[i][0]),int(top2_idx2(y1))), 63, (0,0,255), -1)




cv2.waitKey(0)
cv2.destroyAllWindows()

上記の２箇所のラベリング処理の部分のputtextでx座標とy座標を渡しているので
その値を新たな　zero1.pngの画像に点として表示したいです

そして
このようなエラーが出ました

    coat3=cv2.circle(color_src3,(int(center[i][0]),int(top2_idx2(y1))), 63, (0,0,255), -1)
IndexError: index 11 is out of bounds for axis 0 with size 6

Answer 1

2020-01-05追補2

上 (重心X,矩形Y)を青丸で表示
下 根拠を図示

import cv2
import numpy as np
from matplotlib import pyplot as plt



src= cv2.imread("kuro1.png")



#ユニフォーム抽出
dst1 = cv2.inRange(src, (3,3,3), (10,8,10))
dst2 = cv2.blur(dst1, ksize=(5,5))
ret,thresh1 = cv2.threshold(dst2,127,255,cv2.THRESH_BINARY)

# ラベリング処理（x座標）
label = cv2.connectedComponentsWithStats(thresh1)

stats = label[2]
area = stats[:, cv2.CC_STAT_WIDTH] * stats[:, cv2.CC_STAT_HEIGHT]
top2_idx = area.argsort()[-3:-1]  # 2番目、3番めに面積が大きいラベル (1番目は背景なので除く)

# オブジェクト情報を項目別に抽出
data = label[2]
center = label[3]

# ラベリング結果書き出し用に二値画像をカラー変換
color_src = cv2.cvtColor(thresh1, cv2.COLOR_GRAY2BGR)

# オブジェクト情報を利用してラベリング結果を表示
lst_pos = []
for i in top2_idx:
    # 各オブジェクトの外接矩形を赤枠で表示
    x0 = data[i][0]
    y0 = data[i][1]
    x1 = data[i][0] + data[i][2]
    y1 = data[i][1] + data[i][3]
    cv2.rectangle(color_src, (x0, y0), (x1, y1), (0, 0, 255))

    # 各オブジェクトの重心座標をに黄文字で表示
    cv2.putText(color_src, "X: " + str(int(center[i][0])), (x1 - 10, y1 + 15), cv2.FONT_HERSHEY_PLAIN, 1, (0, 255, 255))

    # 追加コード: 重心座標を書き出す
    # img = thresh1[y0:y1,x0:x1]
    # lst_pos.append((y0,y1,x0,x1,int(center[i][0])))
    lst_pos.append((int(center[i][0]),int(center[i][1])))
    

# x座標の位置
cv2.imshow('x',color_src)

#「背景」となる画像の取り込み（グレースケール）
img_src01 = cv2.imread("zero1.png", 0)

#「差分」をもった画像の取り込み（グレースケール）
img_src02 = cv2.imread("kuro1.png", 0)

#「背景差分」計算用オブジェクトの作成　
# bgObj = cv2.bgsegm.createBackgroundSubtractorMOG()
# 修正点: ただただ手元のOpenCVでエラーが出ないようにするための目的で変更。
bgObj = cv2.createBackgroundSubtractorMOG2()

#差分となっている「前景領域」に対してマスクをかける
fgmask = bgObj.apply(img_src01)
fgmask = bgObj.apply(img_src02)

# ラベリング処理(y座標)
label2 = cv2.connectedComponentsWithStats(fgmask)

### 追加した部分1
stats2 = label2[2]
area2 = stats2[:, cv2.CC_STAT_WIDTH] * stats2[:, cv2.CC_STAT_HEIGHT]
top2_idx2 = area2.argsort()[-3:-1]  # 2番目、3番めに面積が大きいラベル (1番目は背景なので除く)

# オブジェクト情報を項目別に抽出
data2 = label2[2]
center2 = label2[3]

# ラベリング結果書き出し用に二値画像をカラー変換
color_src2 = cv2.cvtColor(fgmask, cv2.COLOR_GRAY2BGR)

# 追加コード: 適当に追加
img_col = cv2.imread("zero1.png")

# オブジェクト情報を利用してラベリング結果を表示
for i in top2_idx2:
    # 各オブジェクトの外接矩形を赤枠で表示
    x0 = data2[i][0]
    y0 = data2[i][1]
    x1 = data2[i][0] + data2[i][2]
    y1 = data2[i][1] + data2[i][3]
    cv2.rectangle(color_src2, (x0, y0), (x1, y1), (0, 0, 255))

    # 各オブジェクトの重心座標をに黄文字で表示
    cv2.putText(color_src2, "Y: " + str(int(y1)), (x1 - 10, y1 + 30), cv2.FONT_HERSHEY_PLAIN, 1, (0, 255, 255))

    ##############################################
    # 内外判定
    poly = np.array([[int(x0),int(y0)], [int(x0),int(y1)], [int(x1),int(y1)], [int(x1),int(y0)]])
    for j in range(len(lst_pos)):

        if cv2.pointPolygonTest( poly, lst_pos[j], False)== 1:
            # print("in")
            # 青で描画
            img_col = cv2.circle(img_col,(lst_pos[j][0],y1), 10, (255,0,0), 2)
        # else:
        #     print("out")
    ##############################################

cv2.imshow('img_col',img_col)

cv2.imshow('y',color_src2)
##
##
### この下からコートの画像に値を点で表したい！！！！！！！！！！！！！！！
##
##midori=cv2.imread("coat2.png")
##
### ！！！！！！！！！！！新たにここを追加！！！！！！！！！！
### この座標部分をどう直したら良いかがわかりません！！！！！！！！！！！
##
##
### 変更点: coatが未定義のため適当に変更
### color_src3 = cv2.cvtColor(coat, cv2.COLOR_GRAY2BGR)
### coat3=cv2.circle(color_src3,(int(center[i][0]),int(top2_idx2(y1))), 63, (0,0,255), -1)
##
### 変更点: 画像を適当に読み込み
### img_col = cv2.imread("zero1.png")
##img_col2 = img_col.copy()
##
### 変更点: Centerを赤で描画
##for i in range(len(center)):
##    img_col2 = cv2.circle(img_col2,(int(center[i][0]),int(center[i][1])), 3, (0,0,255), -1)
##
##
### 変更点: 矩形の頂点を緑で描画
##for i in top2_idx2:
##    x0 = data2[i][0]
##    y0 = data2[i][1]
##    x1 = data2[i][0] + data2[i][2]
##    y1 = data2[i][1] + data2[i][3]
##    img_col2 = cv2.circle(img_col2,(int(x0),int(y0)), 3, (0,255,0), -1)
##    img_col2 = cv2.circle(img_col2,(int(x0),int(y1)), 3, (0,255,0), -1)
##    img_col2 = cv2.circle(img_col2,(int(x1),int(y0)), 3, (0,255,0), -1)
##    img_col2 = cv2.circle(img_col2,(int(x1),int(y1)), 3, (0,255,0), -1)
##
### cv2.imshow("img_col",img_col)
##
### 変更点: Centerのデバグ用
##img_merged = cv2.addWeighted(img_col2, 0.8, color_src, 0.2, 0)
##cv2.imshow("img_merged",img_merged)

cv2.imwrite("img_col2_merged.png",np.vstack((img_col,img_merged)))
cv2.waitKey(0)
cv2.destroyAllWindows()

---

2020-01-05追補1

この画像にプログラムに書かれているラベリング処理を２回した時に求めた
x座標とy座標の値を赤い点として表示させたいです

赤い点と書いてありますがわかりやすくばつ印などでも良いです

プログラム全体の意図が良く掴めていませんが、「座標をプロットする」という点において、希望箇所（！！！の辺り）に対応コードを追加したため、これで解決できると思います。

また、赤い点、とありますが、Centerと矩形と両方とも位置を示すラベルと思いますので、色を分けて表示させました。
※OpenCVで描画する際にバツより丸の方が実装しやすいため丸で実装しています。

赤: Center
緑: 矩形の頂点

コメントの

for i in top2_idx2の部分を
jに変えてみることで解決したりするのでしょうか？

については、今回は二重ループの出番がないので使用していません。
二重ループでiとは別の変数を使いたいときにjの出番があります。

今回は！！！の部分にfor文を追加しましたが、ラベル作成時にこの処理もまとめてやってしまった方がfor文が少なくなりますので、より早くそして、バグの発生する余地の少ないコードになると思います。

import cv2
import numpy as np
from matplotlib import pyplot as plt

src= cv2.imread("kuro1.png")

#ユニフォーム抽出
dst1 = cv2.inRange(src, (3,3,3), (10,8,10))
dst2 = cv2.blur(dst1, ksize=(5,5))
ret,thresh1 = cv2.threshold(dst2,127,255,cv2.THRESH_BINARY)

# ラベリング処理（x座標）
label = cv2.connectedComponentsWithStats(thresh1)

stats = label[2]
area = stats[:, cv2.CC_STAT_WIDTH] * stats[:, cv2.CC_STAT_HEIGHT]
top2_idx = area.argsort()[-3:-1]  # 2番目、3番めに面積が大きいラベル (1番目は背景なので除く)

# オブジェクト情報を項目別に抽出
data = label[2]
center = label[3]

# ラベリング結果書き出し用に二値画像をカラー変換
color_src = cv2.cvtColor(thresh1, cv2.COLOR_GRAY2BGR)

# オブジェクト情報を利用してラベリング結果を表示
for i in top2_idx:
    # 各オブジェクトの外接矩形を赤枠で表示
    x0 = data[i][0]
    y0 = data[i][1]
    x1 = data[i][0] + data[i][2]
    y1 = data[i][1] + data[i][3]
    cv2.rectangle(color_src, (x0, y0), (x1, y1), (0, 0, 255))

    # 各オブジェクトの重心座標をに黄文字で表示
    cv2.putText(color_src, "X: " + str(int(center[i][0])), (x1 - 10, y1 + 15), cv2.FONT_HERSHEY_PLAIN, 1, (0, 255, 255))

cv2.imshow('x',color_src)

#「背景」となる画像の取り込み（グレースケール）
img_src01 = cv2.imread("zero1.png", 0)

#「差分」をもった画像の取り込み（グレースケール）
img_src02 = cv2.imread("kuro1.png", 0)

#「背景差分」計算用オブジェクトの作成　
# bgObj = cv2.bgsegm.createBackgroundSubtractorMOG()
# 修正点: ただただ手元のOpenCVでエラーが出ないようにするための目的で変更。
bgObj = cv2.createBackgroundSubtractorMOG2()

#差分となっている「前景領域」に対してマスクをかける
fgmask = bgObj.apply(img_src01)
fgmask = bgObj.apply(img_src02)

# ラベリング処理(y座標)
label2 = cv2.connectedComponentsWithStats(fgmask)

### 追加した部分1
stats2 = label2[2]
area2 = stats2[:, cv2.CC_STAT_WIDTH] * stats2[:, cv2.CC_STAT_HEIGHT]
top2_idx2 = area2.argsort()[-3:-1]  # 2番目、3番めに面積が大きいラベル (1番目は背景なので除く)

# オブジェクト情報を項目別に抽出
data2 = label2[2]
center2 = label2[3]

# ラベリング結果書き出し用に二値画像をカラー変換
color_src2 = cv2.cvtColor(fgmask, cv2.COLOR_GRAY2BGR)

# オブジェクト情報を利用してラベリング結果を表示
for i in top2_idx2:
    # 各オブジェクトの外接矩形を赤枠で表示
    x0 = data2[i][0]
    y0 = data2[i][1]
    x1 = data2[i][0] + data2[i][2]
    y1 = data2[i][1] + data2[i][3]
    cv2.rectangle(color_src2, (x0, y0), (x1, y1), (0, 0, 255))

    # 各オブジェクトの重心座標をに黄文字で表示
    cv2.putText(color_src2, "Y: " + str(int(y1)), (x1 - 10, y1 + 30), cv2.FONT_HERSHEY_PLAIN, 1, (0, 255, 255))

cv2.imshow('y',color_src2)


# この下からコートの画像に値を点で表したい！！！！！！！！！！！！！！！

midori=cv2.imread("coat2.png")

# ！！！！！！！！！！！新たにここを追加！！！！！！！！！！
# この座標部分をどう直したら良いかがわかりません！！！！！！！！！！！


# 変更点: coatが未定義のため適当に変更
# color_src3 = cv2.cvtColor(coat, cv2.COLOR_GRAY2BGR)
# coat3=cv2.circle(color_src3,(int(center[i][0]),int(top2_idx2(y1))), 63, (0,0,255), -1)

# 変更点: 画像を適当に読み込み
img_col = cv2.imread("zero1.png")

# 変更点: Centerを赤で描画
for i in range(len(center)):
    img_col = cv2.circle(img_col,(int(center[i][0]),int(center[i][1])), 3, (0,0,255), -1)


# 変更点: 矩形の頂点を緑で描画
for i in top2_idx2:
    x0 = data2[i][0]
    y0 = data2[i][1]
    x1 = data2[i][0] + data2[i][2]
    y1 = data2[i][1] + data2[i][3]
    img_col = cv2.circle(img_col,(int(x0),int(y0)), 3, (0,255,0), -1)
    img_col = cv2.circle(img_col,(int(x0),int(y1)), 3, (0,255,0), -1)
    img_col = cv2.circle(img_col,(int(x1),int(y0)), 3, (0,255,0), -1)
    img_col = cv2.circle(img_col,(int(x1),int(y1)), 3, (0,255,0), -1)

cv2.imshow("img_col",img_col)

# 変更点: Centerのデバグ用
# img_merged = cv2.addWeighted(img_col, 0.8, color_src, 0.2, 0)
# cv2.imshow("img_merged",img_merged)

cv2.waitKey(0)
cv2.destroyAllWindows()

---

上記の２箇所のラベリング処理の部分のputtextでx座標とy座標を渡しているので
その値を新たな　zero1.pngの画像に点として表示したいです

説明文に書いてある通り、以下のXY座標を使えば、表示させたい点の中心部分を得られますね。

座標	X	Y
左上	x0	y0
右下	x1	y1

ですので、四角形の頂点に点をつけたいのであれば、

	左	右
上	(x0,y0)	(x1,y0)
下	(x0,y1)	(x1,y1)

が、それぞれの頂点に対応する座標です。

あとは、cv2.circle()の説明に従って点を描画すればOKです。
※circleとありますが、直径を1(ピクセル)にすれば点として描画できます。

cv2.circle(img,(447,63), 63, (0,0,255), -1)は、
画像データimgのx:447,y:63に、直径63で、赤(0,0,255)の円を太さ-1(塗りつぶし)で描画する、という指示になります。

また、ラベル要素をfor文で位置を取得する際に、同時に描画処理をしてしまえば別途ラベルごとにfor文をもう一度書き加える必要はなくスッキリさせられます。

Comments

[0126tami]

四角形の頂点とはどの四角形のことでしょうか？？
１つ目のラベリング処理でx座標、２つ目のラベリング処理でy座標だけを取ってきて描画したいので
値としては１つだけに定まるのですが

circleのアドバイス参考にさせていただきます！

[退会済みユーザー]

> 四角形の頂点とはどの四角形のことでしょうか？？
「各オブジェクトの外接矩形を赤枠で表示」の赤枠の外接矩形です。

夜に対応できたら!!!マークの部分も見ますね。

[退会済みユーザー]

> 夜に対応できたら!!!マークの部分も見ますね。
コード中のfor i in top2_idx2が終わった時のiの値が格納されたまま最後のエラーのところまで処理が進んでいます。

ですので、center[11][0]をやろうとしてエラーが起きているようです。
恐らくcenter[i]のiは0～5の値しか入らないはずです。

最後のエラーが起きている部分の変数をprint()で確認しながら進めるのがよさそうです。

"kuro1.png"、img_src01、img_src02、midoriがあればこちらでも再現できますので、もし回答が必要でしたら質問内容の部分にどの画像がどの変数に使うものかわかる状態で添付ください。

[0126tami]

ありがとうございます
画像も載せてみたのでぜひアドバイスお願いします

for i in top2_idx2の部分を
jに変えてみることで解決したりするのでしょうか？

[0126tami]

そしてラベリング処理での結果については　imshowで表示されると思います
その中で使いたいのは　１つ目のラベリング処理の画像のx座標と２つ目のラベリング処理の画像のy座標です

[退会済みユーザー]

(１つ目のラベリング処理の画像のx座標,２つ目のラベリング処理の画像のy座標)に円を描きたい、ということですか？

[0126tami]

円を書きたいというか　座標がここだと示したいということです
今回はそれがわかりやすいように円でも良いかなという意味です！

[0126tami]

そして
２つのラベリング処理においてx座標とy座標で１つの円が表示されるようにしたいのです
ここでは人間一人に対して２回ラベリング処理します
まず重心のx座標そして足元のy座標という感じです
それを元の人間のいない画像にx座標とy座標がわかるのでそこの円としてプロットしたいという感じなのです。。。

[0126tami]

プログラムの全体の目的としては
人の位置を画像処理で求めて点の位置で見える化しようみたいな感じの目的なのです
わかりにくくてごめんなさい
ご協力とても感謝しています
ぜひお力をお貸しくださいませ

[0126tami]

なんどもありがとうございました！！