NgjddlZddlZddlmZmZddlmZddlZddlZddl m Z ddl m Z ddl m Z m Z ddlmZdd lmZd d id d id Zddddddddid ZgdZdedefdZdZdZGdde ZGdde ZejeejedS)N)AnyOptional)urlparse) PathHandler)HTTPURLHandler) get_cache_dir file_lock)download) PathManagerppyolov2_r50vd_dcn_365ezdhttps://paddle-model-ecology.bj.bcebos.com/model/layout-parser/ppyolov2_r50vd_dcn_365e_publaynet.tarzihttps://paddle-model-ecology.bj.bcebos.com/model/layout-parser/ppyolov2_r50vd_dcn_365e_tableBank_word.tar) PubLayNet TableBankTextTitleListTableFigure)rr )zinference.pdiparamszinference.pdiparams.infozinference.pdmodeltar_filereturnctj|}tjtj|d}tj||}|S)Nr)ospathdirnamesplitextbasenamejoin)r base_path file_name target_folders g/var/www/html/ai-engine/env/lib/python3.11/site-packages/layoutparser/models/paddledetection/catalog.py_get_untar_directoryr%AsZ))I  !1!1(!;!;<>  Y 3A 677  2 W^^BGLL4G4JKK L L2 I---- \)S ) ) 2V ++-- 2 2%811M$ 33#0#))&11"',,y(;;TBB2j$$TYY[[111222222222222222 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 s7 A4F ?(E3' F 3E7 7F :E7 ;F  FFct|}tj|sdStD]B}tjtj||sdSCdS)NFT)r%rrr+r,r )cachedpossible_extracted_model_folderrs r$!is_cached_folder_exists_and_validrAbss&:6&B&B# 7>>9 : :u'w~~bgll+JHUUVV 55  4r&c PeZdZdZdZdZdZ d deded e ed e d ef d Z dS)PaddleModelURLHandlerz@ Supports download and file check for Baidu Cloud links cdgS)Nz*https://paddle-model-ecology.bj.bcebos.comselfs r$_get_supported_prefixesz-PaddleModelURLHandler._get_supported_prefixesss <==r&c||jvSN) cache_map)rHrs r$_isfilezPaddleModelURLHandler._isfilevst~%%r&FNrforce cache_dirkwargsrc |||s4||jvs+tj|j|s7t jt}t|}tj t|tj |j d}| dd}t||jkr&|dddzt!jjz}tj ||} t'| rt)| |j|<n t+| 5tj| s:|d|t3|||} |drDt7| } tj| n#|d | YnxYwdddn #1swxYwY|d || | |j|<|j|S) a As paddle model stores all files in tar files, we need to extract them and get the newly extracted folder path. This function rewrites the base function to support the following situations: 1. If the tar file is not downloaded, it will download the tar file, extract it to the target folder, delete the downloaded tar file, and return the folder path. 2. If the extracted target folder is present, and all the necessary model files are present (specified in _TAR_FILE_NAME_LIST), it will return the folder path. 3. If the tar file is downloaded, but the extracted target folder is not present (or it doesn't contain the necessary files in _TAR_FILE_NAME_LIST), it will extract the tar file to the target folder, delete the tar file, and return the folder path. /Nd_zDownloading {} ...)r9z.tarz'Not able to remove the cached tar file zURL {} cached in {}) _check_kwargsrLrrr+logging getLogger__name__rr rrlstripsplitlenMAX_FILENAME_LENuuiduuid4hexrAr%r isfileinfoformatr endswithr=removewarning) rHrrNrOrPlogger parsed_urlrr9r?r6s r$_get_local_pathz%PaddleModelURLHandler._get_local_pathys0 6""" & 14>))7>>$."677*&x00F!$Jglli(("'//*/:P:PQT:U:U*V*VGzz#r*H8}}t444#DSD>C/$*,,2BBW\\'844F088 1(>&11L $8$?$?$E$EFFF!)$(!K!K!K}}V,,$8$@$@ If----"NN R& R R                188yIIJJJ'0t$~d##s+A>IHIH64II  I )FN) rY __module__ __qualname____doc__r]rIrMstrboolrrrirFr&r$rCrCls>>>&&& #' A$A$A$A$C= A$  A$ A$A$A$A$A$A$r&rCc*eZdZdZdZdZdZddZdS) LayoutParserPaddleModelHandlerz< Resolve anything that's in LayoutParser model zoo. zlp://paddledetection/c|jgSrK)PREFIXrGs r$rIz6LayoutParserPaddleModelHandler._get_supported_prefixess  }r&c |t|jd}|d^}}}|dkr't|d|}nt d|t j|fi|S)NrRweightzUnknown data_type )r\rrr[ MODEL_CATALOGr ValueErrorr get_local_path)rHrrP model_name dataset_name data_type model_urls r$riz.LayoutParserPaddleModelHandler._get_local_paths#dk**,,- /9/?/?/D/D, z9  %l3CHHZ4H4HIII=)==>> >))>>v>>>r&r)c Ntj|||fi|SrK)r r/ri)rHrmoderPs r$_openz$LayoutParserPaddleModelHandler._opens+ 4 4T : :DKKFKKKr&N)r))rYrjrkrlrrrIrir~rFr&r$rprps]%F???LLLLLLr&rp)rrWtypingrr urllib.parserr.r^iopath.common.file_iorrrr iopath.common.downloadr base_catalogr ruLABEL_MAP_CATALOGr,rmr%r=rArCrpregister_handlerrFr&r$rs   !!!!!! ------000000::::::::++++++&&&&&& "$J "$O          7   330N$N$N$N$N$NN$N$N$bLLLLL[LLL0 2244555 ;;==>>>>>r&