Yh*>UddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZddlmZmZmZmZmZmZddlmZddlZddlZddlmZddlmZmZdZ dId e!d ee!ej"fd e#d dfdZ$dJdee!ej"fd e#d e!fdZ%dee!ej"fde!ded e&fdZ'dKdee!ej"fdee!d e&fdZ(dLd e!de#d e!fdZ)d e!d ee!fdZ* dMd e!dee!ej"fd eee!ej"fdee!de#d df dZ+dNdee!ej"fde&d e,e!fdZ-dNdee!ej"fd e!de&d e,e!fd!Z. dOd"e!dee!ej"fd eee!ej"fdee!fd#Z/d$ee!ej"fd%ee!ej"fd&ee!d dfd'Z0e j1e j2d(Z3e4e!e#fe5d)<d$ee!ej"fd%ee!ej"fd&ee!d dfd*Z6e0e6d+Z7e4e!eee!ej"fee!ej"fee!gdffe5d,<ej8ej8ej8d-Z9e4e!ed.effe5d/<d0d0d1d2Z:e4e!e;ee!ee!ffe5d3<d4ee!ej"fd e;e!ee!ee!ffd5Z< dPd$ee!ej"fd%eee!ej"fd6e&d ej"fd7Z= dPd$ee!ej"fd%eee!ej"fd6e&d ee!ej"ffd8Z> dQd e!d9ee!ej"fd:eee!ej"fd eee!ej"fdee!d6e&d dfd;Z?de!eAZB dRd?eBd@ee!dAeeeBdBee!d eBf dCZCdSdDee!ej"fdEe#d ejDfdFZEdGejFd ejFfdHZGdS)TN)Iterable)AnyCallableIOOptionalTypeVarUnion)urlparse)tqdm)#_download_file_from_remote_location_is_remote_location_availablezpytorch/visionurlfilename chunk_sizereturnc:tjtj|dti5}t |d5}t |jdd5}||x}rN| || t|||x}Ndddn #1swxYwYdddn #1swxYwYddddS#1swxYwYdS)N User-AgentheaderswbBT)totalunit unit_scale) urllibrequesturlopenRequest USER_AGENTopenr lengthreadwriteupdatelen)rrrresponsefhpbarchunks l/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/utils.py _urlretriever-s    6 6s\S]D^ 6 _ _ ` `(dl (D ! ! (RHO#Z^)_)_)_ (cg#==444% ( CJJ'''$==444% ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (((((((((((((((((((s[DC8/A&C! C8!C% %C8(C% )C8, D8C< <D?C< DDDfpathc*tjd}t|d5}||x}r,||||x},dddn #1swxYwY|S)NF)usedforsecurityrb)hashlibmd5r"r$r& hexdigest)r/rr4fr+s r, calculate_md5r7#s +e , , ,C eT  avvj)))e  JJu   vvj)))e  ==??sAA66A:=A:r4kwargsc $|t|fi|kSN)r7)r/r4r8s r, check_md5r;.s -0000 00cltj|sdS|dSt||S)NFT)ospathisfiler;)r/r4s r,check_integrityrA2s7 7>>% u {t UC  r<max_hopsc t|}dtd}t|dzD]}tjtj||5}|j|ks|j|cdddcS|j}dddn #1swxYwYtd|d|d|d) NHEAD)Methodrrz Request to z exceeded z( redirects. The last redirect points to .)r!rangerrrr rRecursionError)rrC initial_urlr_r(s r,_get_redirect_urlrM:s3Kz::G 8a<    ^ # #FN$:$:3$:$P$P Q Q U]|s""hl&:          ,C                  i+ i i i icf i i i   sBBB B ct|}tjd|jdStjd|j}|dS|dS)Nz(drive|docs)[.]google[.]comz/file/d/(?P[^/]*)id)r rematchnetlocr?group)rpartsrQs r,_get_google_drive_file_idrUJsW SMME x. ==Et H-uz : :E }t ;;t  r<rootmax_redirect_hopsctj|}|stj|}tjtj||}tj|dt||rdStrt||nt||}t|}|t||||S t||n_#tjjt"f$rA}|dddkr'|dd}t||n|Yd}~nd}~wwxYwt||st'd dS) 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)rChttpszhttps:zhttp:File not found or corrupted.)r>r? expanduserbasenamefspathjoinmakedirsrArr rMrUdownload_file_from_google_driver-rerrorURLErrorOSErrorreplace RuntimeError)rrVrr4rWr/file_ides r, download_urlrkWs 7  d # #D )7##C(( Ibgll422 3 3EKt$$$$uc""$&&+E37777 .?@@@,C00  27D(CPP P  e $ $ $ $ %w/   2A2w'!!kk(G44S%(((()((((  5# & &;9:::;;s/DE7EEFprefixctjfdtjD}|durfd|D}|S)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 cg|]A}tjtj|?|BS)r>r?isdirra).0prVs r, zlist_dir..s=WWW"'-- TST@U@U2V2VW1WWWr<TcPg|]"}tj|#Sror>r?rarqdrVs r,rszlist_dir..s)BBBrw||D!,,BBBr<r>r?r^listdir)rVrl directoriess` r,list_dirr{se 7  d # #DWWWWbj..WWWK ~~BBBBkBBB r<suffixctjfdtjD}|durfd|D}|S)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 cg|]V}tjtj|?|T|WSro)r>r?r@raendswith)rqrrrVr|s r,rszlist_files..sQ i i i1BGNN27<<a;P;P,Q,Q iVWV`V`agVhVh iQ i i ir<TcPg|]"}tj|#Srorurvs r,rszlist_files..s)6661dA&&666r<rx)rVr|rlfiless`` r, list_filesrsh 7  d # #D i i i i i 4(( i i iE ~~6666666 Lr<ric ddl}n#t$rtdwxYwtj|}|s|}tjtj||}tj|dt||rdS| ||dtt||stddS) 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'.TrYF)rOoutputquiet user_agentr]) gdownModuleNotFoundErrorrhr>r?r^r`rarbrAdownloadr!)rirVrr4rr/s r,rcrcs      n    7  d # #D  Ibgll422 3 3EKt$$$$uc"" NNge5ZNPPP 5# & &;9:::;;s! from_pathto_path compressionctj||r d|ddnd5}||ddddS#1swxYwYdS)Nzr:rGr)tarfiler" extractall)rrrtars r, _extract_tarrs i;!O!7k!""o!7!7!7C P P TW w                  sAA A ).bz2.xz_ZIP_COMPRESSION_MAPctj|d|r t|n tj5}||ddddS#1swxYwYdS)Nr)r)zipfileZipFiler ZIP_STOREDr)rrrzips r, _extract_ziprs 3$l$8$E$EZaZl     w                  sAAA).tarz.zip_ARCHIVE_EXTRACTORS)r.gzr._COMPRESSED_FILE_OPENERS)rr)rr)z.tbzz.tbz2z.tgz_FILE_TYPE_ALIASESfilectj|j}|std|d|d}|tvr|gt|RS|t vr||dfS|t vr1t|dkr|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.NrGz&Unknown compression or archive type: 'z'. Known suffixes are: 'z'.) pathlibPathsuffixesrhrrrr'sortedset)rrr|suffix2valid_suffixess r,_detect_file_typers4|D!!*H   kT k k k   b\F###4+F3444$$$vt##))) x==1  rlG---'&88tV##C 233c:M6N6NNQTUmQnQnnooN sssaosss t ttr<remove_finishedc&t|\}}}|std|d|>tjt j||||nd}t|}||d5}t|d5}| | dddn #1swxYwYdddn #1swxYwY|rt j |tj|S)aDecompress a file. The compression is automatically detected from the file name. Args: from_path (str): Path to the file to be decompressed. to_path (str): Path to the decompressed file. If omitted, ``from_path`` without compression extension is used. remove_finished (bool): If ``True``, remove the file after the extraction. Returns: (str): Path to the decompressed file. z*Couldn't detect a compression from suffix rHNr2r) rrhrrr>r`rgrr"r%r$remove) rrrr| archive_typercompressed_file_openerrfhwfhs r, _decompressr"s"):)(D(D%FL+ SQQQQRRR,ry33;;FT`TlLLrtuuvv6kB   4 0 0Cgt9L9LPS #((** ) <  s6C(C; CC CC CC"%C"c Bdtjdtttjfffd }|tj}t\}}}|slttj |tj  |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_pathrcZttrtj|S|Sr:) isinstancestrr>r`)rrs r, path_or_strz$extract_archive..path_or_strZs* i % % 9X&& &Or<Nr)r)rrr rr>r?dirnamerrrar_rgrr) rrrrr|rrr extractors ` r,extract_archiverFs$(gluS',5F/G '//),,(9)(D(D%FL+ %  GLL"'"2"29"="="E"Efb"Q"Q R R+    {8$$$$L1I Ii+... ) ;w|G,, - --r< download_root extract_rootctj|}||}|stj|}t ||||tj||}t |||dSr:)r>r?r^r_rkrar)rrrrr4rarchives r,download_and_extract_archivervsG&&}55M$ )7##C((mXs333gll=(33GG\?;;;;;r<iterablecLddd|DzdzS)N'z', 'c,g|]}t|Sro)r)rqitems r,rsz#iterable_to_str..s===Dc$ii===r<)ra)rs r,iterable_to_strrs- ==H===>> > DDr<Tvaluearg valid_values custom_msgc*t|ts:|d}nd}|t||}t |||S||vr;||}n'd}|||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 ! ! ;;CCNCjjd5kksj33oo  L   !CCdC**5cP\@]@]*^^Coo Lr< file_nameslice_channelsc"t|d5}|}|dvrtdt jd|}|st dd|D\}}t|}|dkrd}| }nd }tj ||d z } d d d n #1swxYwY|d krdnd} | |||  ddd} tj | d} | d |d d d d f} | tjS)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.c34K|]}t|VdSr:)int)rqdims r, z_read_pfm..s(77SC777777r<r<>r6)dtypeNrrBrGr )axis)r"readlinerstriprrPrQ Exceptiongroupsfloatnpfromfilereshape transposeflipastypefloat32) rrr6header dim_matchwhscaleendiandata pfm_channelss r, _read_pfmrs i  2!$$&&  ' '/00 0H11::<<@@  5344 477I$4$4$6$67771ajjll))++,, 199FFEEF{1FSL111#222222222222222&%11QL <<1l + + 5 5aA > >D 74a D AAA% &D ;;rz " ""sC-D  DDtc6|tjjg|j|Rdjg|jdddR|jS)Nr) contiguousviewtorchuint8shape element_sizerr)rs r,_flip_byte_orderrsW- EK((-IqwI8H8HIIINNrRRWjYZY`adbdadYejgijjjoopqpwxxr<)r)r.r:)rB)NNrB)F)NN)NF)NNNF)NNN)r )Hbz2gzipr3lzmar>os.pathrrPrr urllib.errorurllib.requestrcollections.abcrtypingrrrrrr urllib.parser numpyrrtorch.utils.model_zoor _internally_replaced_utilsr rr!rrrr-r7boolr;rArMrUrklistr{rrcr ZIP_BZIP2ZIP_LZMArdict__annotations__rrr"rrtuplerrrrrbytesrrndarrayrTensorrror<r,rs]    $$$$$$>>>>>>>>>>>>>>>>!!!!!! &&&&&&kkkkkkkk  ((c(U3 +<%=(3(_c((((sGL01sUX1U3 ,-1C13141111!!5gl!23!(3-!SW!!!!   3  #  c     3 8C=     48 2;2; 2; W\! "2;uS',./02; # 2;  2;  2;2;2;2;j  5gl*+ T d3i    U3 ,-sDUYZ]U^(48 ";"; "; W\! "";uS',./0"; # ";";";";J S',&' 27W\8I2J YabeYf          ((d38n  S',&' 27W\8I2J YabeYf        wwT#xsGL/@)A5glIZC[]efi]j(kmq(qrrs H 9 9::$sHS"W$556   FFDeHSM8C=$@AAB(uE#w|"34(usHSMS[\_S`?`9a(u(u(u(uZ37!!!!!S',&'!! eC-. /!!!! \ !!!!!!!!L37!-.-.S',&'-. eC-. /-.-. 3  -.-.-.-.f8<37! << <gl*+<5gl!234<uS',./0 < # <  < <<<<(EhE3EEEE GCe *. $   #8A;'   8!#!#sGL01!#3!#rz!#!#!#!#H r<