Opencv cv2.lineがでてこない。

Question

今opencvを使って線を表したいのですができません。言語はpython3です。
３重配列の要素を繋げたいのですがエラーはでないのに動画を再生すると
線が出て来ますせん。どこがいけないのでしょうか？
求めている結果はこんな感じです。

そして下がコードです。

def route(n_id):

    if(len(n_id) > 1):
        for i in range(1,len(n_id)):
            for a in range(0,len(n_id[i])):
                if(len(n_id[i][a]) > 2):
                    cv2.line(frame,(n_id[i][a][0],n_id[i][a][1]),(n_id[i][a+1][0],n_id[i][a+1][1]),(0,20,255),2)

配列はこんな感じで座標が保存されて行きます。

Comments

[can110]

３重配列の各要素やコード上の各ループの説明、またできれば単純なデータ例を示すと回答得られやすいかと思います。

[Cielwoo]

配列の各要素はまず、認識した物体の座標であり、同じ物体の座標を蓄積しています。そしてループは、蓄積された座標を一つ一つ処理するためのものです。最初の[0,0]は無視して結構です。

[can110]

n_id[i][a][0]の[i],[a],[0]or[1]は何を表しているのか、また表示データも「どんな順番で何を出力」しているのかが不明です。

[Cielwoo]

n_id [各物体] [物体の座標群] [座様(a,b)]こんな感じです。表示データは[ [[ ]], [[ ]], [[ ]] ] -> [[ ]]は一つの物体の座標群でしす。座標は順番ずつ入って行きます。表示データは配列の中身を見せるためにアップしました。

Answer 1

具体的な元データ配列構造がいまひとつ理解できていないかもしれませんが、以下のような感じで描画できないでしょうか？
分かりやすさを優先し、あえて冗長なコードとしています。

import cv2
import numpy as np
(W,H) = (640,480)
img = np.zeros((H,W,3), np.uint8) # 幅640x高さ480

# [物体][点群][xory座標]
n_id = np.array(
    [[[100,100],[100,300],[200,300],[200,100]],
    [ [300,200],[300,300],[400,300],[400,200]],
    [ [500,300],[500,400],[600,400],[600,300]]])

for obj_idx in range(len(n_id)): # 物体毎
    obj = n_id[obj_idx] # １物体（領域）
    for pt_idx in range(len(obj)): # 領域を構成する点群毎
        pt_st = obj[pt_idx] # 始点 [0]=x, [1]=y 座標

        # 最後は最初の点と結ぶ
        pt_idx += 1
        if pt_idx >= len(obj):
            pt_idx = 0

        pt_ed = obj[pt_idx] # 終点

        # 始点→終点へ赤線を引く
        cv2.line(img, (pt_st[0],pt_st[1]), (pt_ed[0],pt_ed[1]), (0, 0, 255), obj_idx+1)

cv2.namedWindow("img")
cv2.imshow("img", img)
cv2.waitKey(0)
cv2.destroyAllWindows()

Comments

[Cielwoo]

線が出て来ました！ でも、これを最初と最後の点を結ばないとしたらどうすればいいですか？

[can110]

よかったです。
>これを最初と最後の点を結ばないとしたらどうすればいいですか？
最初と最後の点を結ぶ線は最後に引いているので、点群のループを１回少なくすればよいです。

[Cielwoo]

だいぶ時間が経っていますが、そのコードでfor文を降順でやるにはどうすればよいでしょうか？
reverseを使ってみたのですができません。