File: laplace.py

package info (click to toggle)
opencv 2.1.0-3%2Bsqueeze1
  • links: PTS, VCS
  • area: main
  • in suites: squeeze
  • size: 68,800 kB
  • ctags: 52,010
  • sloc: cpp: 554,793; xml: 475,942; ansic: 153,396; python: 18,622; sh: 428; makefile: 111
file content (49 lines) | stat: -rwxr-xr-x 1,456 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
#!/usr/bin/python
from opencv.cv import *
from opencv.highgui import *
import sys

if __name__ == "__main__":
    laplace = None
    colorlaplace = None
    planes = [ None, None, None ]
    capture = None
    
    if len(sys.argv)==1:
        capture = cvCreateCameraCapture( 0 )
    elif len(sys.argv)==2 and sys.argv[1].isdigit():
        capture = cvCreateCameraCapture( int(sys.argv[1]) )
    elif len(sys.argv)==2:
        capture = cvCreateFileCapture( sys.argv[1] ) 

    if not capture:
        print "Could not initialize capturing..."
        sys.exit(-1)
        
    cvNamedWindow( "Laplacian", 1 )

    while True:
        frame = cvQueryFrame( capture )
        if not frame:
            cvWaitKey(0)
            break

        if not laplace:
            for i in range( len(planes) ):
                planes[i] = cvCreateImage( cvSize(frame.width,frame.height), 8, 1 )
            laplace = cvCreateImage( cvSize(frame.width,frame.height), IPL_DEPTH_16S, 1 )
            colorlaplace = cvCreateImage( cvSize(frame.width,frame.height), 8, 3 )

        cvSplit( frame, planes[0], planes[1], planes[2], None )
        for plane in planes:
            cvLaplace( plane, laplace, 3 )
            cvConvertScaleAbs( laplace, plane, 1, 0 )

        cvMerge( planes[0], planes[1], planes[2], None, colorlaplace )

        cvShowImage("Laplacian", colorlaplace )

        if cvWaitKey(10) != -1:
            break

    cvDestroyWindow("Laplacian")