,}h?UddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZmZmZmZmZmZmZmZmZmZddlmZddlZddlZddlmZddl m!Z!m"Z"dZ#dDde$d ee$ejJfd e&d dfd Z'dEd ee$ejJfd e&d e$fdZ(d ee$ejJfde$ded e)fdZ*dFd ee$ejJfdee$d e)fdZ+dGde$de&d e$fdZ,de$d ee$fdZ- dHde$dee$ejJfd eee$ejJfdee$de&d df dZ.dIdee$ejJfde)d ee$fdZ/dIdee$ejJfde$de)d ee$fdZ0 dJde$dee$ejJfd eee$ejJfdee$fdZ1dee$ejJfd ee$ejJfd!ee$d dfd"Z2e jfe jhd#Z5ee$e&fe6d$<dee$ejJfd ee$ejJfd!ee$d dfd%Z7e2e7d&Z8ee$eee$ejJfee$ejJfee$gdffe6d'<ejrejrejrd(Z:ee$ed)effe6d*<d+d+d,d-Z;ee$eee$ee$ffe6d.<d/ee$ejJfd ee$ee$ee$ffd0Z< dKdee$ejJfd eee$ejJfd1e)d ejJfd2Z= dKdee$ejJfd eee$ejJfd1e)d ee$ejJffd3Z> dLde$d4ee$ejJfd5eee$ejJfd eee$ejJfdee$d1e)d dfd6Z?d7ed e$fd8Z@ed9e$eAZB dMd:eBd;ee$dZCdNd?ee$ejJfd@e&d ejfdAZEdBejd ejfdCZGy)ON) AnyCallableDictIOIterableListOptionalTupleTypeVarUnion)urlparse)tqdm)#_download_file_from_remote_location_is_remote_location_availablezpytorch/visionurlfilename chunk_sizereturnctjjtjj|dti5}t |d5}t |j5}|j|x}r?|j||jt||j|x}r?dddddddddy#1swYxYw#1swYxYw#1swYyxYw)N User-Agentheaderswb)total) urllibrequesturlopenRequest USER_AGENTopenrlengthreadwriteupdatelen)rrrresponsefhpbarchunks Y/var/www/html/test/engine/venv/lib/python3.12/site-packages/torchvision/datasets/utils.py _urlretriever,s    6 6s\S]D^ 6 _ `(dl (D ! (RHOO)D (#==44%4 CJ'$==44%4 ( ((( ( ( ( (((s= C-C!)AC<C!C-C C!!C* &C--C6fpathcntjdk\rtjd}ntj}t |d5}|j |x}r%|j ||j |x}r%ddd|jS#1swY|jSxYw)N) F)usedforsecurityrb)sys version_infohashlibmd5r!r#r% hexdigest)r-rr6fr*s r+ calculate_md5r9#s 6!kk%0kkm eT avvj))e) JJu vvj))e) ==? ==?s 8BB4r6kwargsc |t|fi|k(SN)r9)r-r6r:s r+ check_md5r=1s -00 00c`tjj|sy|yt||S)NFT)ospathisfiler=)r-r6s r+check_integrityrC5s* 77>>%  { UC  r>max_hopsc v|}dtd}t|dzD]}}tjj tjj ||5}|j |k(s |j |cdddcS|j }dddtd|d|d|d#1swYxYw) NHEAD)Methodrrz Request to z exceeded z( redirects. The last redirect points to .)r rangerrrrrRecursionError)rrD initial_urlr_r's r+_get_redirect_urlrN=sKz:G 8a<   ^^ # #FNN$:$:3$:$P Q U]||s"hll&:  ,,C     +j :bcfbggh i    sB/ B//B8 ct|}tjd|jytjd|j}|y|j dS)Nz(drive|docs)[.]google[.]comz/file/d/(?P[^/]*)id)r rematchnetlocrAgroup)rpartsrRs r+_get_google_drive_file_idrVMsP SME xx. =E HH-uzz :E } ;;t r>rootmax_redirect_hopsctjj|}|stjj|}tjtjj ||}tj |dt||rtd|zytr t||nIt||}t|}|t||||S td|zdz|zt||t||s t)d y#tj j"t$f$rG}|ddd k(r3|j'd d }td |zdz|zt||n|Yd}~}d}~wwxYw)aDownload a file from a url and place it in root. Args: url (str): URL to download file from root (str): Directory to place downloaded file in filename (str, optional): Name to save the file under. If None, use the basename of the URL md5 (str, optional): MD5 checksum of the download. If None, do not check max_redirect_hops (int, optional): Maximum number of redirect hops allowed Texist_okz$Using downloaded and verified file: N)rDz Downloading  to httpszhttps:zhttp:z;Failed download. Trying https -> http instead. Downloading File not found or corrupted.)r@rA expanduserbasenamefspathjoinmakedirsrCprintrrrNrVdownload_file_from_google_driver,rerrorURLErrorOSErrorreplace RuntimeError)rrWrr6rXr-file_ides r+ download_urlrnZsj 77  d #D 77##C( IIbggll42 3EKKt$uc" 4u<=$&+E37 .?@,C0  27D(CP P  .3&/%7 8 e $ 5# &9:: ' %%w/ 2Aw'!kk(G4SVYY\bbejjkS%()  s& D#F=FFprefixc ztjj|}tj|Dcgc]B}tjj tjj ||sA|D}}|dur-|Dcgc]"}tjj ||$}}|Scc}wcc}w)a List all directories at a given root Args: root (str): Path to directory whose folders need to be listed prefix (bool, optional): If true, prepends the path to each result, otherwise only returns the name of the directories found T)r@rAr`listdirisdirrc)rWrop directoriesds r+list_dirrvs 77  d #D jj.W"''-- TST@U2V1WKW ~6ABrww||D!,B B XBsAB3:B3'B8suffixc tjj|}tj|Dcgc]T}tjj tjj ||sA|j |sS|V}}|dur-|Dcgc]"}tjj ||$}}|Scc}wcc}w)aList all files ending with a suffix at a given root Args: root (str): Path to directory whose folders need to be listed suffix (str or tuple): Suffix of the files to match, e.g. '.png' or ('.jpg', '.png'). It uses the Python "str.endswith" method and is passed directly prefix (bool, optional): If true, prepends the path to each result, otherwise only returns the name of the files found T)r@rAr`rqrBrcendswith)rWrwrorsfilesrus r+ list_filesr{s 77  d #D 4( i1BGGNN277<<a;P,QVWV`V`agVhQ iE i ~0561dA&66 L j6sAC:C C'C rlc ddl}tjj |}|s|}tj tjj||}tj|dt||rtd|rdndd |y|j||d t t||s td y#t$r tdwxYw) afDownload a Google Drive file from and place it in root. Args: file_id (str): id of file to be downloaded root (str): Directory to place downloaded file in filename (str, optional): Name to save the file under. If None, use the id of the file. md5 (str, optional): MD5 checksum of the download. If None, do not check rNz`To download files from GDrive, 'gdown' is required. You can install it with 'pip install gdown'.TrZzUsing downloaded z and verified zfile: F)rPoutputquiet user_agentr_) gdownModuleNotFoundErrorrkr@rAr`rbrcrdrCredownloadr )rlrWrr6rr-s r+rfrfs  77  d #D  IIbggll42 3EKKt$uc" !S/b!AwOP NNge5ZNP 5# &9:: '%   n   s CC from_pathto_path compressionctj||rd|ddnd5}|j|dddy#1swYyxYw)Nzr:rHr)tarfiler! extractall)rrrtars r+ _extract_tarrsH i;2k!"o%6!7C P TW w   s <A).bz2.xz_ZIP_COMPRESSION_MAPctj|d|r t|ntj5}|j |dddy#1swYyxYw)Nr)r)zipfileZipFiler ZIP_STOREDr)rrrzips r+ _extract_ziprsL 3$8$EZaZlZl   w   s A  A).tarz.zip_ARCHIVE_EXTRACTORS)r.gzr._COMPRESSED_FILE_OPENERS)rr)rr)z.tbzz.tbz2z.tgz_FILE_TYPE_ALIASESfilectj|j}|std|d|d}|tvr |gt|S|t vr||dfS|t vr(t|dkDr|d}|t vr||z||fS|d|fSttttt ztt z}td|d|d ) aDetect the archive type and/or compression of a file. Args: file (str): the filename Returns: (tuple): tuple of suffix, archive type, and compression Raises: RuntimeError: if file has no suffix or suffix is not supported zFile 'zP' has no suffixes that could be used to detect the archive type and compression.NrHz&Unknown compression or archive type: 'z'. Known suffixes are: 'z'.) pathlibPathsuffixesrkrrrr&sortedset)rrrwsuffix2valid_suffixess r+_detect_file_typers||D!**H TFj k  b\F##4+F344$$vt##)) x=1 rlG--'&88tV##C 23c:M6NNQTUmQnnoN ?xG`ao`pprs ttr>remove_finishedct|\}}}|std|d| CC&C# C&&C/c dtjdtttjfffd }|tj j }t\}}}|sattj j|tj jj|d|}||St|}||||rt j|tj|S)aDExtract an archive. The archive type and a possible compression is automatically detected from the file name. If the file is compressed but not an archive the call is dispatched to :func:`decompress`. Args: from_path (str): Path to the file to be extracted. to_path (str): Path to the directory the file will be extracted to. If omitted, the directory of the file is used. remove_finished (bool): If ``True``, remove the file after the extraction. Returns: (str): Path to the directory the file was extracted to. ret_pathrcRttrtj|S|Sr<) isinstancestrr@rb)rrs r+ path_or_strz$extract_archive..path_or_stras! i %99X& &Or>r})r)rrr rr@rAdirnamerrrcrarjrr) rrrrrwrrr extractors ` r+extract_archiverMs(glluS',,5F/G ''//),(9)(D%FL+   GGLL"''"2"29"="E"Efb"Q R+  8$$$L1I i+. ) w||G, --r> download_root extract_rootc$tjj|}||}|stjj|}t ||||tjj ||}t d|d|t|||y)Nz Extracting r\)r@rAr`rarnrcrer)rrrrr6rarchives r+download_and_extract_archiver}s}GG&&}5M$ 77##C(mXs3ggll=(3G Ky\N 34G\?;r>iterablec hddj|Dcgc] }t|c}zdzScc}w)N'z', ')rcr)ritems r+iterable_to_strrs- H=Dc$i=> > DD=s/ Tvaluearg valid_values custom_msgct|ts.|d}nd}|jt||}t |||S||vr9| |}t |d}|j||t |}t ||S)Nz'Expected type str, but got type {type}.z:Expected type str for argument {arg}, but got type {type}.)typerzPUnknown value '{value}' for argument {arg}. Valid values are {{{valid_values}}}.)rrr)rrformatr ValueErrorr)rrrrmsgs r+verify_str_argrs eS ! ;;CNCjjd5ksj3o  L  !CoeC**5cP\@]*^Co Lr> file_nameslice_channelsct|d5}|jj}|dvr tdt j d|j}|s t dd|jD\}}t|jj}|dkrd}| }nd }tj||d z } d d d d k(rdnd}  j| jddd} tj| d} | d |d d d d f} | jtjS#1swYzxYw)aDRead file in .pfm format. Might contain either 1 or 3 channels of data. Args: file_name (str): Path to the file. slice_channels (int): Number of channels to slice out of the file. Useful for reading different data formats stored in .pfm files: Optical Flows, Stereo Disparity Maps, etc. r2)PFsPfzInvalid PFM files^(\d+)\s(\d+)\s$zMalformed PFM header.c32K|]}t|ywr<)int).0dims r+ z_read_pfm..s7SC7sr<>r8)dtypeNrr/rHr)axis)r!readlinerstriprrQrR Exceptiongroupsfloatnpfromfilereshape transposeflipastypefloat32) rrr8header dim_matchwhscaleendiandata pfm_channelss r+ _read_pfmrs7 i 2!$$&  '/0 0HH11::<@ 34 47I$4$4$671ajjl))+, 19FFEF{{1FSL1#2&%1QL <<1l + 5 5aA >D 774a D A% &D ;;rzz ""122s CE  Etc>|jjtjjg|j|j j djg|jdddj|jS)Nr) contiguousviewtorchuint8shape element_sizerr)rs r+_flip_byte_orderrsW- EKK(--IqwwI8HINNrRWWjYZY`Y`adbdYejgijoopqpwpwxr>)i)ir<)r/)NNr/)F)NN)NF)NNNF)NNN)r)Hbz2gzipr5lzmar@os.pathrrQr3rr urllib.errorurllib.requestrtypingrrrrrrr r r r urllib.parser numpyrrtorch.utils.model_zoor_internally_replaced_utilsrrr rrrr,r9boolr=rCrNrVrnrvr{rfr ZIP_BZIP2ZIP_LZMAr__annotations__rrr!rrrrrrrbytesrrndarrayrTensorrr>r+r s    [[[! &k  (c(U3 +<%=(3(_c( sGLL01 s UX 1U3 ,-1C13141!5gll!23!(3-!SW!  3  #  c  3 8C=  48 5; 5; W\\! "5;uS',,./05; # 5;  5;  5;p 5gll*+ T d3i U3 ,-sDUYZ]U^(48 #; #; W\\! "#;uS',,./0#; # #;L S',,&' 27W\\8I2J YabeYf         (d38n  S',,&' 27W\\8I2J YabeYf     wT#xsGLL/@)A5gllIZC[]efi]j(kmq(qrrs HH 99 99:$sHS"W$556   FDeHSM8C=$@AAB(uE#w||"34(usHSMS[\_S`?`9a(uZ37!!!S',,&'!! eC-. /!!!! \\ !!L37!-.S',,&'-. eC-. /-.-. 3  -.f8<37! < <gll*+<5gll!234<uS',,./0 < # <  < <*EhE3E Ce *. $   #8A;'   8!#sGLL01!#3!#rzz!#H r>