U KÉhV'ã@s\ddlZddlZddlZddlZddlZddlZe ejj d¡Z e ejj d¡Z zej   ¡ZWn"ek r„edƒeƒYnXdZdZdZdZd Zdad d „Zd d „Zdd„Zdd„Zdd„Zdd„Zdd„ZedkrXedƒe dƒZ!e!dkree dƒƒqàe!dkreƒqàe!dkr,eƒqàe!dkr>eƒqàe!d krNqXqàed!ƒqàdS)"éNz#haarcascade_frontalface_default.xmlzhaarcascade_eye.xmlzYError: OpenCV contrib package is missing. Install with: pip install opencv-contrib-pythonz.http://community.scrubbed.net:8080/api/checkinÚregistered_faceszattendance_log.csvzface_recognizer.ymlzlast_user_id.txtcCsêtdƒg}tdƒD](}t |¡}| ¡r| |¡| ¡q|sNtdƒdStdƒ|D]}td|›d|›ƒqZzRtdt›d ƒ}|d kr’Wqæt |ƒ}||kr¼|atd t›d ƒWqæntd ƒWqtt k râtdƒYqtXqtdS)z:Lists available cameras and allows the user to select one.zDetecting available cameras...é zNo cameras found.NzAvailable cameras:z z : Camera zSelect a camera (current: z): ÚzCamera z selected.zInvalid camera index.zPlease enter a number.) ÚprintÚrangeÚcv2Ú VideoCaptureÚisOpenedÚappendÚreleaseÚinputÚselected_camera_indexÚintÚ ValueError)Zavailable_camerasÚiÚcapÚindexÚchoice©rú"/var/www/fran/franai/livestream.pyÚ select_cameras2     rcCsTt |d¡}t |¡}t |dd¡}tjddd}| |¡}t |dddtj¡}|S) zPEnhanced preprocessing for better recognition accuracy in real-world conditions.)éÈr)érrgø?)ér)Z clipLimitZ tileGridSizeNéÿ)rÚresizeÚ equalizeHistÚ GaussianBlurÚ createCLAHEÚapplyÚ normalizeÚ NORM_MINMAX)Úface_imgZclaherrrÚpreprocess_face>s   r#c Cs¢tj t|¡}tj|ddt t¡}d}td|›ƒtdƒtdƒ|dkrz|  ¡\}}|sfqzt  |tj ¡}t j |dd d d }|D]Æ\}} } } || | | …||| …f} | d krŠ| d krŠt ||| f|| | | fd d¡t |d|›d|| dftjdd d¡tj |d|d›d¡} t | | ¡|d7}t d¡|dkrŠqRqŠt d|¡t d¡d@tdƒkrHqzqH| ¡t ¡td|›d|›ƒ|S)zXCaptures face images for training, ensuring variations (with glasses, different angles).T)Úexist_okrz+Starting training data collection for User z?Please move your head slowly in different directions and anglesz"Press 'q' to stop collection earlyédgš™™™™™ñ?é)éPr'©Ú scaleFactorÚ minNeighborsÚminSizer'©rrréz Capturing: z/100rgffffffæ?Z training_Z03dú.jpgrzTraining Data CollectionrÚqz,Training data collection complete. Captured z images for User )ÚosÚpathÚjoinÚFACE_DIRÚmakedirsrrr rÚreadÚcvtColorÚCOLOR_BGR2GRAYÚ face_cascadeÚdetectMultiScaleÚ rectangleÚputTextÚFONT_HERSHEY_SIMPLEXÚimwriteÚwaitKeyÚimshowÚordr ÚdestroyAllWindows)Úuser_idÚ user_folderrÚcountÚretÚframeÚgrayÚfacesÚxÚyÚwÚhr"Úimg_pathrrrÚcollect_training_dataSsH   "ÿ   rNc sÎgg}}t t¡D]b}tj t|¡‰tj ˆ¡rz t|ƒ}Wn&tk rftd|›ƒYqYnXdddddg}t ‡fdd„|Dƒƒ}|r|D]z}tj ˆ|¡}tj  |¡r–t |¡D]P}|  d ¡r¾tj ||¡} t   | t j¡} | d kròq¾t| ƒ} | | ¡| |¡q¾q–qt ˆ¡D]X}|  d ¡rtj ˆ|¡} t   | t j¡} | d krXqt| ƒ} | | ¡| |¡qq|sŒtd ƒd St |t |¡¡t t¡td t|ƒ›d tt|ƒƒ›dƒd S)zNTrains LBPH face recognizer with additional preprocessing for better accuracy.z Skipping non-integer directory: ZfrontÚleftÚrightÚupZdownc3s$|]}tj tj ˆ|¡¡VqdS)N)r0r1Úisdirr2)Ú.0Úangle©rCrrÚ ”sz(train_face_recognizer..r.NzNo training data found.zTraining complete with z images from z users.)r0Úlistdirr3r1r2rRrrrÚanyÚexistsÚendswithrÚimreadÚIMREAD_GRAYSCALEr#r Ú recognizerÚtrainÚnpÚarrayÚsaveÚ MODEL_FILEÚlenÚset) rHÚlabelsZ user_id_strrBZ angle_dirsZhas_angle_structurerTZ angle_folderÚfilenamerMÚimgrrUrÚtrain_face_recognizer…sN           rhc Cs\tj t¡stdƒdSt t¡t t ¡}| ¡\}}|s@qHt  |tj ¡}t j |dddd}|D]º\}}}}||||…|||…f} t| ƒ} t | ¡\} } | dkrÐd| ›d t| d ƒ›d } t| ƒd } nd } d} t |||f||||f| d ¡t || ||dftjd| d tj¡qdt d|¡t d¡d@tdƒkr,qHq,| ¡t ¡dS)zKRecognizes a user even if wearing glasses or has slight appearance changes.z3Error: Model file not found. Train the model first.Ng333333ó?é)é2rjr(r'zUser: z (Acc: r-z%)r,ÚUnknown)rrrrrzFace Recognitionrr/)r0r1rYrbrr]r5rrr r6r7r8r9r#ÚpredictÚroundÚlog_attendancer:r;r<ÚLINE_AAr?r>r@r rA)rrErFrGrHrIrJrKrLr"rBÚ confidenceÚtextÚcolorrrrÚrecognize_user¹s6    "& rsc Csžtj ¡}| d¡}| d¡}tj t¡s^ttddd }t  |¡}|  dddg¡W5QRXttd dd(}t  |¡}|  ||d g¡t |ƒW5QRXd S) zGLogs attendance while ensuring a single check-in and check-out per day.z%Y-%m-%d %H:%M:%Sz%Y-%m-%drKr)ÚnewlineÚ TimestampzUser IDÚStatusÚazCheck-inN) ÚdatetimeÚnowÚstrftimer0r1rYÚLOG_FILEÚopenÚcsvÚwriterÚwriterowÚsend_post_request)rBryÚ timestampÚtodayÚfiler~rrrrnàs      rnc Cs˜d|i}ddi}zNtjt||d}|jdkrDtd|›d|j›ƒntd|j›d |j›ƒWn4tjjk r’}ztd |›ƒW5d }~XYnXd S) zSends attendance to API.rBz Content-Typezapplication/json)ÚjsonÚheadersru*✅ Successfully sent attendance for User z to API., Response: uâš ï¸ Failed. Status: z , Response: u⌠Error sending attendance: N)ÚrequestsÚpostÚ API_ENDPOINTÚ status_coderrqÚ exceptionsÚRequestException)rBÚpayloadr…ÚresponseÚerrrr€ñs r€Ú__main__zK 1: Register User 2: Train Model 3: Recognize Face 4: Select Camera 5: ExitzChoose an option: Ú1zEnter user ID: Ú2Ú3Ú4Ú5zInvalid choice. Try again.)"rÚnumpyr_r0rxr}r†ÚCascadeClassifierÚdataÚ haarcascadesr8Z eye_cascadeÚfaceÚLBPHFaceRecognizer_creater]ÚAttributeErrorrÚexitrˆr3r{rbZ USER_ID_FILEr rr#rNrhrsrnr€Ú__name__r rrrrrÚsN #24'