Jak wykrywać proste kształty geometryczne za pomocą OpenCV

Jeśli masz tylko te regularne kształty, istnieje prosta procedura w następujący sposób:

1. Znajdź kontury w obrazie (obraz powinien być binarny, jak podano w twoim pytaniu)
2. przybliżenie każdego konturu za pomocą funkcji approxPolyDP.
3. najpierw sprawdź liczbę elementów w przybliżonych konturach wszystkich kształtów. To rozpoznawanie kształtu. Na przykład kwadrat będzie miał 4, pentagon będzie miał 5. Kręgi będą miały więcej, nie wiem, więc je znajdziemy. (Mam 16 za kółko i 9 za półkole.)
4. Teraz Przypisz kolor, Uruchom kod do testowego obrazu, Sprawdź jego numer, wypełnij go odpowiednimi kolorami.

Poniżej mój przykład w Pythonie:

import numpy as np
import cv2

img = cv2.imread('shapes.png')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

ret,thresh = cv2.threshold(gray,127,255,1)

contours,h = cv2.findContours(thresh,1,2)

for cnt in contours:
    approx = cv2.approxPolyDP(cnt,0.01*cv2.arcLength(cnt,True),True)
    print len(approx)
    if len(approx)==5:
        print "pentagon"
        cv2.drawContours(img,[cnt],0,255,-1)
    elif len(approx)==3:
        print "triangle"
        cv2.drawContours(img,[cnt],0,(0,255,0),-1)
    elif len(approx)==4:
        print "square"
        cv2.drawContours(img,[cnt],0,(0,0,255),-1)
    elif len(approx) == 9:
        print "half-circle"
        cv2.drawContours(img,[cnt],0,(255,255,0),-1)
    elif len(approx) > 15:
        print "circle"
        cv2.drawContours(img,[cnt],0,(0,255,255),-1)

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

Poniżej wyjście:

Pamiętaj, że działa tylko dla regularnych kształtów.

Alternatywnie, aby znaleźć okręgi, możesz użyć houghcircles. Tutorial znajdziesz tutaj .

Jeśli chodzi o iOS, deweloperzy OpenCV opracowują próbki iOS tego lata, więc odwiedź ich stronę : www.code.opencv.org i skontaktować się z nimi.

Slajdy z ich samouczka znajdziesz tutaj: http://code.opencv.org/svn/gsoc2012/ios/trunk/doc/CVPR2012_OpenCV4IOS_Tutorial.pdf