Подсчитайте количество прямоугольников в изображении ⇐ Python

[Anonymous]

Фон

У меня есть два изображения, и я хочу посчитать количество прямоугольников в них обоих.
Я написал код, который находит контуры и использует его для поиска прямоугольников. Но он не работает должным образом, поэтому мне нужна помощь:

Проблема

Я не понимаю, почему код находит такое количество прямоугольников. Например, на первом изображении насчитывается 8, я ожидал бы 4.

На втором изображении насчитывается 16, что я считаю правильным (15 внутренних и 1 внешнее). ).

Код

Мой код выглядит следующим образом:

Код: Выделить всё

import cv2
import numpy as np

pic = 'boxes1'
image = cv2.imread(f'../Computer Vision/{pic}.jpg', 1)

blur = cv2.pyrMeanShiftFiltering(image, 11, 21)
gray = cv2.cvtColor(blur, cv2.COLOR_BGR2GRAY)
thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)[1]

contours = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
contours = contours[0] if len(contours) == 2 else contours[1]

rect_list = []

for cont in contours:
peri = cv2.arcLength(cont, True)
approx = cv2.approxPolyDP(cont, 0.015 * peri, True)
if len(approx) == 4:
x,y,w,h = cv2.boundingRect(approx)
rect = x,y,w,h
rect_list.append(rect)
cv2.rectangle(image,(x,y),(x+w,y+h),(36,255,12),2)

cv2.imshow('thresh', thresh)
cv2.imwrite(f'output_{pic}.png', thresh)
cv2.waitKey(0)
cv2.destroyAllWindows()
print(len(rect_list))

Текущие выходные данные

Код находит 8 прямоугольников на первом изображении и 16 во втором. Я думаю, что первое должно быть 4, а второе, вероятно, правильное (?) (15 внутренних и 1 внешнее).

Код сохраняет следующие выходные данные:

Подробнее здесь: https://stackoverflow.com/questions/605 ... n-an-image