Exemplo n.º 1
0
    def Init(self):
        glutKeyboardFunc(self.keyboard)
        glutSpecialFunc(self.special)

        if 'darwin' not in os.uname()[0].lower():
            capture = self.getCapture()
            b = 0.213794155745
            c = 0.131609063828
            #cv.SetCaptureProperty(capture,CVtypes.CV_CAP_PROP_BRIGHTNESS,0.514000152471)
            #cv.SetCaptureProperty(capture,CVtypes.CV_CAP_PROP_CONTRAST,0.161806668155)
            cv.SetCaptureProperty(capture, CVtypes.CV_CAP_PROP_BRIGHTNESS, b)
            cv.SetCaptureProperty(capture, CVtypes.CV_CAP_PROP_CONTRAST, c)
Exemplo n.º 2
0
    def special(self, key, x, y):
        capture = self.getCapture()
        b = cv.GetCaptureProperty(capture, CVtypes.CV_CAP_PROP_BRIGHTNESS)
        c = cv.GetCaptureProperty(capture, CVtypes.CV_CAP_PROP_CONTRAST)
        if key == GLUT_KEY_DOWN:
            b -= .1
        elif key == GLUT_KEY_UP:
            b += .1
        elif key == GLUT_KEY_LEFT:
            c -= .01
        elif key == GLUT_KEY_RIGHT:
            c += .01

        cv.SetCaptureProperty(capture, CVtypes.CV_CAP_PROP_BRIGHTNESS, b)
        cv.SetCaptureProperty(capture, CVtypes.CV_CAP_PROP_CONTRAST, c)
        b = cv.GetCaptureProperty(capture, CVtypes.CV_CAP_PROP_BRIGHTNESS)
        c = cv.GetCaptureProperty(capture, CVtypes.CV_CAP_PROP_CONTRAST)
        print 'Brightness:', b
        print 'Contrast:', c
        print
Exemplo n.º 3
0
class WebcamDetector:
    def __init__(self, resolution=(640, 480), show_cam=False):
        self.setup(resolution, show_cam)

    def __del__(self):
        # release the capture device
        cv.ReleaseCapture(self.capture)
        if self.show_cam:
            # destroy the camera window
            cv.DestroyWindow('Camera')
        pass

    def setup(self, (width, height), show_cam):
        """Setup the webcam device and different windows."""
        device = 0  # assume we want the first device
        capture = cv.CreateCameraCapture(0)

        # set the width/height of the captured image
        # this won't work on Windows (*sigh*)
        cv.SetCaptureProperty(capture, cv.CAP_PROP_FRAME_WIDTH, width)
        cv.SetCaptureProperty(capture, cv.CAP_PROP_FRAME_HEIGHT, height)

        # get the width/height of the captured image
        # this won't work on Windows (*sigh*)
        self.width = cv.GetCaptureProperty(capture, cv.CAP_PROP_FRAME_WIDTH)
        self.height = cv.GetCaptureProperty(capture, cv.CAP_PROP_FRAME_HEIGHT)

        if self.width == 0 and self.height == 0:  # Windows
            self.width, self.height = 320, 240  # set to default

        # check if capture device is OK
        if not capture:
            print "Error opening capture device"

        self.capture = capture

        self.show_cam = show_cam
        if self.show_cam:
            cv.NamedWindow('Camera', cv.WINDOW_AUTOSIZE)
Exemplo n.º 4
0
def detect(image):
    image_size = cv.GetSize(image)

    # create grayscale version
    grayscale = cv.CreateImage(image_size, 8, 1)
    cv.CvtColor(image, grayscale, cv.BGR2GRAY)

    # create storage
    storage = cv.CreateMemStorage(0)
    cv.ClearMemStorage(storage)

    # equalize histogram
    cv.EqualizeHist(grayscale, grayscale)

    # detect objects
    cascade = cv.LoadHaarClassifierCascade('haarcascade_frontalface_alt.xml',
                                           cv.Size(1, 1))
    faces = cv.HaarDetectObjects(grayscale, cascade, storage, 1.2, 2,
                                 cv.HAAR_DO_CANNY_PRUNING, cv.Size(50, 50))

    if faces:
        print 'face detected!'
        for i in faces:
            cv.Rectangle(image, cv.Point(int(i.x), int(i.y)),
                         cv.Point(int(i.x + i.width), int(i.y + i.height)),
                         cv.RGB(0, 255, 0), 3, 8, 0)

    # create windows
    cv.NamedWindow('Camera', cv.WINDOW_AUTOSIZE)

    # create capture device
    device = 0  # assume we want first device
    capture = cv.CreateCameraCapture(0)
    cv.SetCaptureProperty(capture, cv.CAP_PROP_FRAME_WIDTH, 640)
    cv.SetCaptureProperty(capture, cv.CAP_PROP_FRAME_HEIGHT, 480)

    # check if capture device is OK
    if not capture:
        print "Error opening capture device"
        sys.exit(1)

    while 1:
        # do forever

        # capture the current frame
        frame = cv.QueryFrame(capture)
        if frame is None:
            break

        # mirror
        cv.Flip(frame, None, 1)

        # face detection
        detect(frame)
        # display webcam image
        cv.ShowImage('Camera', frame)

        # handle events
        k = cv.WaitKey(10)

        if k == 0x1b:  # ESC
            print 'ESC pressed. Exiting ...'
            break
Exemplo n.º 5
0
    try:
        # try to get the device number from the command line
        device = int(sys.argv[1])

        # got it ! so remove it from the arguments
        del sys.argv[1]
    except (IndexError, ValueError):
        # no device number on the command line, assume we want the 1st device
        device = 0

    if len(sys.argv) == 1:
        # no argument on the command line, try to use the camera
        capture = cv.CreateCameraCapture(device)

        # set the wanted image size from the camera
        cv.SetCaptureProperty(capture, cv.CAP_PROP_FRAME_WIDTH, 320)
        cv.SetCaptureProperty(capture, cv.CAP_PROP_FRAME_HEIGHT, 240)
    else:
        # we have an argument on the command line,
        # we can assume this is a file name, so open it
        capture = cv.CreateFileCapture(sys.argv[1])

    # check that capture device is OK
    if not capture:
        print "Error opening capture device"
        sys.exit(1)

    # create an image to put in the histogram
    histimg = cv.CreateImage(cv.Size(320, 240), 8, 3)

    # init the image of the histogram to black