Пример #1
0
class trainLimitFrame(object):
    def __init__(self,framelength=100,label=0,xmlfile='./train/mytrainSet.xml'):
        self._framelen=framelength
        self._xmlfile=xmlfile
        self._label=label
        self._core=core(faceRecognizeXmlfile=self._xmlfile)
        self._iosteam=IOSteam()
        self._window=PreviewWindowManager('TrainWindows')
        self._window_faceZone=PreviewWindowManager('TrainWindows.FaceZone')
    def run(self):
        self._window.createWindow()
        self._window_faceZone.createWindow()
        self._iosteam.run()
        for i in range(self._framelen):
            logging.info('current frame: {}'.format(i))
            frame=self._iosteam.frame
            
            
            
            self._core.train(frame,self._label)
#            self._core._face_detection(frame)
            
            whattoShow=self._core.displayFrame
            self._window.show(whattoShow)
            if self._core.faceSetofImageinSameFrameResized!=[]:
                self._window_faceZone.show(self._core.faceSetofImageinSameFrameResized[0])
            cv2.waitKey(1)
        self._iosteam.stop()
        self._core.saveTrainSet()
        self._window.destroyWindow()
        self._window_faceZone.destroyWindow()
Пример #2
0
 def __init__(self,framelength=100,label=0,xmlfile='./train/mytrainSet.xml'):
     self._framelen=framelength
     self._xmlfile=xmlfile
     self._label=label
     self._core=core(faceRecognizeXmlfile=self._xmlfile)
     self._iosteam=IOSteam()
     self._window=PreviewWindowManager('TrainWindows')
     self._window_faceZone=PreviewWindowManager('TrainWindows.FaceZone')
Пример #3
0
    def __init__(self,xmlfile='./train/mytrainSet.xml',imgDB='faceDB'):

        self._xmlfile=xmlfile

        self._trainedImgPostion=imgDB
        self._imgMat=[]
        self._label=[]

        self._core=core(faceRecognizeXmlfile=self._xmlfile)
        self._iosteam=IOSteam()
        self._window=PreviewWindowManager('TrainWindows',self.onKeypress)
        self._window_faceZone=PreviewWindowManager('TrainWindows.FaceZone')

        self._trained_file_used=False
        self._keep_run=True
        self._apply_recognize=False

        self.train_setted_img()
Пример #4
0
class application(object):
    def __init__(self,xmlfile='./train/mytrainSet.xml',imgDB='faceDB'):

        self._xmlfile=xmlfile

        self._trainedImgPostion=imgDB
        self._imgMat=[]
        self._label=[]

        self._core=core(faceRecognizeXmlfile=self._xmlfile)
        self._iosteam=IOSteam()
        self._window=PreviewWindowManager('TrainWindows',self.onKeypress)
        self._window_faceZone=PreviewWindowManager('TrainWindows.FaceZone')

        self._trained_file_used=False
        self._keep_run=True
        self._apply_recognize=False

        self.train_setted_img()
    def _link_img_lable(self,img,label):
        self._imgMat.append(img)
        self._label.append(label)

    
        
    def _load_image_trained(self):
        oswalk=list(os.walk(self._trainedImgPostion))
        for current_dir,sub_dir,sub_file in oswalk:
            if sub_file!=[]:
                for eachfile in sub_file:
                    fileposition=current_dir+'\\'+eachfile
                    img=cv2.imread(fileposition)
                    label=current_dir.split('\\')[-1]
                    self._link_img_lable(img,int(label))
                    
                    logging.debug('file walker:filePosition:{}  label:{}'.format(
                                                                fileposition, label  ))
                    
                    
    def onKeypress(self, keycode):
        """Handle a keypress.

        space  -> Take a screenshot.
        escape -> Quit.
        r      -> Reverse mode of faceRecognize,(default:off)
        m      -> Move png file to faceSetDataBase,Please input an integer[1,2,3,4,5...] to tell the programe how to settle current *.png.after operation,png files will be moved to (./faceDB/[label]/i.png)
        """
        if keycode == 32: # space
            self._iosteam.dumpFrame()

        elif keycode == 27: # escape
            #self._windowManager.destroyWindow()
            self._keep_run=False
        elif keycode == ord('r'):
            #activate face recognize
            self._apply_recognize=not self._apply_recognize
            logging.info('[+]app recognize mode change to {}'.format(
                                                        self._apply_recognize))
                                                        
                                                        
            if self._apply_recognize:
                #self.train_setted_img()
                pass
            
        elif keycode == ord('m'):
            #move file .png to label
            label=raw_input('[*]Please input the interger ID of current dirs\' photo:\n')
            try:
                os.mkdir('./faceDB/'+label)
            except WindowsError as e:
                logging.info('Directory error{},means that existed'.format(e))
            filename=[i for i in os.listdir('.') if i.endswith('png')]
            logging.info('file:{}'.format(filename))
            if 0==os.system('mv *.png ./faceDB/'+label):
                logging.info('[+]*.png moved Successfully')
            else:
                logging.info('[-]*.png moved Fail')

            
    def train_setted_img(self):
        """
        run during the init of app,pre train gived image with label
        """
        self._load_image_trained()
        for i in range(len(self._label)):
            self._core.train( self._imgMat[i],self._label[i] )
            logging.debug('pre train step used img{} ,label{} '.format(
                                            self._imgMat[i].shape,self._label[i]))
        self._trained_file_used=True
        
    def _runwithIOSteam(self):
        self._window.createWindow()
        self._window_faceZone.createWindow()
        self._iosteam.run()
        while self._keep_run:
            
            frame=self._iosteam.frame
            if(frame==None):
                continue
            #face recognize
            if self._apply_recognize:
                self._core.recognize(frame)
                result=self._core.result
            else:
                #just detect faceZone
                self._core._face_detection(frame)
            #preview result here
            whattoShow=self._core.displayFrame
            if self._apply_recognize:
                self._draw_name_in_frame(whattoShow)
            
            self._window.show(whattoShow)
            if self._core.faceSetofImageinSameFrameResized!=[]:
                self._window_faceZone.show(self._core.faceSetofImageinSameFrameResized[0])
            self._window.processEvents()

        self._iosteam.stop()
        self._window.destroyWindow()
        self._window_faceZone.destroyWindow()
    def _draw_name_in_frame(self,frame):
        for eachPeople in self._core.result:
            true_name=faceIDmatchName[eachPeople['who'][0]]
            dist=str(eachPeople['who'][1])
            pos=tuple(eachPeople['zone'][:2])
            draw_name(frame,true_name+dist,pos)
            
    def run(self):
        if(self._trained_file_used):
            self._runwithIOSteam()
        else:
            logging.debug('faceDB do not used in Train')
Пример #5
0
# -*- coding: utf-8 -*-
"""
Created on Sun Jun 19 17:33:35 2016

test FaceDetectation

@author: pip
"""

import cv2
import logging
from cameraIO import IOSteam
import sys


from trackers import FaceTracker

front=IOSteam()
tracker=FaceTracker()

front.run()
tracker.update( front.frame)
facelist=tracker.faces
front.stop()
print(facelist)