輪郭に対する外接矩形の情報取得について

Question

実現したいこと

画像中の輪郭抽出を行い、それに対して外接矩形を描画するプログラムを作っています。
検出精度を上げるために、画像を分割処理を行いそのひとつひとつの画像に対して輪郭抽出、外接矩形の描画を行っています。

最終的に、描画した外接矩形の高さ、幅を取得したいです。

下記ソースコードのout_imgに分割画像１枚ごとの情報を格納しており、rectsに外接矩形の情報を格納しています。
分割された複数画像のうち一枚の画像の中で描画されている外接矩形の情報を取得する方法を教えていただきたいです。
（例. 300枚に分割した画像のうち、151番目の画像に描画された外接矩形の情報の取得）

サンプル画像

発生している問題・エラーメッセージ

エラーメッセージ

該当のソースコード

↓高さ、幅を求めるコード

def get_rect_info(pt):
    
    #長方形の中心
    center = (pt[0] + pt[2]) / 2
    
    #長さが異なる2つの辺の長さをそれぞれ計算する。
    vec1 = pt[0] - pt[1]
    vec2 = pt[1] - pt[2]
    vec1_len = np.linalg.norm(vec1)
    vec2_len = np.linalg.norm(vec2)
    
    #長辺が幅、短辺が高さとする。
    if vec1_len < vec2_len:
        width, height = vec2_len, vec1_len
        vecw = vec2
    else:
        width, height = vec1_len, vec2_len
        vecw = vec1
        
    return {'center': center, 'size': (width, height)}

↓輪郭抽出⇒外接矩形の描画を行うコード

import cv2
import numpy as np

def DrawCont(img):
    
    # 画像の読み込み
    #img = cv2.imread('test.jpg')

    # リサイズ
    #img = cv2.resize(img, dsize=None, fx=0.2, fy=0.2)

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

    # 二値化
    ret, binary = cv2.threshold(gray, 3, 255, cv2.THRESH_BINARY_INV)

    #輪郭抽出
    _, contours, _ = cv2.findContours(binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

    # 面積が小さい輪郭削除
    area = img.shape[0] * img.shape[1]
    contours = list(filter(lambda cnt: 1 < cv2.contourArea(cnt), contours))

    # 外接矩形に変換
    #global rects
    rects = [cv2.boxPoints(cv2.minAreaRect(cnt)).astype(int) for cnt in contours]
    #print(rects)

    # 外接矩形を描画
    cv2.drawContours(img, rects, -1, (0, 255, 0), 2)

↓分割処理を行うコード

import cv2
import numpy as np


img = cv2.imread('C:/Users/ito/Anaconda3/envs/Sample/pic/testimg/IMG_0723.jpg')
#img = cv2.resize(img, (1275, 997))
    
size = 200
v_size = img.shape[0] // size * size
h_size = img.shape[1] // size * size
img = img[:v_size, :h_size]
 
v_split = img.shape[0] // size
h_split = img.shape[1] // size
out_img = []

[out_img.extend(np.hsplit(h_img, h_split)) for h_img in np.vsplit(img, v_split)]

out_img = np.array(out_img)

i=0
for i, lst in enumerate(out_img):
    #関数DrawContの呼び出し（輪郭抽出→外接矩形の描画）
    DrawCont(out_img[i])
    
    #分割処理後に外接矩形描画した画像を保存
    cv2.imwrite('C:/Users/ito/Anaconda3/envs/Sample/pic/split/I0723/split' + str(i) +'.png', out_img[i])

    
    
    #######################　矩形の長さを求める　　##########################
get_rect_info(out_img[157])

試したこと

関数get_rect_infoの引数を変えてみる

開発環境

windows7 professional
python3.6.5
opencv3.3.1

Answer 1

rects = [cv2.boxPoints(cv2.minAreaRect(cnt)).astype(int) for cnt in contours]

の直後に以下を入れると、外接矩形の情報が取得できると思います。

for rect in rects:
    print(get_rect_info(rect))

Comments

[Taka11]

ご回答いただきありがとうございます！