PhI*UddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZ ddlZddlZddlmZmZmZmZmZmZmZmZmZmZddlmZddlZddl Z ddl!Z!ddl"m#Z#ddl$m%Z%m&Z&dZ' dIdee(d e)d ee*d dfd Z+dJd e)de)de*d dfdZ,dKde)de*d e)fdZ-de)de)ded e.fdZ/dIde)dee)d e.fdZ0dLd e)de*d e)fdZ1d e)d ee)fdZ2 dMd e)de)dee)dee)de*d df dZ3dNde)de.d ee)fdZ4dNde)de)de.d ee)fd Z5dJde*d ee(ee(ffd!Z6dOd"e)de)dee)dee)fd#Z7d$e)d%e)d&ee)d dfd'Z8ejrejtd(Z;ee)e*feee)ee)e)ee)gdffee)e(ZG dRd?eGd@ee)dAeeeGdBee)d eGf dCZHdSdDe)dEe*d ejfdFZJdGe!jd e!jfdHZLy)TN) AnyCallableDictIOIterableIteratorListOptionalTupleTypeVar)urlparse)tqdm)#_download_file_from_remote_location_is_remote_location_availablezpytorch/visioncontent destinationlengthreturnct|d5}t|5}|D]0}|s|j||jt |2 ddddddy#1swYxYw#1swYyxYw)Nwb)total)openrwriteupdatelen)rrrfhpbarchunks eC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/datasets/utils.py_save_response_contentr!s] k4 B6(:dE HHUO KKE # ); (:(: s" A.6A"A."A+ 'A..A7urlfilename chunk_sizectjjtjj|dti5t t fdd|jdddy#1swYyxYw)N User-Agentheadersc&jSN)read)r$responsesr z_urlretrieve..0sHMM*,Er.)r)urllibrequesturlopenRequest USER_AGENTr!iterr)r"r#r$r,s `@r _urlretriever5.sZ    6 6s\S]D^ 6 _ `dlt$EsKX^f^m^mn a ` `s 'A77Bfpathcntjdk\rtjd}ntj}t |d5}|j |x}r%|j ||j |x}r%ddd|jS#1swY|jSxYw)N) F)usedforsecurityrb)sys version_infohashlibmd5rr+r hexdigest)r6r$r?frs r calculate_md5rB3s 6!kk%0kkm eT avvj))e) JJu vvj))e)  ==?  ==?s 8BB4r?kwargsc |t|fi|k(Sr*)rB)r6r?rCs r check_md5rEAs -00 00r.c`tjj|sy|yt||S)NFT)ospathisfilerE)r6r?s r check_integrityrJEs* 77>>%  { UC  r.max_hopsc v|}dtd}t|dzD]}}tjj tjj ||5}|j |k(s |j |cdddcS|j }dddtd|d|d|d#1swYxYw) NHEAD)Methodr&r'z Request to z exceeded z( redirects. The last redirect points to .)r3ranger/r0r1r2r"RecursionError)r"rK initial_urlr(_r,s r _get_redirect_urlrUMsKz:G 8a<  ^^ # #FNN$:$:3$:$P QU]||s"hll&:R Q,,C R Q!+j :bcfbggh i  R QsB/ B//B8 ct|}tjd|jytjd|j}|y|j dS)Nz(drive|docs)[.]google[.]comz/file/d/(?P[^/]*)id)r rematchnetlocrHgroup)r"partsrYs r _get_google_drive_file_idr]]sP SME xx. =E HH-uzz :E } ;;t r.rootmax_redirect_hopsctjj|}|stjj|}tjj ||}tj |dt ||rtd|zytr t||nIt||}t|}|t||||S td|zdz|zt||t ||s t'd y#tjj 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)rKz Downloading  to httpszhttps:zhttp:z;Failed download. Trying https -> http instead. Downloading zFile not found or corrupted.)rGrH expanduserbasenamejoinmakedirsrJprintrrrUr]download_file_from_google_driver5r/errorURLErrorOSErrorreplace RuntimeError)r"r^r#r?r_r6file_ides r download_urlrsjsa 77  d #D 77##C( GGLLx (EKKt$uc" 4u<=$&+E37 .?@,C0  27D(CP P  .3&/%7 8 e $ 5# &9:: ' %%w/ 2Aw'!kk(G4SVYY\bbejjkS%()  s D #E5.=E00E5prefixc 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)rGrHrflistdirisdirrh)r^rtp directoriesds r list_dirr{s 77  d #D jj.W."''-- TST@U2V1.KW ~6ABkrww||D!,k 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)rGrHrfrvrIrhendswith)r^r|rtrxfilesrzs r list_filesrs 77  d #D 4( i(1BGGNN277<<a;P,QVWV`V`agVhQ(E i ~0561dA&6 L j6sAC:C C'C c|j|}d}|st|}|stj|g|} t j d|j }||dnd}||fS#t$rd}Y||fSwxYw)Nz3Google Drive - (?P<api_response>.+?) api_response) iter_contentnext itertoolschainrXsearchdecodeUnicodeDecodeError)r,r$r first_chunkrYrs r _extract_gdrive_api_responsers##J/GK7m oo{mW5G OQ\QcQcQef050Au^,t        s-A-- A>=A>rqcftjj|}|s|}tjj||}tj|dt ||rt d|rdndd|yd}t|d }tj5}|j||d }|jjD]\} } | jd s| } nt|\} } | d k(rdnd} | -|j|t|| d }t|\} }  dk(rtd|dt! |dddtj"|j$dkrtt'j(t*5t-|5}|j/}t1j2d|rt5j6d|dddddd|rt9||std|dyy#1swYxYw#1swY]*\s*>|(&(?:[\w\d]+|#\d+|#x[a-f\d]+);)zWe detected some HTML elements in the downloaded file. This most likely means that the download triggered an unhandled API response by GDrive. Please report this to torchvision at https://github.com/pytorch/vision/issues including the response: z&The MD5 checksum of the download file z does not match the one on record.Please delete the file and try again. If the issue persists, please report this to torchvision at https://github.com/pytorch/vision/issues.)rGrHrfrhrirJrjdictrequestsSessiongetcookiesitems startswithrrpr!statst_size contextlibsuppressrrr+rXrwarningswarnrE)rqr^r#r?r6r"rsessionr,keyvaluetokenrrrtexts r rkrks 77  d #D  GGLLx (EKKt$uc" !S/b!AwOP 'C WZ 0F    w;;s6$;?"**002JC~~013 %A$J !L''+??CTE  {{3tFE/JSW{XH$@$J !L' + +.xj9BC  w.- 2 wwu~ )  !3 4d5kR779DyyQSWX ()-v/ 7B 4 9UC(4UG i;2k!"o%6!7C PTW w Q P Ps <A).bz2.xz_ZIP_COMPRESSION_MAPctj|d|r t|ntj5}|j |dddy#1swYyxYw)Nr)r)zipfileZipFiler ZIP_STOREDr)rrrzips r _extract_ziprsE 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.NrOz&Unknown compression or archive type: 'z'. Known suffixes are: 'z'.) pathlibPathsuffixesrprrrrsortedset)rrr|suffix2valid_suffixess r _detect_file_typer5s||D!**H TFj k  b\F##4+F344$$vt##)) x=1 rlG--'&88tV##C 23c:M6NNQTUmQnnoN ?xG`ao`pprs ttr.remove_finishedcrt|\}}}|std|d||j|||nd}t|}||d5}t |d5}|j |j dddddd|rtj||S#1swY*xYw#1swY.xYw)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 rPNrr;r) rrprorrrr+rGremove) rrrr| archive_typercompressed_file_openerrfhwfhs r _decompressr`s):)(D%FL+ GxqQRR##FL > DD=s/ Trarg 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) isinstancerformatr 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. r;)PFsPfzInvalid PFM files^(\d+)\s(\d+)\s$zMalformed PFM header.c32K|]}t|ywr*)int).0dims r z_read_pfm..s7$6SC$6sr<>rA)dtypeNrr8rOr)axis)rreadlinerstriprrXrY Exceptiongroupsfloatnpfromfilereshape transposeflipastypefloat32) rrrAheader dim_matchwhscaleendiandata pfm_channelss r _read_pfmr s0 i !$$&  '/0 0HH11::<@ 34 47I$4$4$671ajjl))+, 19FFEF{{1FSL1# &%1QL <<1l + 5 5aA >D 774a D A% &D ;;rzz ""1  s CE  Erc>|jjtjjg|j|j j djg|jdddj|jS)Nr) contiguousviewtorchuint8shape element_sizerr)rs r _flip_byte_orderrsW- EKK(--IqwwI8HINNrRWWjYZY`Y`adbdYejgijoopqpwpwxr.r*)i)i)r8)NNr8)F)NN)NF)NNNF)NNN)r)Mbz2rgzipr>rlzmarGos.pathrrXr<rr/ urllib.errorurllib.requestrrtypingrrrrrrr r r r urllib.parser numpyrrrtorch.utils.model_zoor_internally_replaced_utilsrrr3bytesrrr!r5rBboolrErJrUr]rsr{rrrkr ZIP_BZIP2ZIP_LZMAr__annotations__rrrrrrrrrrrrndarrayr Tensorrr.r r)s     ^^^! &k  ! $ e_ $ $ SM $ $ocoSoco$o  #  1S1s1c1d1!3!Xc]!d!  3  #  c  3 8C= no1; 1;1;#+C=1;>Fsm1;gj1; 1;h 3  c S#tS " !s !5QVX`afXgQgKh ! B SB B xPS}B bjknboB J C # HSM d      (d38n  C # HSM d   MT#xc8C=(A4(GHHI HH 99 99:$sHS"W$556   FDeHSM8C=$@AAB(uC(uE#x}hsm*K$L(uV3#PTad@!s!Xc]!TX!eh!N#'"! < <<3-<sm < # <  < <*EhE3E Ce *. $   #8A;'   8!#!#c!#"**!#H r.