Как определить восьмиугольные формы с помощью контуров - PullRequest
Новая
       35

Как определить восьмиугольные формы с помощью контуров

0 голосов
/ 30 мая 2018

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

Знак остановки:

image

Мне уже удается обнаружить треугольники 

    Mat ROI = new Mat();
    Mat bgrClone = bgr.clone();
    MatOfPoint approxContour = new MatOfPoint();
    MatOfPoint2f approxContour2f = new MatOfPoint2f();
    List<MatOfPoint> contourDraw = new ArrayList<MatOfPoint>();

    for(int i = 0; i < contourList.size(); i++) {
        MatOfPoint2f contour2f = new MatOfPoint2f(contourList.get(i).toArray());
         double approxDistance = Imgproc.arcLength(contour2f, true) * 0.02;//0.225
         Imgproc.approxPolyDP(contour2f, approxContour2f, approxDistance, true);

       approxContour2f.convertTo(approxContour, CvType.CV_32S);
    if (approxContour.size().height == 3 && (Imgproc.contourArea(contour2f) > 3000) ) { //&& (Imgproc.contourArea(contour2f) > 5000)

        contourDraw.add(approxContour);
        Imgproc.drawContours(bgr, contourDraw, -1, new Scalar(0,255,0), 1);

        Rect cord = Imgproc.boundingRect(approxContour);
        Core.rectangle(bgr, new Point(cord.x, cord.y), new Point(cord.x+cord.width, cord.y+cord.height),new Scalar(0,255,0), 1);

        ROI = bgrClone.submat(cord.y, cord.y+cord.height, cord.x, cord.x+cord.width);
        showResult(ROI);
    }
  }

Я использую Java, но любой на любом языке будет признателен.

1 Ответ

0 голосов
/ 07 июня 2018

Вы можете выполнить определение формы, используя контуры с OpenCV.Я реализовал следующее в Python.

Шаги:

  1. Поскольку вы имеете дело с дорожными знаками, большинство из них будет красного цвета.В цветовом пространстве RGB количество красного является наибольшим в красном канале.Но так как белый цвет также включает в себя красный, я не использовал его.Я скорее использовал синий канал, где красное содержание минимально.

Изображение синего канала:

enter image description here

Затем я применил порог Оцу к изображению выше и инвертировал его.

Порог Оцу и инверсия:

enter image description here

Наконец я нашел контуры и отметил эти контуры, имеющие около 8 дуг вокруг него.

Окончательный результат:

enter image description here

Код: 

import numpy as np
import cv2
import os

path = r'C:\Desktop\Stack\contour\shape_detection'
filename = 'stop.png'

img = cv2.imread(os.path.join(path, filename), 1)
cv2.imshow('img1',img[:,:,0])

ret,thresh1 = cv2.threshold(img[:,:,0], 0, 255,cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
cv2.imshow('thresh1', thresh1)

_, contours, hierarchy = cv2.findContours(thresh1, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)

for cnt in contours:
    approx = cv2.approxPolyDP(cnt,0.01*cv2.arcLength(cnt,True),True)
    print len(approx)
    if len(approx)==8:
        print "octagon"
        cv2.drawContours(img, [cnt], 0, (0, 255, 0), 6)

cv2.imshow('sign', img)       
cv2.waitKey(0)
cv2.destroyAllWindows()
...