在两个给定点之间画线（OpenCV，Python）

Question

我现在正在努力解决这个问题一小时......

我的图像里面有一个矩形：

这是我为寻找角落点而编写的代码：

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)

结果如下：

正如您所看到的，它完美无缺。我想要的是沿着上/下点（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)

但它不起作用。

有什么想法吗？

结果应该是这样的：

两条线必须沿着点的坐标传递。上面的print(points[0])给出了下一个输出，例如：

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

由于

Answer 1

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)

Answer 2

首先，让我们看看你的印刷品，它说点[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)

这里有两点需要注意：

1. 正如你所看到的那样，我并不关心第二点 它会在同一水平线上，如果没有，它会得到一个 有点复杂，但并不难。

2. img.shape返回 因为我们，图像细节为（行，列，通道）的元组 需要我们采取的cols [1]。