Search code examples
pythonopencvpointlines

Draw line between two given points (OpenCV, Python)

I am struggling with this problem for an hour by now...

I have an image with a rectangle inside:

rect

This is the code I wrote to find the points for the corners:

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)

This is the result:

rst

As you can see, it works perfect. What I want is to draw a line along the upper/lower points (x1,x2 - x3,x4).

What I produced since now is this...

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

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

But it doesn't work.

Any idea ?

The result should be like this:

out

The two lines must pass along the coordinates of the points. print(points[0]) above give the next output, as example:

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

Thanks

Solution

  • So first of all, let'S look at your print, it says that points[0] is 

    [[561 168]]

    but opencv point is like 

    (561, 168)

    You can unpack it like you did with the circle and then do the tuple

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

    or you can use

    tuple(points[0].ravel())

    or

    tuple(points[0][0])

    Edit

    You wanted from one side of the screen to the other one, that is also easy. What you need to do is change the x value to be 0 in one point and column value in the other point. I think the easiest way is to do it like this:

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

    Two things to note here:

    1. As you can see I did not care about the second point, since I assumed that it will be in the same horizontal line, if not, it will get a little bit more complicated, but not hard.

    2. img.shape returns the tuple with the image details as (rows, cols, channels), since we need cols we took [1].