表示した外接矩形の詳細情報を求める

Question

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

#三角(赤)
def find_rect_of_target_color1(image):
    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV_FULL)
    h = hsv[:, :, 0]
    s = hsv[:, :, 1]
    v = hsv[:, :, 2]
    mask = np.zeros(h.shape, dtype=np.uint8)
    mask[(h > 240) & (s > 100) & (v > 100)] = 251
    #輪郭抽出
    _, contours, _ = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
    rects1 = []
    for contour in contours:
      approx = cv2.convexHull(contour)
      rect = cv2.boundingRect(approx)
      rects1.append(np.array(rect))
    return rects1

#半円(黄)
def find_rect_of_target_color2(image):
    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV_FULL)
    h = hsv[:, :, 0]
    s = hsv[:, :, 1]
    v = hsv[:, :, 2]
    mask = np.zeros(h.shape, dtype=np.uint8)
    mask[((h > 20) & (h < 40)) & (s > 100) & (v > 200)] = 30
    #輪郭抽出
    _, contours, _ = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
    rects2 = []
    for contour in contours:
        approx = cv2.convexHull(contour)
        rect = cv2.boundingRect(approx)
        rects2.append(np.array(rect))
    return rects2

if __name__ == "__main__":
    img = cv2.imread('./sample.jpg')

    #赤
    rects1 = find_rect_of_target_color1(img)
    if len(rects1) > 0:
        rect = max(rects1, key=(lambda x: x[2] * x[3]))
        cv2.rectangle(img, tuple(rect[0:2]),
        tuple(rect[0:2] + rect[2:4]), (0, 0, 255), 2)
                                     # B  G   R
    #黄
    rects2 = find_rect_of_target_color2(img)
    if len(rects2) > 0:
        rect = max(rects2, key=(lambda x: x[2] * x[3]))
        cv2.rectangle(img, tuple(rect[0:2]),
        tuple(rect[0:2] + rect[2:4]), (0, 255, 255), 2)

cv2.imshow('findrect', img)
k = cv2.waitKey(0)
if k == ord('q'):
    cv2.destroyAllWindows()

上記は画像をHSV変換して特定色の外接矩形を表示するプログラムになります。
表示した外接矩形の縦の長さと横の長さ、面積とアスペクト比を計算するコードを追加したいのですが、いまいちわかりません。
様々なサイトを参考にして実行してみましたがpython初心者なものでコンパイルエラーが多数出てしまいます。
どなたか解決方法がわかる方いらっしゃいましたらご教授願います。

Answer 1

ソースコードを見る限り、各外接矩形にてrect[2]が幅、rect[3]が高さのようなので

w,h = rect[2],rect[3] # 幅と高さ
a = w*h # 面積
ar = w/h# アスペクト比

でよいかと思います。

Comments

[xxxxxxx]

できました！それと求めたそれらの値を画像内の外接矩形の下に表示させたいのですが、矩形のどこかの座標がわからなくて困っています。。

[can110]

rect[1]が矩形の左上のy座標のようなので、それにhを足すと矩形下辺のy座標になります。x座標はrect[0]のままでよいでしょう。
実際に紙に描いてみると理解しやすいかと思います。

[xxxxxxx]

大変わかりやすい回答ありがとうございました！
なんとか思った通りにできました。。。