--- /dev/null
+#!/usr/bin/python
+#
+# The full "Square Detector" program.
+# It loads several images subsequentally and tries to find squares in
+# each image
+#
+
+from opencv.cv import *
+from opencv.highgui import *
+from math import sqrt
+
+thresh = 50;
+img = None;
+img0 = None;
+storage = None;
+wndname = "Square Detection Demo";
+
+def angle( pt1, pt2, pt0 ):
+ dx1 = pt1.x - pt0.x;
+ dy1 = pt1.y - pt0.y;
+ dx2 = pt2.x - pt0.x;
+ dy2 = pt2.y - pt0.y;
+ return (dx1*dx2 + dy1*dy2)/sqrt((dx1*dx1 + dy1*dy1)*(dx2*dx2 + dy2*dy2) + 1e-10);
+
+def findSquares4( img, storage ):
+ N = 11;
+ sz = cvSize( img.width & -2, img.height & -2 );
+ timg = cvCloneImage( img ); # make a copy of input image
+ gray = cvCreateImage( sz, 8, 1 );
+ pyr = cvCreateImage( cvSize(sz.width/2, sz.height/2), 8, 3 );
+ # create empty sequence that will contain points -
+ # 4 points per square (the square's vertices)
+ squares = cvCreateSeq( 0, sizeof_CvSeq, sizeof_CvPoint, storage );
+ squares = CvSeq_CvPoint.cast( squares )
+
+ # select the maximum ROI in the image
+ # with the width and height divisible by 2
+ subimage = cvGetSubRect( timg, cvRect( 0, 0, sz.width, sz.height ))
+
+ # down-scale and upscale the image to filter out the noise
+ cvPyrDown( subimage, pyr, 7 );
+ cvPyrUp( pyr, subimage, 7 );
+ tgray = cvCreateImage( sz, 8, 1 );
+ # find squares in every color plane of the image
+ for c in range(3):
+ # extract the c-th color plane
+ channels = [None, None, None]
+ channels[c] = tgray
+ cvSplit( subimage, channels[0], channels[1], channels[2], None )
+ for l in range(N):
+ # hack: use Canny instead of zero threshold level.
+ # Canny helps to catch squares with gradient shading
+ if( l == 0 ):
+ # apply Canny. Take the upper threshold from slider
+ # and set the lower to 0 (which forces edges merging)
+ cvCanny( tgray, gray, 0, thresh, 5 );
+ # dilate canny output to remove potential
+ # holes between edge segments
+ cvDilate( gray, gray, None, 1 );
+ else:
+ # apply threshold if l!=0:
+ # tgray(x,y) = gray(x,y) < (l+1)*255/N ? 255 : 0
+ cvThreshold( tgray, gray, (l+1)*255/N, 255, CV_THRESH_BINARY );
+
+ # find contours and store them all as a list
+ count, contours = cvFindContours( gray, storage, sizeof_CvContour,
+ CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE, cvPoint(0,0) );
+
+ if not contours:
+ continue
+
+ # test each contour
+ for contour in contours.hrange():
+ # approximate contour with accuracy proportional
+ # to the contour perimeter
+ result = cvApproxPoly( contour, sizeof_CvContour, storage,
+ CV_POLY_APPROX_DP, cvContourPerimeter(contours)*0.02, 0 );
+ # square contours should have 4 vertices after approximation
+ # relatively large area (to filter out noisy contours)
+ # and be convex.
+ # Note: absolute value of an area is used because
+ # area may be positive or negative - in accordance with the
+ # contour orientation
+ if( result.total == 4 and
+ abs(cvContourArea(result)) > 1000 and
+ cvCheckContourConvexity(result) ):
+ s = 0;
+ for i in range(5):
+ # find minimum angle between joint
+ # edges (maximum of cosine)
+ if( i >= 2 ):
+ t = abs(angle( result[i], result[i-2], result[i-1]))
+ if s<t:
+ s=t
+ # if cosines of all angles are small
+ # (all angles are ~90 degree) then write quandrange
+ # vertices to resultant sequence
+ if( s < 0.3 ):
+ for i in range(4):
+ squares.append( result[i] )
+
+ return squares;
+
+# the function draws all the squares in the image
+def drawSquares( img, squares ):
+ cpy = cvCloneImage( img );
+ # read 4 sequence elements at a time (all vertices of a square)
+ i=0
+ while i<squares.total:
+ pt = []
+ # read 4 vertices
+ pt.append( squares[i] )
+ pt.append( squares[i+1] )
+ pt.append( squares[i+2] )
+ pt.append( squares[i+3] )
+
+ # draw the square as a closed polyline
+ cvPolyLine( cpy, [pt], 1, CV_RGB(0,255,0), 3, CV_AA, 0 );
+ i+=4
+
+ # show the resultant image
+ cvShowImage( wndname, cpy );
+
+def on_trackbar( a ):
+ if( img ):
+ drawSquares( img, findSquares4( img, storage ) );
+
+names = ["../c/pic1.png", "../c/pic2.png", "../c/pic3.png",
+ "../c/pic4.png", "../c/pic5.png", "../c/pic6.png" ];
+
+if __name__ == "__main__":
+ # create memory storage that will contain all the dynamic data
+ storage = cvCreateMemStorage(0);
+ for name in names:
+ img0 = cvLoadImage( name, 1 );
+ if not img0:
+ print "Couldn't load %s" % name
+ continue;
+ img = cvCloneImage( img0 );
+ # create window and a trackbar (slider) with parent "image" and set callback
+ # (the slider regulates upper threshold, passed to Canny edge detector)
+ cvNamedWindow( wndname, 1 );
+ cvCreateTrackbar( "canny thresh", wndname, thresh, 1000, on_trackbar );
+ # force the image processing
+ on_trackbar(0);
+ # wait for key.
+ # Also the function cvWaitKey takes care of event processing
+ c = cvWaitKey(0);
+ # clear memory storage - reset free space position
+ cvClearMemStorage( storage );
+ if( c == '\x1b' ):
+ break;
+ cvDestroyWindow( wndname );