複数図形 重心計算 Python

Question

前提・実現したい

findCounter関数を用い、複数の図形に対し重心計算を行いたいのですが、どこにfor文を入れて繰り返し処理を行えばよいかわかりません。
図形が複数存在する場合、複数の重心が計算されず、１つの重心が計算されます。

該当のソースコード

import cv2
import numpy as np
import csv

threshold=0
lower_S=np.array([0,0,20]) 
upper_S=np.array([180,83,255])
lower_V=np.array([0,0,94])
upper_V=np.array([180,255,255])

picture="026537.jpg"
img= cv2.imread( picture)
# hsv色空間に変換
hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
#mask処理
img_mask1=cv2.inRange(hsv,lower_S,upper_S)
img_mask2=cv2.inRange(hsv,lower_V,upper_V)
img_mask=cv2.bitwise_or(img_mask1,img_mask2)
img_masked=cv2.bitwise_and(img,img,mask=img_mask)
#grayscale変換
dst = cv2.cvtColor(img_masked,cv2.COLOR_BGR2GRAY)
#輪郭検出
image,contours =cv2.findContours(dst,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
#面積計算
areas = np.array(list(map(cv2.contourArea,contours)))
max_idx = np.argmax(areas)
max_area = areas[max_idx]
result = cv2.moments(contours[max_idx])
x = int(result["m10"]/result["m00"])
y = int(result["m01"]/result["m00"])
coordination=([x],[y])
#CSVファイル作成
with open('Cordinations.csv','a',newline="") as f:
    writer=csv.writer(f)
    writer.writerow([x,y])
    f.close() 

補足情報（FW/ツールのバージョンなど）

使用言語:Python
作業環境:Visual Studio
OpenCV(4.4)

Answer 1

1つの輪郭を引数にとり、その重心 cx, cy を返す関数を作成し、map() で輪郭の一覧 contours に適用すればよいです。

あと、cv2.findContours() の返り値の受け取り型が間違っています。

- image,contours = cv2.findContours(dst, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+ contours, hierarchy = cv2.findContours(dst, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

• OpenCV - findContours で画像から輪郭を抽出する方法
• OpenCV - 輪郭の特徴分析について

修正後のコード

import cv2
import numpy as np
import csv

threshold = 0
lower_S = np.array([0, 0, 20])
upper_S = np.array([180, 83, 255])
lower_V = np.array([0, 0, 94])
upper_V = np.array([180, 255, 255])

picture = "sample.jpg"

img = cv2.imread(picture)

# hsv色空間に変換
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

# mask処理
img_mask1 = cv2.inRange(hsv, lower_S, upper_S)
img_mask2 = cv2.inRange(hsv, lower_V, upper_V)
img_mask = cv2.bitwise_or(img_mask1, img_mask2)
img_masked = cv2.bitwise_and(img, img, mask=img_mask)
# grayscale変換
dst = cv2.cvtColor(img_masked, cv2.COLOR_BGR2GRAY)

#################################################
# 以下修正したコード
#################################################

# 輪郭検出
contours, hierarchy = cv2.findContours(dst, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)


# 面積
contour = list(map(cv2.contourArea, contours))
print(contour)

# 重心
def calc_center(contour):
    # 輪郭のモーメントを計算する。
    M = cv2.moments(contour)
    if M["m00"] == 0:  # ゼロ除算回避
        return tuple(contour[0, 0])

    # モーメントから重心を計算する。
    cx = M["m10"] / M["m00"]
    cy = M["m01"] / M["m00"]

    return cx, cy


centers = list(map(calc_center, contours))
print(centers)

np.savetxt("centers.txt", centers, fmt="%.3f")