o i @sddlmZddlZddlZddlZddlmZddlZddl Z ddl Z ddl Z ddl ZddZdddZdd d Zd d Zd dZeededejddddejddddejddeddddZedkrvedSdS))print_functionNcCsg}g}t|D]A}tjtj|d}t|}t|}t|dkr.t d |t|dkr=t d |q| || |dq||fS)NrzIWARNING: More than one face found in {}. Only considering the first face.z-WARNING: No faces found in {}. Ignoring file.) image_files_in_folderospathsplitextbasenameface_recognitionload_image_fileface_encodingslenclickechoformatappend)known_people_folder known_namesknown_face_encodingsfilerimg encodingsrkC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\face_recognition/face_recognition_cli.pyscan_known_peoples      rFcCs.|r td|||dStd||dS)Nz{},{},{}z{},{})printr)filenamenamedistance show_distancerrr print_result#sr333333?c st}t|jdkrtj|}|dtjjt |}t |}|D](}t ||} t | |k} d| vrGfddt| || Dq&tddq&|sZtdddSdS)N@)r!r!Tcs$g|]\}}}|rt||qSr)r).0Zis_matchrrimage_to_checkrrr :s$ztest_image..Zunknown_personZno_persons_found)r r maxshapePILZImageZ fromarrayZ thumbnailZLANCZOSnparrayr Z face_distancelistzipr) r$rr tolerancerZ unknown_imageZpil_imgZunknown_encodingsZunknown_encodingZ distancesresultrr#r test_image*s      r/csfddtDS)Ncs,g|]}tjd|tjdrtj|qS)z.*\.(jpg|jpeg|png))flags)rematchIrrjoin)r"ffolderrrr%Ds,z)image_files_in_folder..)rlistdirr6rr6rrCsrc Csp|dkrd}n|}t}dtvrtd}|j|d}t|t|t|t|t|} |t| dS)NZ forkserver) processes) multiprocessingZget_all_start_methodsZ get_contextZPoolr, itertoolsrepeatstarmapr/) Zimages_to_checkrrZnumber_of_cpusr-rr:contextpoolZfunction_parametersrrrprocess_images_in_process_poolGs   rArr$z--cpusrzmnumber of CPU cores to use in parallel (can speed up processing lots of images). -1 means "use all in system")defaulthelpz --tolerancezkTolerance for face comparisons. Default is 0.6. Lower this if you get multiple matches for the same person.z--show-distancez.) rsys version_infor rrrisdirrrAr/)rr$Zcpusr-rrrGrmain_s    rK__main__)F)r F) __future__rr rr1Zface_recognition.apiapir r;r<rHZ PIL.Imager(Znumpyr)rrr/rrAcommandZargumentoptionboolrK__name__rrrrs2