使用Python检测图像中的矩形

import cv2
import numpy as np

image = cv2.imread('image.jpg')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

edges = cv2.Canny(gray, 50, 150)

contours, _ = cv2.findContours(edges, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

rectangles = []
for contour in contours:
    perimeter = cv2.arcLength(contour, True)
    approx = cv2.approxPolyDP(contour, 0.02 * perimeter, True)
    if len(approx) == 4:
        rectangles.append(approx)

for rectangle in rectangles:
    cv2.drawContours(image, [rectangle], -1, (0, 255, 0), 2)

cv2.imshow('Rectangles', image)
cv2.waitKey(0)
cv2.destroyAllWindows()