Ei*>&%SSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK r SSK r SSK r SSK r SSK Jr SSKJrJrJrJrJrJr SSKJr SSKrSSKrSSKJr SSKJrJr Sr SES \!S \\!\RD4S \#S S4S jjr$SFS\\!\RD4S \#S \!4Sjjr%S\\!\RD4S\!S\S \&4Sjr'SGS\\!\RD4S\\!S \&4Sjjr(SHS \!S\#S \!4Sjjr)S \!S \\!4Sjr*SIS \!S\\!\RD4S \\\!\RD4S\\!S\#S S4 Sjjr+SJS\\!\RD4S\&S \,\!4Sjjr-SJS\\!\RD4S\!S\&S \,\!4Sjjr.SKS\!S\\!\RD4S \\\!\RD4S\\!4Sjjr/S \\!\RD4S!\\!\RD4S"\\!S S4S#jr0\ Rb\ RdS$.r3\4\!\#4\5S%'S \\!\RD4S!\\!\RD4S"\\!S S4S&jr6\0\6S'.r7\4\!\\\!\RD4\\!\RD4\\!/S44\5S('\Rp\Rp\RpS).r9\4\!\S*\44\5S+'S,S,S-S..r:\4\!\;\\!\\!44\5S/'S0\\!\RD4S \;\!\\!\\!44S1jr<SLS \\!\RD4S!\\\!\RD4S2\&S \RD4S3jjr=SLS \\!\RD4S!\\\!\RD4S2\&S \\!\RD44S4jjr>SMS \!S5\\!\RD4S6\\\!\RD4S \\\!\RD4S\\!S2\&S S4S7jjr?S8\S \!4S9jr@\"S:\!\A5rBSNS;\BS<\\!S=\\\BS>\\!S \B4 S?jjrCSOS@\\!\RD4SA\#S \R4SBjjrESC\RS \R4SDjrGg)PN)Iterable)AnyCallableIOOptionalTypeVarUnion)urlparse)tqdm)#_download_file_from_remote_location_is_remote_location_availablezpytorch/visionurlfilename chunk_sizereturnc[RR[RRUS[0S95n[ US5n[ URSSS9nURU5=n(aEURU5 UR[U55 URU5=n(aMESSS5 SSS5 SSS5 g!,(df  N=f!,(df  N(=f!,(df  g=f)N User-AgentheaderswbBT)totalunit unit_scale) urllibrequesturlopenRequest USER_AGENTopenr lengthreadwriteupdatelen)rrrresponsefhpbarchunks W/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torchvision/datasets/utils.py _urlretriever,s    6 6s\S]D^ 6 _ `dl (D !RHOO#Z^)_cg#==44%4 CJ'$==44%4*` ! a `)_)_ ! ! a `s= C>C-%ACC- C> C* &C-- C; 7C>> D fpathc:[R"SS9n[US5nURU5=n(a+UR U5 URU5=n(aM+SSS5 UR 5$!,(df  UR 5$=f)NF)usedforsecurityrb)hashlibmd5r!r#r% hexdigest)r-rr2fr*s r+ calculate_md5r5#s~ ++e ,C eT avvj))e) JJu vvj))e)  ==?   ==?s AA== Br2kwargsc U[U40UD6:H$N)r5)r-r2r6s r+ check_md5r9.s -00 00cj[RRU5(dgUcg[X5$)NFT)ospathisfiler9)r-r2s r+check_integrityr?2s+ 77>>%  { U  r:max_hopsc UnS[S.n[US-5H|n[RR [RR XS95nUR U:Xd UR c UsSSS5 s $UR nSSS5 M~ [SUSUSUS35e!,(df  M=f) 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 QU]||s"hll&:R Q,,C R Q!+j :bcfbggh i  R QsB/ B// B> c[U5n[R"SUR5cg[R"SUR5nUcgUR S5$)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_hopsc[RRU5nU(d[RRU5n[R"[RR X55n[R "USS9 [XS5(ag[5(a [XP5 O/[XS9n[U5nUb [XaX#5$[X5 [XS5(d ['S 5eg![RR ["4a2nUSSS:Xa"UR%SS5n[X5 SnANkUeSnAff=f) 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$u""$&&+E7 @,C0  27(P P   $ 5 & &9:: ' %%w/ 2Aw'!kk(G4S((  s D$E)&EEEprefixc [RRU5n[R"U5Vs/sHHn[RR [RR X55(dMFUPMJ nnUSLa/UVs/sH"n[RR X5PM$ nnU$s snfs snf)zList 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."''-- T@U2V1.KW ~6ABkrww||D,k B XBsAB?B?)Csuffixc [RRU5n[R"U5Vs/sH`n[RR [RR X55(dMFUR U5(dM^UPMb nnUSLa/UVs/sH"n[RR X5PM$ nnU$s snfs snf)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( i(1BGGNN277<<;P,QQVWV`V`agVhQ(E i ~0561d&6 L j6sACCC*)CrfcSSKn[RR U5nU(dUn[R "[RRX55n[R"USS9 [XS5(agURXS[S9 [XS5(d [S5eg![a [S5ef=f) aNDownload 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$u"" NNg5ZNP 5 & &9:: '#   n   s B88C from_pathto_path compressionc[R"X(aSUSS3OS5nURU5 SSS5 g!,(df  g=f)Nzr:rDr)tarfiler! extractall)r}r~rtars r+ _extract_tarrs@ i;2k!"o%6!7C PTW w Q P Ps A A).bz2.xz_ZIP_COMPRESSION_MAPc[R"USU(a [UO[RS9nUR U5 SSS5 g!,(df  g=f)Nr)r)zipfileZipFiler ZIP_STOREDr)r}r~rzips r+ _extract_ziprsG 3$8$EZaZlZl   w   s A A).tarz.zip_ARCHIVE_EXTRACTORS)r.gzr._COMPRESSED_FILE_OPENERS)rr)rr)z.tbzz.tbz2z.tgz_FILE_TYPE_ALIASESfilec[R"U5RnU(d[SUS35eUSnU[;a U/[UQ7$U[ ;aX"S4$U[ ;a)[U5S:aUSnU[ ;aX2-X24$USU4$[[[5[[ 5-[[ 5-5n[SUSUS 35e) zDetect 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$$t##)) x=1 rlG--'88tV##C 23c:M6NNQTUmQnnoN ?xG`ao`pprs ttr:remove_finishedc[U5up4nU(d[SUS35eUc>[R"[R "U5R X4bUOS55n[UnU"US5n[US5nURUR55 SSS5 SSS5 U(a[R"U5 [R"U5$!,(df  NI=f!,(df  NR=f)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 rENr0r) rrerrr<r]rdrr!r$r#remove) r}r~rrq archive_typercompressed_file_openerrfhwfhs r+ _decompressr"s"):)(D%F+ GxqQRR,,ryy3;;FTlLrtuv6kB  4 0Cgt9LPS #((*:M 0 ) <<  :M9L 0 0s$8 C9 C(%C9( C6 2C99 Dc .^S[RS[[[R44U4SjjnUc[R R T5n[T5upEnU(d^[T[R RU[R RT5RUS55US9nU"U5$[UnU"TX5 U(a[R"T5 U"[R"U55$)aExtract 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_pathrc^>[T[5(a[R"U5$U$r8) isinstancestrr<r])rr}s r+ path_or_str$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%F+   GGLL"''"2"29"="E"Efb"Q R+  8$$$L1I i. ) w||G, --r: download_root extract_rootc[RRU5nUcUnU(d[RRU5n[ XX45 [RR X5n[ XbU5 gr8)r<r=r[r\rhr^r)rrrrr2rarchives r+download_and_extract_archivervs`GG&&}5M$ 77##C(X3ggll=3GG?;r:iterablec nSSRUVs/sHn[U5PM sn5-S-$s snf)N'z', ')r^r)ritems r+iterable_to_strrs1 H=HDc$iH=> > DD=s2 Tvaluearg valid_values custom_msgc[U[5(d,UcSnOSnUR[U5US9n[ U5eUcU$X;a,UbUnOSnURX[ U5S9n[ U5eU$)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    !CdC**5P\@]*^Co Lr: file_nameslice_channelsc[US5nUR5R5nUS;a [S5e[R "SUR55nU(d [ S5eSUR55upV[UR5R55nUS:aSnU*nOS n[R"X(S -S 9n S S S 5 WS :XaSOSn W RWWU 5RSSS5n [R"U SS9n U S U2S S 2S S 24n U R[R5$!,(df  N~=f)a0Read 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.c38# UHn[U5v M g7fr8)int).0dims r+ _read_pfm..s7$6SC$6sr<>r4)dtypeNrrDr )axis)r!readlinerstriprrMrN Exceptiongroupsfloatnpfromfilereshape transposeflipastypefloat32) rrr4header dim_matchwhscaleendiandata pfm_channelss r+ _read_pfmrs, i !$$&  '/0 0HH11::<@ 34 47I$4$4$67ajjl))+, 19FFEF{{1SL1# &%1QL <<1l + 5 5aA >D 774a D A% &D ;;rzz ""1  s C E Etc>UR5R[R5R"/URQUR 5P76R S5R"/URSSQSP76RUR5$)Nr) contiguousviewtorchuint8shape element_sizerr)rs r+_flip_byte_orderrs~ 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: