如何使用OpenCV和Python查找梯形的角点,类似于查找正方形的角点?

2024-06-26 14:52:44 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用下面的代码查找图像中最大正方形的角。我想做同样的事情,但如果正方形恰好弯曲(像梯形),它仍然可以找到形状的角。如何使用Python的OpenCV模块实现这一点

importedImage = 'shapes.png'
originalImg = cv.imread(importedImage)

#filters image bilaterally and displays it
bilatImg = cv.bilateralFilter(originalImg, 5, 175, 175)

#finds edges of bilaterally filtered image and displays it
edgeImg = cv.Canny(bilatImg, 75, 200)

#gets contours (outlines) for shapes and sorts from largest area to smallest area
contours, hierarchy = cv.findContours(edgeImg, cv.RETR_TREE, cv.CHAIN_APPROX_SIMPLE)
contours = sorted(contours, key=cv.contourArea, reverse=True)

#boxes in largest rectangle
rectangle = cv.minAreaRect(contours[0])
corners = cv.boxPoints(rectangle).astype(np.int32)

Tags: andimageitcv正方形shapesdisplaysrectangle
1条回答
网友
1楼 · 发布于 2024-06-26 14:52:44

再见

为了解决您的问题,您可以使用此代码段。我对新的部分留下了评论,以便更容易理解

import cv2
import numpy as np

importedImage = 'shapes.png'
originalImg = cv2.imread(importedImage)

#filters image bilaterally and displays it
bilatImg = cv2.bilateralFilter(originalImg, 5, 175, 175)

#finds edges of bilaterally filtered image and displays it
edgeImg = cv2.Canny(bilatImg, 75, 200)

#gets contours (outlines) for shapes and sorts from largest area to smallest area
contours, hierarchy = cv2.findContours(edgeImg, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
contours = sorted(contours, key=cv2.contourArea, reverse=True)

# drawing red contours on the image
for con in contours:
    cv2.drawContours(originalImg, con, -1, (0, 0, 255), 3)

# and double-checking the outcome
cv2.imshow("Contours check",originalImg)
cv2.waitKey()
cv2.destroyWindow("Contours check")

# find the perimeter of the first closed contour
perim = cv2.arcLength(contours[0], True)
# setting the precision
epsilon = 0.02*perim
# approximating the contour with a polygon
approxCorners = cv2.approxPolyDP(contours[0], epsilon, True)
# check how many vertices has the approximate polygon
approxCornersNumber = len(approxCorners)
print("Number of approximated corners: ", approxCornersNumber)

# can also be used to filter before moving on [if needed]
# i.e. if approxCornersNumber== 4:

# printing the position of the calculated corners
print("Coordinates of approximated corners:\n", approxCorners)

额外好处:如果您想检查哪个顶点是哪个顶点,您只需根据this answer添加最后一部分,其中(x, y)是上面为每个顶点提供的坐标

originalImg = cv2.circle(originalImg, (x,y), radius=10, color=(255, 255, 255), thickness=-1)
cv2.imshow("Vertex position",originalImg)
cv2.waitKey()
cv2.destroyWindow("Vertex position")

祝你度过愉快的一天,
安东尼诺

相关问题 更多 >