複数の図形における重心計算

Question

二値化した画像中の複数の図形の重心計算を行いたいです。

まずは１つの図形の重心計算を行うために下記のようなプログラムを書いたのですが、複数の図形それぞれにおいて重心を求めるとなると、どこでfor文などを用いて繰り返し処理をすればいいのかがわかりません。

アドバイスをいただけると助かります。

# 画像読み込み
img = cv2.imread("test.jpg")
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
###############画像の二値化＆画像から重心求める##############
ret,thresh = cv2.threshold(img, 127, 255, 0)
img1 = np.array(thresh)
img2 = cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY)

mu = cv2.moments(img2,False)
x,y= mu["m10"]/mu["m00"] , mu["m01"]/mu["m00"]
plt.plot(x,y,marker='.')
plt.imshow(img2, cmap = "gray")
plt.colorbar()
plt.show()

Answer 1

以下のようにfindContoursで輪郭抽出して、それぞれの重心を計算すればよいかと思います。

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

img = cv2.imread("test.jpg")
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret,thresh = cv2.threshold(img, 127, 255, 0)
img1 = np.array(thresh)
img2 = cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY)

contours, _ = cv2.findContours(img2, cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
for c in contours:
    M = cv2.moments(c)
    x = int(M["m10"] / M["m00"])
    y = int(M["m01"] / M["m00"])
    plt.plot(x,y,marker='.')

plt.imshow(img2, cmap = "gray")
plt.colorbar()
plt.show()

Comments

[runrun5]

M = cv2.moments(c)　の引数にfor文の数を入れれることを初めて知りました。
勉強になりました。ありがとうございます。

[can110]

for文の中にprint(c)入れたら分かりますが、数というかcontourですね。