o 1j @sUddlZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZddlZddlZddlmZejddlZdejd<ddlZddlmZddlmZd d lmZdZz ddl m!m"Z#e#ZWne$yZ%z e&d e%WYdZ%[%ndZ%[%wwd d l'm(Z(m)Z)d d l*m+Z+d dl,m-Z-m.Z.m/Z/ddl0m1Z1gdZ2gdZ3hdZ4e e5e6d<ddhZ7e e5e6d<e4e7BZ8e e5e6d<hdZ9e e5e6d<hdZ:e e5e6d<de5de5fdd Z;de5d!e1de5fd"d#Z dAde5d&e e e1dee5d'ffd(d)Z?d*ee5d'fde5fd+d,Z@d-e$fd.d/ZAd0eBde eCfd1d2ZDd-e$de=fd3d4ZEGd5d6d6eZFGd7d8d8eFZGGd9d:d:eFZHGd;d<dd>eFZJGd?d@d@ZKeKZLdS)BN)ABCabstractmethod)Path)OptionalSequenceSetTuplecriticalZ AISTUDIO_LOG) NotExistError)snapshot_download)loggingzFailed to import `modelscope.hub.errors` (%r). ModelScope downloads can still be used; not-found detection will use generic fallbacks.) CACHE_DIR FILE_LOCK_DIR)download_and_extract)DISABLE_MODEL_SOURCE_CHECKHUGGING_FACE_ENDPOINT MODEL_SOURCE)LocalModelFormat(%ZResNet18Z ResNet18_vdZResNet34Z ResNet34_vdZResNet50Z ResNet50_vdZ ResNet101Z ResNet101_vdZ ResNet152Z ResNet152_vdZ ResNet200_vd PaddleOCR-VLPaddleOCR-VL-1.5zPP-LCNet_x0_25PP-LCNet_x0_25_textline_orizPP-LCNet_x0_35z PP-LCNet_x0_5zPP-LCNet_x0_75z PP-LCNet_x1_0PP-LCNet_x1_0_doc_oriPP-LCNet_x1_0_textline_oriz PP-LCNet_x1_5z PP-LCNet_x2_5z PP-LCNet_x2_0zPP-LCNetV2_smallzPP-LCNetV2_basezPP-LCNetV2_largeZMobileNetV3_large_x0_35ZMobileNetV3_large_x0_5ZMobileNetV3_large_x0_75ZMobileNetV3_large_x1_0ZMobileNetV3_large_x1_25ZMobileNetV3_small_x0_35ZMobileNetV3_small_x0_5ZMobileNetV3_small_x0_75ZMobileNetV3_small_x1_0ZMobileNetV3_small_x1_25Z ConvNeXt_tinyZConvNeXt_smallZConvNeXt_base_224ZConvNeXt_base_384ZConvNeXt_large_224ZConvNeXt_large_384ZMobileNetV2_x0_25ZMobileNetV2_x0_5ZMobileNetV2_x1_0ZMobileNetV2_x1_5ZMobileNetV2_x2_0ZMobileNetV1_x0_25ZMobileNetV1_x0_5ZMobileNetV1_x0_75ZMobileNetV1_x1_0Z'SwinTransformer_tiny_patch4_window7_224Z(SwinTransformer_small_patch4_window7_224Z'SwinTransformer_base_patch4_window7_224Z(SwinTransformer_base_patch4_window12_384Z(SwinTransformer_large_patch4_window7_224Z)SwinTransformer_large_patch4_window12_384z PP-HGNet_tinyzPP-HGNet_smallz PP-HGNet_basez PP-HGNetV2-B0z PP-HGNetV2-B1z PP-HGNetV2-B2z PP-HGNetV2-B3z PP-HGNetV2-B4z PP-HGNetV2-B5z PP-HGNetV2-B6z FasterNet-Lz FasterNet-Mz FasterNet-Sz FasterNet-T0z FasterNet-T1z FasterNet-T2z StarNet-S1z StarNet-S2z StarNet-S3z StarNet-S4ZMobileNetV4_conv_smallZMobileNetV4_conv_mediumZMobileNetV4_conv_largeZMobileNetV4_hybrid_mediumZMobileNetV4_hybrid_largeZCLIP_vit_base_patch16_224ZCLIP_vit_large_patch14_224zPP-LCNet_x1_0_MLzPP-HGNetV2-B0_MLzPP-HGNetV2-B4_MLzPP-HGNetV2-B6_MLZ ResNet50_MLZCLIP_vit_base_patch16_448_MLzPP-YOLOE_plus-XzPP-YOLOE_plus-LzPP-YOLOE_plus-MzPP-YOLOE_plus-Sz RT-DETR-Lz RT-DETR-Hz RT-DETR-XzYOLOv3-DarkNet53zYOLOv3-MobileNetV3zYOLOv3-ResNet50_vd_DCNzYOLOX-LzYOLOX-MzYOLOX-NzYOLOX-SzYOLOX-TzYOLOX-Xz RT-DETR-R18z RT-DETR-R50z PicoDet-Sz PicoDet-Lz Deeplabv3-R50zDeeplabv3-R101zDeeplabv3_Plus-R50zDeeplabv3_Plus-R101zPP-ShiTuV2_reczPP-ShiTuV2_rec_CLIP_vit_basezPP-ShiTuV2_rec_CLIP_vit_largez PP-LiteSeg-Tz PP-LiteSeg-BzOCRNet_HRNet-W48zOCRNet_HRNet-W18z SegFormer-B0z SegFormer-B1z SegFormer-B2z SegFormer-B3z SegFormer-B4z SegFormer-B5ZSeaFormer_tinyZSeaFormer_smallZSeaFormer_baseZSeaFormer_largezMask-RT-DETR-HzMask-RT-DETR-LPP-OCRv4_server_reczMask-RT-DETR-SzMask-RT-DETR-MzMask-RT-DETR-XZSOLOv2zMaskRCNN-ResNet50zMaskRCNN-ResNet50-FPNzMaskRCNN-ResNet50-vd-FPNzMaskRCNN-ResNet101-FPNzMaskRCNN-ResNet101-vd-FPNzMaskRCNN-ResNeXt101-vd-FPNzCascade-MaskRCNN-ResNet50-FPNz'Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPNzPP-YOLOE_seg-SPP-OCRv3_mobile_recen_PP-OCRv3_mobile_reckorean_PP-OCRv3_mobile_recjapan_PP-OCRv3_mobile_recchinese_cht_PP-OCRv3_mobile_recte_PP-OCRv3_mobile_recka_PP-OCRv3_mobile_recta_PP-OCRv3_mobile_reclatin_PP-OCRv3_mobile_recarabic_PP-OCRv3_mobile_reccyrillic_PP-OCRv3_mobile_recdevanagari_PP-OCRv3_mobile_recen_PP-OCRv4_mobile_recPP-OCRv4_server_rec_docPP-OCRv4_mobile_recPP-OCRv4_server_detPP-OCRv4_mobile_detPP-OCRv3_server_detPP-OCRv3_mobile_detPP-OCRv4_server_seal_detPP-OCRv4_mobile_seal_detch_RepSVTR_rec ch_SVTRv2_recz"PP-LCNet_x1_0_pedestrian_attributezPP-LCNet_x1_0_vehicle_attributePicoDet_layout_1xPicoDet_layout_1x_tableSLANet SLANet_plus LaTeX_OCR_rec UniMERNetPP-FormulaNet-SPP-FormulaNet-LPP-FormulaNet_plus-SPP-FormulaNet_plus-MPP-FormulaNet_plus-LzFasterRCNN-ResNet34-FPNzFasterRCNN-ResNet50zFasterRCNN-ResNet50-FPNzFasterRCNN-ResNet50-vd-FPNz!FasterRCNN-ResNet50-vd-SSLDv2-FPNzFasterRCNN-ResNet101zFasterRCNN-ResNet101-FPNzFasterRCNN-ResNeXt101-vd-FPNzFasterRCNN-Swin-Tiny-FPNzCascade-FasterRCNN-ResNet50-FPNz)Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPNUVDocZDLinearZNLinearZRLinearZ NonstationaryZTimesNetZTiDEZPatchTSTZ DLinear_adZAutoEncoder_adZNonstationary_adZ PatchTST_adZ TimesNet_adZ TimesNet_clsZSTFPMz FCOS-ResNet50zDETR-R50zPP-YOLOE-L_vehiclezPP-YOLOE-S_vehiclezPP-ShiTuV2_detzPP-YOLOE-S_humanzPP-YOLOE-L_humanz PicoDet-Mz PicoDet-XSzPP-YOLOE_plus_SOD-LzPP-YOLOE_plus_SOD-SzPP-YOLOE_plus_SOD-largesize-LzCenterNet-DLA-34zCenterNet-ResNet50PicoDet-S_layout_3clsPicoDet-S_layout_17clsPicoDet-L_layout_3clsPicoDet-L_layout_17clsRT-DETR-H_layout_3clsRT-DETR-H_layout_17clsZPicoDet_LCNet_x2_5_faceZ BlazeFacezBlazeFace-FPN-SSHzPP-YOLOE_plus-S_faceZ MobileFaceNetZ ResNet50_facez PP-YOLOE-R-LzCo-Deformable-DETR-R50zCo-Deformable-DETR-Swin-Tz Co-DINO-R50zCo-DINO-Swin-L whisper_large whisper_basewhisper_medium whisper_small whisper_tinyzPP-TSM-R50_8frames_uniformz PP-TSMv2-LCNetV2_8frames_uniformz!PP-TSMv2-LCNetV2_16frames_uniformZMaskFormer_tinyZMaskFormer_smallPP-LCNet_x1_0_table_cls SLANeXt_wiredSLANeXt_wirelessRT-DETR-L_wired_table_cell_det!RT-DETR-L_wireless_table_cell_detZYOWOzPP-TinyPose_128x96zPP-TinyPose_256x192zGroundingDINO-Tz SAM-H_boxz SAM-H_pointPP-DocLayoutV2PP-DocLayoutV3PP-DocLayout-LPP-DocLayout-MPP-DocLayout-SPP-DocLayout_plus-LPP-DocBlockLayoutZ BEVFusionzYOLO-Worldv2-L PP-DocBee-2B PP-DocBee-7BPP-Chart2TablePP-OCRv5_server_detPP-OCRv5_mobile_detPP-OCRv5_server_recPP-OCRv5_mobile_receslav_PP-OCRv5_mobile_rec PP-DocBee2-3Blatin_PP-OCRv5_mobile_reckorean_PP-OCRv5_mobile_recth_PP-OCRv5_mobile_recel_PP-OCRv5_mobile_recen_PP-OCRv5_mobile_recarabic_PP-OCRv5_mobile_recte_PP-OCRv5_mobile_recta_PP-OCRv5_mobile_recdevanagari_PP-OCRv5_mobile_reccyrillic_PP-OCRv5_mobile_recZ G2PWModelZfastspeech2_csmscZ pwgan_csmsc)Nr%r r1r2r&r'rr(r]rr"rr`r7r$r_rcrarbrrr3r4rBrAr@r?r^rXrVrWrUrOrPrQrRrTrSr:r=r<r;r9rrrJrr.rr-r,r*r0r+r)rr/rZr\rYr[rDrCrMrNr5r6rKrLr#r!r8r>rdrerfrgrh> rUr\r5rMrrgr>r6rLrdrrarcrYrTrOrXrfrJrhr[rerPr`rKrrNrZr]rbr_r=(SAFETENSORS_SUPPORTED_MODELS_WITH_SUFFIXPaddleOCR-VL-0.9BzPaddleOCR-VL-1.5-0.9B+SAFETENSORS_SUPPORTED_MODELS_WITHOUT_SUFFIXSAFETENSORS_SUPPORTED_MODELS>rHrGrFr^rIrErVrWPADDLE_DYN_SUPPORTED_MODELS>rUr\rrKrJrrMrNr[rrZrYrLrTONNX_SUPPORTED_MODELS model_namereturncCs|dvrdS|S)N>rjrrjrorqrqh/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/utils/official_models.py _canonical_download_support_namesrt model_formatcCsb|dvr|S|dkr |tvr|dS|tvr|Std||dkr)|dStd|d) N> paddle_dynpaddle safetensors _safetensorsz Unknown safetensors model name: onnx_onnxUnknown official model format: .)rirk ValueError)rorurqrqrs_format_download_model_names  rcCs`t|}|dkr dS|dkr|tvS|dkr|tvS|dkr"|tvS|dkr(dStd|d ) NrwTrvrxrzomFr|r})rtrmrlrnr~)rorucanonical_namerqrqrs#_is_supported_official_model_formatsr model_formats.cCs|dur|fSt|}g}g}|D]}t||s||qt||}||vr,||q|sOt|dkrBtd|dd|dtdt|d|dt|S)Nz)Official model source does not provide a rz package for model r}zEOfficial model source does not provide any of the requested packages z for model )tuplerappendrlenr~list)rorformats model_namesZunsupported_formatsruZdownload_model_namerqrqrs_resolve_download_model_names%s8      rrcCsJtjtd}tj|ddtd|d}tj||dS)zCCross-process lock path for a resolved official model download key.official_modelsT)exist_okzutf-8z.lock) ospathjoinrmakedirshashlibsha256encode hexdigest)rZlock_dirkeyrqrqrs"_official_model_download_lock_pathBsrexcccsp|}t}|dur2t||vr4|t||Vt|ddp%t|dd}|dur6t||vsdSdSdSdS)N __cause__ __context__)setidaddgetattr)rcurrentvisitedrqrqrs_iter_exception_chainJs$rexc_objcCs^t|dd}t|dd}t|tr|St|ddpdD]}t|dd}t|tr,|SqdS)Nresponse status_codeargsrq)r isinstanceint)rrcodeargscrqrqrs_exception_http_status_codeUs     rcCstdur%t|D]}t|tjrdSt|tjr"t|dkr"dSqdSt|D])}t|tjr:t|dkr:dS|jjdkrCdS|jjdkrRt|dkrRdSq)dS)zKDetect ModelScope 'model not found' errors with or without `ms_hub_errors`.NTFr HTTPError) ms_hub_errorsrrr rrrequests __class____name__)rrrqrqrs,_modelscope_is_model_package_not_found_errords(          rc@sneZdZdZgZdZdZddZeddZ dd Z d d Z e d d Z e dedefddZeddZdS)_BaseModelHosterNrcCs ||_dSN) _save_dir)selfsave_dirrqrqrs__init__ z_BaseModelHoster.__init__cCs.dD]}||r|dt| Sq|S)N)ryr{)endswithr)rosuffixrqrqrs_strip_repo_suffixs  z#_BaseModelHoster._strip_repo_suffixcCs||jvrdS|||jvS)NT) model_listrrrorqrqrssupports_models z_BaseModelHoster.supports_modelcCsr||sJd|d|jjd|j|}td|d|d|||td|d|jd |S) Nz The model z is not supported on hosting !zUsing official model (zB), the model files will be automatically downloaded and saved in ``.`z4` model files has been download from model source: `z`!) rrrrr info _downloaddebugalias)rro model_dirrqrqrs get_models  z_BaseModelHoster.get_modelcCtrNotImplementedErrorrrqrqrsrz_BaseModelHoster._downloadrrpcCrrrrrrqrqrs is_model_package_not_found_errorrz1_BaseModelHoster.is_model_package_not_found_errorcCsV|jdurdSztj|j|jd}|jdkWSty*td|jdYdSw)NT)timeoutzThe model hosting platform(z) is unreachable!F) healthcheck_urlrhead_healthcheck_timeoutok Exceptionr rr)clsrrqrqrs is_availables   z_BaseModelHoster.is_available)r __module__ __qualname__rrrrr staticmethodrrrrrrboolr classmethodrrqrqrqrsr}s   rc@sJeZdZeZdZdZdZdZddddd d Z d d Z d e de fddZ dS)_BosModelHosterZbosz*https://paddle-model-ecology.bj.bcebos.comz paddle3.0.0zKhttps://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_modelzwhisper_large.tarzwhisper_base.tarzwhisper_medium.tarzwhisper_small.tarzwhisper_tiny.tar)rErFrGrHrIcCsN||jvr |j|}n|d}|jd|jd|}t||j|dddS)Nz _infer.tar/F) overwrite)special_model_fnbase_urlversionrparent)rrorfnurlrqrqrsrs   z_BosModelHoster._downloadrrpcCs<t|D]}t|tjr|j}|dur|jdkrdSqdS)NrTF)rrrrrrrrrrrqrqrsrs  z0_BosModelHoster.is_model_package_not_found_errorN)rrr ALL_MODELSrrrrrrrrrrrqrqrqrsrsrc@s2eZdZeZdZeZddZde de fddZ dS) _HuggingFaceModelHosterZ huggingfacectfdd}tj|r||dSt}tj|d}||t||WddS1s3wYdS)Ncstjd|tddS)N PaddlePaddle/)repo_id local_dirZendpoint)hf_hubr rrrrrqrs_clones  z1_HuggingFaceModelHoster._download.._clonetemp_dirrrexiststempfileTemporaryDirectoryrshutilmoverrorrtdrrqrrrsrs    "z!_HuggingFaceModelHoster._downloadrrpcCst|D]C}t|tjtjtjfrdSt|tjrG|j}|dur)|jdkr)dS|durG|jdkrGdt |vsDdt |vsDdt |vrGdSqdS)NTrizRepository Not FoundzEntry Not FoundzRevision Not FoundF) rr hf_hub_utilsZRepositoryNotFoundErrorZEntryNotFoundErrorZRevisionNotFoundErrorZHfHubHTTPErrorrrstrrrqrqrsrs*      z8_HuggingFaceModelHoster.is_model_package_not_found_errorN) rrr OCR_MODELSrrrrrrrrrqrqrqrsrs rc@2eZdZeZdZdZddZdede fddZ d S) _ModelScopeModelHoster modelscopezhttps://modelscope.cncr)Ncstjd|ddS)Nrrr)rr rrrrqrsrs  z0_ModelScopeModelHoster._download.._clonerrrrqrrrsrs    "z _ModelScopeModelHoster._downloadrrpcCst|Sr)rrrqrqrsrsz7_ModelScopeModelHoster.is_model_package_not_found_errorN rrrrrrrrrrrrqrqrqrsr s rc@r) _AIStudioModelHosterZaistudiozhttps://aistudio.baidu.comcr)Ncs4dvrtd|ddStd|ddS)NrrrzPaddleX/)aistudio_downloadrrrrqrsr(s   z._AIStudioModelHoster._download.._clonerrrrqrrrsr's    "z_AIStudioModelHoster._downloadrrpcCsLt|D]}t|trdSt|tjr#|j}|dur#|jdkr#dSqdS)NTrF)rrr rrrrrrqrqrsr8s   z5_AIStudioModelHoster.is_model_package_not_found_errorNrrqrqrqrsr"s rc@seZdZeZeedZee e e gZ dddZ ddZdd Zd d Zdd d edeeefddZddZddZddZdS) _ModelManagerrrpNcCsd|_t|_dSr)_hosters threadingLock _hosters_lockrrqrqrsrMsz_ModelManager.__init__c Cstr(tdg}|jD]}|jtkr|d||jq |||jq |Stdg}|jD]!}|jtkrG| rF|d||jq2| rS|||jq2t |dkrptdt j dt j dtj dtj d |S) NzsConnectivity check to the model hoster has been skipped because `PADDLE_PDX_DISABLE_MODEL_SOURCE_CHECK` is enabled.rzChecking connectivity to the model hosters, this may take a while. To bypass this check, set `PADDLE_PDX_DISABLE_MODEL_SOURCE_CHECK` to `True`.zvNo model hoster is available! Please check your network connection to one of the following model hoster: HuggingFace (z), ModelScope (z ), AIStudio (z ), or BOS (z,). Otherwise, only local models can be used.)rr warninghoster_candidatesrrinsertrrrrrrrrr)rhostersZ hoster_clsrqrqrs_build_hostersQs6     $z_ModelManager._build_hosterscCs`|jdur-|j|jdur||_Wd|jSWd|jS1s(wY|jSr)rr rrrqrqrs _get_hostersps     z_ModelManager._get_hostersc Csht|tr|fnt|}g}|D]}d|vr|dd}||qd}|D]}|j|}tj|rAt d|d|}nq&|durt t|}t |E|D]}|j|}tj|rot d|d|}nqT|dur|}t|dkrd} t | t| |||}Wdn1swY|ddkr|d} | r| r| S|S) Nrz-0.9BrzeModel files already exist. Using cached files. To redownload, please delete the directory manually: `rrzTNo available model hosting platforms detected. Please check your network connection.rj)rrrreplacerrrrrr rrfilelockFileLockrrerrorr_download_from_hosteris_dir) rrorZresolved_namescandidate_namerZ candidate_dirZ lock_pathr msgZ vl_model_dirrqrqrs_get_model_local_pathwsV              z#_ModelManager._get_model_local_path)rrorcCst||}||Sr)rr)rrorZdownload_model_namesrqrqrsget_model_paths  z_ModelManager.get_model_pathcCst|tr|fnt|}d}t|D]\}}g}d}t|D]\} } || s(q|| z || } | WSty} z`| }|| } | rn| dt |k}|rht d| d|j d|| ddWYd} ~ qd}|dt |krtd|j d | t d | d |j d | d ||dj d WYd} ~ nd} ~ ww|r|r|dt |krnt d|d|j d||dj dq|rqqtd|dd|)NTrzModel package `z` was not found on z, trying fallback package `rFz-Encounter exception when download model from zN. No model source is available! Please check network or use local model files!z,Encountering exception when download model `z` from z: z2, will try to download from other model sources: `zModel packages z were not found on z, will try model source `z(No model source is available for model `rzA`! Please check model name and network, or use local model files!) rrr enumeraterrrrrrr r r)rr rorZlast_exceptionidxZhosterZattempted_candidatesZ"all_attempted_candidates_not_foundZ candidate_idxrZ model_patheZ is_not_foundZ has_fallbackrqrqrsrs         z#_ModelManager._download_from_hostercCs ||jvSr)rrrqrqrs __contains__rz_ModelManager.__contains__cCs ||Sr)rrrqrqrs __getitem__rz_ModelManager.__getitem__)rpN)rrrrrrrrrrrrr rrrrrrrrrrrrrqrqrqrsrCs*  4   8 rr)MrrrrrabcrrpathlibrtypingrrrrrZhuggingface_hubrZhuggingface_hub.utilsutilsrr Zset_verbosity_errorrenvironrZaistudio_sdk.errorsr Zaistudio_sdk.snapshot_downloadr rrZmodelscope.hub.errorsZhuberrorsZ_ms_hub_errorsrrrZ utils.cacherrZutils.downloadrZ utils.flagsrrrZmodels.utils.model_pathsrrrrir__annotations__rkrlrmrnrtrrrrrr BaseExceptionrrrrrrrrrrrqrqrqrss          +Q$      ;"1! -