Di*>UddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZddlmZmZmZmZmZmZddlmZddlZddlZddlmZddlmZmZdZ dEd e!d ee!ejDfd e#d dfd Z$dFdee!ejDfd e#d e!fdZ%dee!ejDfde!ded e&fdZ'dGdee!ejDfdee!d e&fdZ(dHd e!de#d e!fdZ)d e!d ee!fdZ* dId e!dee!ejDfd eee!ejDfdee!de#d df dZ+dJdee!ejDfde&d e,e!fdZ-dJdee!ejDfde!de&d e,e!fdZ. dKde!dee!ejDfd eee!ejDfdee!fdZ/d ee!ejDfd!ee!ejDfd"ee!d dfd#Z0e jbe jdd$Z3e4e!e#fe5d%<d ee!ejDfd!ee!ejDfd"ee!d dfd&Z6e0e6d'Z7e4e!eee!ejDfee!ejDfee!gdffe5d(<ejpejpejpd)Z9e4e!ed*effe5d+<d,d,d-d.Z:e4e!e;ee!ee!ffe5d/<d0ee!ejDfd e;e!ee!ee!ffd1Z< dLd ee!ejDfd!eee!ejDfd2e&d ejDfd3Z= dLd ee!ejDfd!eee!ejDfd2e&d ee!ejDffd4Z> dMd e!d5ee!ejDfd6eee!ejDfd eee!ejDfdee!d2e&d dfd7Z?d8ed e!fd9Z@ed:e!eAZB dNd;eBdee!d eBf d?ZCdOd@ee!ejDfdAe#d ejfdBZEdCejd ejfdDZGy)PN)Iterable)AnyCallableIOOptionalTypeVarUnion)urlparse)tqdm)#_download_file_from_remote_location_is_remote_location_availablezpytorch/visionurlfilename chunk_sizereturnctjjtjj|dti5}t |d5}t |jdd5}|j|x}r?|j||jt||j|x}r?dddddddddy#1swYxYw#1swYxYw#1swYyxYw)N User-AgentheaderswbBT)totalunit unit_scale) urllibrequesturlopenRequest USER_AGENTopenr lengthreadwriteupdatelen)rrrresponsefhpbarchunks \/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/torchvision/datasets/utils.py _urlretriever,s    6 6s\S]D^ 6 _ `(dl (D ! (RHOO#Z^)_ (cg#==44%4 CJ'$==44%4 ( ((( ( ( ( (((s= C/C#+AC>C#C/C C##C, (C//C8fpathctjd}t|d5}|j|x}r%|j ||j|x}r%ddd|j S#1swY|j SxYw)NF)usedforsecurityrb)hashlibmd5r!r#r% hexdigest)r-rr2fr*s r+ calculate_md5r5#s ++e ,C eT avvj))e) JJu vvj))e) ==? ==?s 8A44B r2kwargsc |t|fi|k(SN)r5)r-r2r6s r+ check_md5r9.s -00 00c`tjj|sy|yt||S)NFT)ospathisfiler9)r-r2s r+check_integrityr?2s* 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)rr@ initial_urlr_r's r+_get_redirect_urlrJ: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 rematchnetlocr=group)rpartsrNs r+_get_google_drive_file_idrRJsP SME xx. =E HH-uzz :E } ;;t r:rootmax_redirect_hopsctjj|}|stjj|}tjtjj ||}tj |dt||rytr t||n5t||}t|}|t||||S t||t||s t'd y#tjj t"f$r3}|dddk(r|j%dd}t||n|Yd}~id}~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_okN)r@httpszhttps:zhttp:File not found or corrupted.)r<r= expanduserbasenamefspathjoinmakedirsr?rr rJrRdownload_file_from_google_driver,rerrorURLErrorOSErrorreplace RuntimeError)rrSrr2rTr-file_ides r+ download_urlrhWs* 77  d #D 77##C( IIbggll42 3EKKt$uc"$&+E37 .?@,C0  27D(CP P  e $ 5# &9:: ' %%w/ 2Aw'!kk(G4S%() s C<<#E)E  Eprefixc 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<r=r[listdirisdirr^)rSrip directoriesds r+list_dirrps 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<r=r[rkr>r^endswith)rSrqrirmfilesros r+ list_filesrus 77  d #D 4( i1BGGNN277<<a;P,QVWV`V`agVhQ iE i ~0561dA&66 L j6sAC:C C'C rfc ddl}tjj |}|s|}tj tjj||}tj|dt||ry|j||dtt||s tdy#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'.TrVF)rLoutputquiet user_agentrZ) gdownModuleNotFoundErrorrer<r=r[r]r^r_r?downloadr )rfrSrr2rzr-s r+r`r`s  77  d #D  IIbggll42 3EKKt$uc" NNge5ZNP 5# &9:: '#   n   s B11C from_pathto_path compressionctj||rd|ddnd5}|j|dddy#1swYyxYw)Nzr:rDr)tarfiler! extractall)r}r~rtars 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)r}r~rzips 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.NrDz&Unknown compression or archive type: 'z'. Known suffixes are: 'z'.) pathlibPathsuffixesrerrrr&sortedset)rrrqsuffix2valid_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|Sr8) isinstancestrr<r])rr}s r+ path_or_strz$extract_archive..path_or_strZs! i %99X& &Or:r)r)rrr rr<r=dirnamerrr^r\rdrr) r}r~rrrqrrr extractors ` r+extract_archiverFs(glluS',,5F/G ''//),(9)(D%FL+   GGLL"''"2"29"="E"Efb"Q R+  8$$$L1I i+. ) w||G, --r: download_root extract_rootctjj|}||}|stjj|}t ||||tjj ||}t |||yr8)r<r=r[r\rhr^r)rrrrr2rarchives r+download_and_extract_archivervshGG&&}5M$ 77##C(mXs3ggll=(3GG\?;r:iterablec hddj|Dcgc] }t|c}zdzScc}w)N'z', ')r^r)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. r0)PFsPfzInvalid PFM files^(\d+)\s(\d+)\s$zMalformed PFM header.c32K|]}t|ywr8)int).0dims r+ z_read_pfm..s7SC7sr<>r4)dtypeNrrDr )axis)r!readlinerstriprrMrN Exceptiongroupsfloatnpfromfilereshape transposeflipastypefloat32) rrr4header 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)ir8)r)NNr)F)NN)NF)NNNF)NNN)r )Hbz2gzipr1lzmar<os.pathrrMrr urllib.errorurllib.requestrcollections.abcrtypingrrrrrr urllib.parser numpyrrtorch.utils.model_zoor _internally_replaced_utilsr rr rrrr,r5boolr9r?rJrRrhlistrprur`r ZIP_BZIP2ZIP_LZMArdict__annotations__rrr!rrtuplerrrrrbytesrrndarrayrTensorrr:r+r s    $>>! &k  (c(U3 +<%=(3(_c(sGLL01sUX1U3 ,-1C13141!5gll!23!(3-!SW!  3  #  c  3 8C=  48 2; 2; W\\! "2;uS',,./02; # 2;  2;  2;j 5gll*+ T d3i U3 ,-sDUYZ]U^(48 "; "; W\\! "";uS',,./0"; # ";J 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: