请参阅This C++ code我试图在python中进行更改以完成相同的操作,但我仍然坚持使用for循环来查找aspect_ratio。我没有得到任何线索来获取python中RotatedRect的最小和最大宽度和高度,因为Python中C ++的RotatedRect的替代是cv2.boxPoints
import cv2
from matplotlib import pyplot as plt
img = cv2.imread('C:/Users/Zeeesh/Desktop/1.jpg', cv2.IMREAD_GRAYSCALE)
ret,thresh = cv2.threshold(img,200,255,0)
img1,contours, hierarchy = cv2.findContours(thresh,cv2.RETR_TREE,cv2.CHAIN_APPROX_NONE)
out = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
for contour in contours:
rect = cv2.minAreaRect(contour)
(x, y), (width, height), angle = rect
aspect_ratio = min(width, height) / max(width, height)
thresh1 = 0.2
if (aspect_ratio > thresh1):
print('Straight Line')
for pt in contour:
cv2.drawContours(out, [pt], 0,255,0)
else:
print('Curve')
for pt in contour:
cv2.drawContours(out, [pt], 0,0,255)
plt.imshow(out)
你可以找到旋转矩形的width
和height
:
rect = cv2.minAreaRect(contour)
(x, y), (width, height), angle = rect
然后纵横比是:
aspect_ratio = min(width, height) / max(width, height)