Как рассчитать количество ребер в многоугольнике, используя PIL / numpy - PullRequest
Новая
       17

Как рассчитать количество ребер в многоугольнике, используя PIL / numpy

0 голосов
/ 07 февраля 2020

enter image description here

Вопрос

С учетом черно-белого многоугольника, какой метод расчета числа ребер является наиболее эффективным ?

Например, приведенное выше изображение должно возвращать значение 7.

1 Ответ

1 голос
/ 07 февраля 2020

Чтобы рассчитать количество ребер, вы можете использовать контурное приближение. Идея состоит в том, что кривая может быть аппроксимирована серией коротких прямых . Это приводит к полученной аппроксимированной кривой, которая состоит из подмножества точек, определенных исходной кривой.

Вот решение с использованием OpenCV. Контурная аппроксимация уже реализована как cv2.approxPolyDP, которая аппроксимирует многоугольную кривую. Длина возвращаемого значения будет количеством ребер многоугольника.

Код 

import cv2

# Load image, grayscale, Otsu's threshold
image = cv2.imread('1.jpg')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)[1]

# Find contours and perform contour approximation
cnts = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cnts = cnts[0] if len(cnts) == 2 else cnts[1]
for c in cnts:
    peri = cv2.arcLength(c, True)
    approx = cv2.approxPolyDP(c, 0.03 * peri, True)
    print(len(approx))

Результат 

7
...