U A{i; @sddlmZddlZddlZddlZddlmZddlZddl Z ddl Z ddZ ddZ ddZ d d Zeed ejd d ddejddddddZedkredS))print_functionNcCs&|\}}}}td|||||dS)Nz{},{},{},{},{})printformat)filenamelocationtoprightZbottomleftr G/tmp/pip-unpacked-wheel-pbnq7y3j/face_recognition/face_detection_cli.py print_result s r cCs2t|}tj|d|d}|D]}t||qdS)Nr)Znumber_of_times_to_upsamplemodel)face_recognitionZload_image_fileface_locationsr )image_to_checkr Z unknown_imagerZ face_locationr r r test_images rcsfddtDS)Ncs,g|]$}tjd|tjdrtj|qS)z.*\.(jpg|jpeg|png))flags)rematchIospathjoin).0ffolderr r sz)image_files_in_folder..)rlistdirrr rr image_files_in_foldersrcCsX|dkrd}n|}t}dtkr,td}|j|d}t|t|}|t|dS)NZ forkserver) processes) multiprocessingZget_all_start_methods get_contextZPoolzip itertoolsrepeatstarmapr)Zimages_to_checkZnumber_of_cpusr r!contextpoolZfunction_parametersr r r process_images_in_process_pools   r*rz--cpuszDnumber of CPU cores to use in parallel. -1 means "use all in system")defaulthelpz--modelZhogz>Which face detection model to use. Options are "hog" or "cnn".csltjdkr |dkr tdd}tj|r^|dkrLfddt|Dqhtt||n t |dS)N)r+zmWARNING: Multi-processing support requires Python 3.4 or greater. Falling back to single-threaded processing!csg|]}t|qSr )r)rZ image_filer r r r>szmain..) sys version_infoclickZechorrisdirrr*r)rZcpusr r r0r main2s  r5__main__) __future__rr3rrZface_recognition.apiapirr"r1r%r rrr*commandargumentoptionr5__name__r r r r s$