NiravJ's profile - activity

import base64
import time
import urllib2
import cv2
import numpy as np
import argparse
import os
from cStringIO import StringIO

DEFAULT_OP_PATH = 'FaceCaps\\'
DEFAULT_CASCADE_IP_PATH = 'haarcascade-frontalface-alt.xml'


# This is class which gets URL from IP camera and it reads url response put it into byte array of numpy


class VideoCap:
    def __init__(self):
        response = urllib2.urlopen("http://192.168.0.153:5000/video_feed")
        self.videosource =cv2.VideoCapture(np.asarray(bytearray(response.read()), dtype=np.uint8))
        self.count = 0
        self.argobj = Parse()
        self.facecascade = cv2.CascadeClassifier(self.argobj.input_path)

     def CaptureFrames(self):

        while True:

            #Create Frame Number for each frame we are assigning the 08d with number which is there in self.count
            framenumber = '%08d' % (self.count)

            #Capture frame by Frame
            ret, frame =videosource.read()

            #set screen color to gray so harr cascade can detect edges and faces
            screencolor = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)

            #customize how cascade detect your face
            faces = self.facecascade.detectMultiScale(screencolor,scaleFactor = 1.1,minNeighbors=5,minSize = (30,30),flags = cv2.CASCADE_SCALE_IMAGE)

            #Display resulting frame.
            cv2.imshow('Spying on you',screencolor)

            # If face length is 0 then it indicate that no faces detected
            if len(faces) == 0:
                pass
            #If faces is detected then it will return 1 or more than 1 depend upon faces
            elif len(faces) > 0:
                print('Face Detected')

                #Graph face and draw rectangle arround it.
                for (x,y,w,h) in faces:
                    cv2.rectangle(frame,(x,y),(x+w,y+h),(0,255,0),2)
                    sub_face= frame[x:x+w,y:y+h]
                    cv2.imwrite(DEFAULT_OP_PATH + framenumber + '.jpg',sub_face)

            #Increment count so we get uniquename for each frame
            self.count +=1

            if cv2.waitKey(1) == 27:
                break

        #When we are done capture we will close the release web connection
        self.videosource.release()
        cv2.waitKey(500)
        cv2.destroyAllWindows()
        cv2.waitKey(500)


def Parse():
    parser = argparse.ArgumentParser(description='Cascade path for face detection')
    parser.add_argument('-i','--input_path',type=str,default = DEFAULT_CASCADE_IP_PATH,help='Cascade input path')
    parser.add_argument('-o','--output_path',type=str,default = DEFAULT_OP_PATH , help = 'Output path for pics taken')
    args = parser.parse_args()
    return args

def ClearImageFolder():
    if not(os.path.exists(DEFAULT_OP_PATH)):
        os.makedirs(DEFAULT_OP_PATH)
    else:
        for files in os.listdir(DEFAULT_OP_PATH):
            filepath = os.path.join(DEFAULT_OP_PATH,files)
            if os.path.isfile(filepath):
                os.unlink(filepath)
            else:
                continue

def main():
    ClearImageFolder()

    #instaniate class object
    faceDetect = VideoCap()

    #Call Capture frames from class to begin face detection
    faceDetect.CaptureFrames()

if __name__ == '__main__':
    main()

Your help and time are most appreciated.