OpenCV – 連結成分のラベリングを行う cv2.connectedComponents の使い方

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

# 画像を読み込む。
img = cv2.imread("sample.jpg")

# グレースケールに変換する。
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 2値化する
ret, bin_img = cv2.threshold(gray, 20, 255, cv2.THRESH_BINARY)

# 連結成分のラベリングを行う。
n_labels, labels = cv2.connectedComponents(bin_img)

# ラベル数
print("number of labels:", n_labels)

fig, ax = plt.subplots(figsize=(7, 7))
ax.imshow(labels)

plt.show()

# 連結成分のラベリングを行う。
retval, labels, stats, centroids = cv2.connectedComponentsWithStats(bin_img)

for i, row in enumerate(stats):
    print(f"ラベル {i}")
    print(f"* 左上の座標: ({row[cv2.CC_STAT_LEFT]}, {row[cv2.CC_STAT_TOP]})")
    print(f"* 大きさ: ({row[cv2.CC_STAT_WIDTH]}, {row[cv2.CC_STAT_HEIGHT]})")
    print(f"* 面積: {row[cv2.CC_STAT_AREA]}")

for i, row in enumerate(centroids):
    print(f"ラベル: {i}, 座標: ({row[0]:.2f}, {row[1]:.2f})")