U h$ @sdZddlmZdgZddlZddlZddlZddlmZm Z ddd DZ d Z d Z d d Z ddZejdddfddZdS)zp This code file mainly comes from https://github.com/dmlc/gluon-cv/blob/master/gluoncv/model_zoo/model_store.py )print_functionget_model_fileN)download check_sha1cCsi|]\}}||qSr).0ZchecksumnamerrE/tmp/pip-unpacked-wheel-5oclok7i/insightface/model_zoo/model_store.py sr ))Z(95be21b58e29e9c1237f229dae534bd854009ce0Zarcface_r100_v1)Zarcface_mfn_v1)Z(39fd1e087a2a2ed70a154ac01fecaa86c315d01bZretinaface_r50_v1)Z(2c9de8116d1f448fd1d4661f90308faae34c990aZretinaface_mnet025_v1)Z(0db1d07921d005e6c9a5b38e059452fc5645e5a4Zretinaface_mnet025_v2)Z(7dd8111652b7aac2490c5dcddeb268e53ac643e6Z genderage_v1zhttps://insightface.ai/files/z {repo_url}models/{file_name}.zipcCs(|tkrtdj|dt|ddS)Nz-Pretrained model for {name} is not available.r ) _model_sha1 ValueErrorformatr rrr short_hashs  rcCs>tj|sdStd|}t|dkr.dSt|}|dS)Nz %s/*.paramsr)ospathexistsgloblensorted)dir_pathpathsrrr find_params_file$s  r~z .insightfacemodelsc Cs|}tj|}tj||}t|}t|}|dk rNt||rD|Stdntdtj|slt |tj|st |tj||d}t }|ddkr|d}t t j ||d|dd t|}||W5QRXt|t|}t||r|Std dS) aReturn location for the pretrained on local file system. This function will download from online model zoo when model cannot be found or has mismatch. The root directory will be created if it doesn't exist. Parameters ---------- name : str Name of the model. root : str, default '~/.mxnet/models' Location for keeping the model parameters. Returns ------- file_path Path to the requested pretrained model file. NzBMismatch in the content of model file detected. Downloading again.z%Model file is not found. Downloading.z.zipr/)repo_url file_nameT)r overwritez5Downloaded file has different hash. Please try again.)rr expanduserjoinrrrprintrmakedirs base_repo_urlr _url_formatrzipfileZipFile extractallremover) r rootr!r file_pathZ sha1_hashZ zip_file_pathr zfrrr r.sB          )__doc__ __future__r__all__rr)rutilsrrrr'r(rrrr$rrrrr s