o *j @sddlZddlmZddlmZmZmZddlm Z ddl m Z m Z ddl mZddlmZddlmZddlmZmZmZd d lmZeZe je jfd ed ed eedeefddZefdedeefddZdeeeeffddZddZ ddZ!dS)N)ListOptionalUnion) HTTPError)LicensesModelVisibility)model_file_download)snapshot_download)Config)DEFAULT_MODEL_REVISION ConfigFields ModelFile) get_loggermodel_id chinese_name visibilitylicensecCsH||rtd|ddS|j||||dtd|ddS)Nzmodel z already exists, skip creation.F)rrrrz successfully created.T)Z repo_existsloggerinfoZ create_model)apirrrrrU/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/utils/hub.pycreate_model_if_not_exists rmodel_id_or_pathrevisioncCs\tj|st|tj|d}ntj|rtj|tj}n tj|r'|}ndSt |S)z Read config from hub or local path Args: model_id_or_path (str): Model repo name or local directory path. revision: revision of the model when getting from the hub Return: config (:obj:`Config`): config object )rN) ospathexistsrr CONFIGURATIONisdirjoinisfiler from_file)rrZ local_pathrrr read_config(s    r$modelcCs2t|trt|st|}|Sdd|D}|S)NcSs"g|] }t|s t|n|qSr)osprr ).0mrrr Dszauto_load..) isinstancestrr&rr )r%rrr auto_load?s  r,c CszRt|tj}t|d}t|r:t|}t|jdr$|jj j WSt|jdr5t|jds5|jj WS|jj WSt|rPt|}t|drM|j WSdWSWdSt ym}zt d|WYd}~dSd}~ww)aGet the model type from the configuration. This method will try to get the model type from 'model.backbone.type', 'model.type' or 'model.model_type' field in the configuration.json file. If this file does not exist, the method will try to get the 'model_type' field from the config.json. Args: model_dir: The local model dir to use. @return: The model type string, returns None if nothing is found. config.jsonbackbone model_typetypeNz%parse config file failed with error: )r&r!r rr"r r#hasattrr%r.r0r/ Exceptionrerror) model_dirZconfiguration_file config_filecfgerrrget_model_typeKs,           r8c Csddl}ddl}d}|j|tj}|j|rz'parse_label_mapping..label2idid2labelcSi|]\}}||qSrrr'idlabelrrrr<r=cSr@rrrArrrr<r=r-cSr@rrrArrrr<r=cSsi|] \}}|t|qSr)int)r'rCrBrrrr<s)jsonrrr!r Z LABEL_MAPPINGropenloaditemsrr r#r1r r%r>r?Z preprocessor) r4rErr>Z label_pathfZ label_mappingZ config_pathconfigr?rrrparse_label_mappingjsR            rK)"rZos.pathrr&typingrrrrequestsrZmodelscope.hub.constantsrrZmodelscope.hub.file_downloadrZ modelscope.hub.snapshot_downloadr Zmodelscope.utils.configr Zmodelscope.utils.constantr r r rrZPUBLICZ APACHE_V2r+rDrr$r,r8rKrrrrs: