web-dev-qa-db-ja.com

与えられた2つのポイントの間に線を引きます(OpenCV、Python)

私は今までにこの問題に1時間苦労しています...

内部に長方形の画像があります。

rect

これは、コーナーのポイントを見つけるために私が書いたコードです。

_import cv2
import numpy as np


img = cv2.imread('rect.png')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
gray = np.float32(gray)

points = cv2.goodFeaturesToTrack(gray, 100, 0.01, 10)
points = np.int0(points)

for point in points:
    x, y = point.ravel()
    cv2.circle(img, (x, y), 3, (0, 255, 0), -1)

print(points[0])
print(points[1])
print(points[2])
print(points[3])

cv2.imshow('img', img)
cv2.waitKey(0)
cv2.imwrite('rect.png', img)
_

結果は次のとおりです。

rst

ご覧のとおり、完璧に機能します。私が欲しいのは、上下の点(x1、x2-x3、x4)に沿って線を引くことです。

今から作ったのはこれ….

_cv2.line(img, (points[0]), (points[1]), (0, 255, 0), thickness=3, lineType=8)

cv2.imshow('img', img)
cv2.waitKey(0)
_

しかし、それは機能しません。

何か案が ?

結果は次のようになります。

out

2本の線は点の座標に沿って通過する必要があります。上記のprint(points[0])は、例として次の出力を示します。

_[[561 168]]
[[155 168]]
[[561  53]]
[[155  53]] 
_

ありがとう

4
BlueTrack

それで、まず最初に、あなたのプリントを見てみましょう、それはpoints [0]がであると言います

[[561 168]]

しかし、opencvポイントは次のようなものです

(561, 168)

サークルで行ったように開梱してから、タプルを実行できます

x, y = points[0].ravel()
(x,y)

またはあなたが使用することができます

Tuple(points[0].ravel())

または

Tuple(points[0][0])

編集

画面の片側から反対側に移動したかったのですが、これも簡単です。あなたがする必要があるのは、一方のポイントでx値を0に、もう一方のポイントで列値に変更することです。最も簡単な方法は、次のようにすることだと思います。

y = points[0].ravel()[1]
cv2.line(img, (0, y), (img.shape[1], y), (0, 255, 0), thickness=3, lineType=8)

ここで注意すべき2つのこと:

  1. ご覧のとおり、2番目のポイントは同じ水平線上にあると想定していたので、気にしませんでした。そうでない場合は、少し複雑になりますが、難しくはありません。

  2. img.shapeは、取得した列が必要なため、画像の詳細を(行、列、チャネル)としてタプルを返します[1]。

3
api55
points = cv2.goodFeaturesToTrack(gray, 100, 0.01, 10)
points = np.int0(points).reshape(-1,2)

for point in points:
    x, y = point.ravel()
    cv2.circle(img, (x, y), 3, (0, 255, 0), -1)


y1 = min(points[:,1])
y2 = max(points[:,1])

## small and big enough 
cv2.line(img, (0, y1), (1000, y1), (0, 255, 0), thickness=3, lineType=8)
cv2.line(img, (0, y2), (1000, y2), (0, 255, 0), thickness=3, lineType=8)
3
Kinght 金