o >h @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_encodingsfilerZimg encodingsrk/home/www/facesmatcher.com/frenv_dlib/lib/python3.10/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..)rlistdirr5rr5rrCsrc 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,rr9contextpoolZfunction_parametersrrrprocess_images_in_process_poolGs   r@rr#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 rrrisdirrr@r.)rr#Zcpusr,rrrFrmain_s    rJ__main__)F)rF) __future__rr rr0Zface_recognition.apiapir r:r;rGZ PIL.Imager'numpyr(rrr.rr@commandargumentoptionboolrJ__name__rrrrs2