opencv python 輪郭の特徴点の座標

Question

輪郭抽出した後に，特徴点の座標が知りたいです。

import matplotlib.pyplot as plt
from matplotlib.patches import Polygon

img = cv2.imread("./pu1/1124_2_0.jpg")
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)

kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
binary = cv2.dilate(binary, kernel)
contours, hierarchy = cv2.findContours(binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

def draw_contours(ax, img, contours):
    ax.imshow(img)
    ax.set_axis_off()

    for i, cnt in enumerate(contours):
        if cv2.contourArea(cnt) < 1900:
            continue
        else :
            cnt = cnt.squeeze(axis=1)
            ax.add_patch(Polygon(cnt, color="b", fill=None, lw=2))
            ax.plot(cnt[:, 0], cnt[:, 1], "ro", mew=0, ms=4)
            ax.text(cnt[0][0], cnt[0][1], i, color="orange", size="20")
            print(contours)


fig, ax = plt.subplots(figsize=(8, 8))
draw_contours(ax, img, contours)

plt.savefig('./rinnkaku/.jpg')
コード

このコードで元画像から下のような画像が得られるのですが，

面積の小さな部分の輪郭の座標も抽出してしまいます。
一番大きな輪郭（ここでいう2ばん）のところだけの座標群を知りたいです。

どなたかよろしくお願いします。

pythonのversionは3.7.3で，
opencvのversionは4.1.1です。

Comments

[tiitoi]

findContours の返り値 contours がそのまま求めてる座標値ではないのでしょうか？

[reonald]

コードや質問内容を修正したので再度確認してもらえたら嬉しいです。

[tiitoi]

輪郭が描画されていない元画像を添付していただけますか

[reonald]

追加しました。

Answer 1

一番大きな輪郭（ここでいう2ばん）のところだけの座標群を知りたいです。

cv2.contourArea() で輪郭の面積を取得できるので、この関数を max() の key 引数に指定して、面積が最大の輪郭を取得する。

target_contour = max(contours, key=lambda x: cv2.contourArea(x))

これで取得できた輪郭は (点の数, 1, 2) の numpy 配列で、これが輪郭の点を構成する座標になっています。(画像の左上が原点 (0, 0) で単位はピクセルの画像座標系)

import cv2
import matplotlib.pyplot as plt
import numpy as np
from matplotlib.patches import Polygon


def draw_contours(ax, img, contours):
    ax.imshow(img)
    ax.set_axis_off()

    for i, cnt in enumerate(contours):
        cnt = cnt.squeeze(axis=1)
        ax.add_patch(Polygon(cnt, color="b", fill=None, lw=2))
        ax.plot(cnt[:, 0], cnt[:, 1], "ro", mew=0, ms=4)
        ax.text(cnt[0][0], cnt[0][1], i, color="orange", size="20")


img = cv2.imread("sample.jpg")

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

ret, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)

kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
binary = cv2.dilate(binary, kernel)

contours, hierarchy = cv2.findContours(
    binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE
)

# 一番面積が大きい輪郭を抽出する。
target_contour = max(contours, key=lambda x: cv2.contourArea(x))


fig, ax = plt.subplots(figsize=(8, 8))
draw_contours(ax, img, [target_contour])

# 輪郭を構成する点を CSV に保存する。
np.savetxt(
    "target_contour.csv", target_contour.squeeze(axis=1), delimiter=",", header="X,Y"
)

Answer 2

1.面積の小さいところ（これなら0-7の輪郭）を輪郭抽出したくない。

contourAreaを使って面積をチェックしてはじいてみて下さい。

2.赤点の座標が知りたいです。

draw_contoursの中で座標を取得しているコードを書いているように見受けられるので、ちょっと何が分からないのかつかめないです。