46 lines
1.4 KiB
Python
Executable File
46 lines
1.4 KiB
Python
Executable File
#!/usr/bin/python
|
|
import urllib2
|
|
import cv2.cv as cv
|
|
import sys
|
|
|
|
if __name__ == "__main__":
|
|
laplace = None
|
|
colorlaplace = None
|
|
planes = [ None, None, None ]
|
|
capture = None
|
|
|
|
if len(sys.argv) == 1:
|
|
capture = cv.CreateCameraCapture(0)
|
|
elif len(sys.argv) == 2 and sys.argv[1].isdigit():
|
|
capture = cv.CreateCameraCapture(int(sys.argv[1]))
|
|
elif len(sys.argv) == 2:
|
|
capture = cv.CreateFileCapture(sys.argv[1])
|
|
|
|
if not capture:
|
|
print "Could not initialize capturing..."
|
|
sys.exit(-1)
|
|
|
|
cv.NamedWindow("Laplacian", 1)
|
|
|
|
while True:
|
|
frame = cv.QueryFrame(capture)
|
|
if frame:
|
|
if not laplace:
|
|
planes = [cv.CreateImage((frame.width, frame.height), 8, 1) for i in range(3)]
|
|
laplace = cv.CreateImage((frame.width, frame.height), cv.IPL_DEPTH_16S, 1)
|
|
colorlaplace = cv.CreateImage((frame.width, frame.height), 8, 3)
|
|
|
|
cv.Split(frame, planes[0], planes[1], planes[2], None)
|
|
for plane in planes:
|
|
cv.Laplace(plane, laplace, 3)
|
|
cv.ConvertScaleAbs(laplace, plane, 1, 0)
|
|
|
|
cv.Merge(planes[0], planes[1], planes[2], None, colorlaplace)
|
|
|
|
cv.ShowImage("Laplacian", colorlaplace)
|
|
|
|
if cv.WaitKey(10) != -1:
|
|
break
|
|
|
|
cv.DestroyWindow("Laplacian")
|