camera.py

import cv2
from Facecv import FaceCV

class VideoCamera(object):
    def __init__(self):
        # Using OpenCV to capture from device 0. If you have trouble capturing
        # from a webcam, comment the line below out and use a video file
        # instead.
        
        self.video = cv2.VideoCapture(0)
        self.face = FaceCV(depth=16, width=8)
        # If you decide to use video.mp4, you must have this file in the folder
        # as the main.py.
        # self.video = cv2.VideoCapture('video.mp4')
    
    def __del__(self): 
        self.video.release()
    
    def get_frame(self):
        success, image = self.video.read()
        # We are using Motion JPEG, but OpenCV defaults to capture raw images,
        # so we must encode it into JPEG in order to correctly display the
        # video stream.

        outimg,ages,genders, g_label = self.face.detect_and_predict(image)
        ret, jpeg = cv2.imencode('.jpg', outimg)
        return jpeg.tobytes(), ages, g_label