kh+ddlZddlmZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZmZmZmZddlmZGdd eZGd d eZGd d eZy)N)Path)AnyCallableDictListOptionalTupleUnion)default_loader)check_integritydownload_and_extract_archive download_urlverify_str_arg) VisionDatasetceZdZdZdZddddZddd d d d d dZddddZdZddde fde e e fde de de de ede ededee gefddffd Zdefd Zd'd!Zd"e d#e ee fde fd$Zde fd%Zdefd&ZxZS)(_LFWzlfw-pyz http://vis-www.cs.umass.edu/lfw/)lfwzlfw.tgz a17d05bd522c52d84eca14327a23d494) lfw_funneledzlfw-funneled.tgz 1b42dfed7d15c9b2dd63d5e5840c86ad)zlfw-deepfunneledzlfw-deepfunneled.tgz 68331da3eb755a505a502b5aacb3c201)originalfunneled deepfunneled 9f1ba174e4e1c508ff7cdf10ac338a7d 5132f7440eb68cf58910c8a45a2ac10b 4f27cbf15b2da4a85c1907eb4181ad21 450f0863dd89e85e73936a6d71a3474b e4bf5be0a43b5dcd9dc5ccfcb8fb19c5 54eaac34beb6d042ed3a7d883e247a21 a6d0a479bd074669f656265a6e693f6d)z pairs.txtzpairsDevTest.txtzpairsDevTrain.txtz people.txtzpeopleDevTest.txtzpeopleDevTrain.txt lfw-names.txtDevTrainDevTest10foldtraintestr#NFrootsplit image_setview transformtarget_transformdownloadloaderreturnc t |tjj ||j ||t |jd|jj|_ |j|j\} |_ |_ t |jdddg|_ t |jdgd|_|j|j|jd|_g|_|r|j%|j's t)d tjj |j*| |_||_y) N)r/r0r-r.peoplepairsr,r'z.txtzHDataset not found or corrupted. You can use download=True to download it)super__init__ospathjoin base_folderrlower file_dictkeysr-filenamemd5r.r, annot_file labels_filedatar1_check_integrity RuntimeErrorr+ images_dir_loader) selfr+r,r-r.r/r0r1r2rG __class__s T/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/datasets/lfw.pyr8z _LFW.__init__ s dD,<,<=euv' (9;H[H[H]^.2nnT^^.L+ DM48"4::<(G9LM #EKKM7sbggll499dmmDdhhObggll499d6F6FGX\XhXhIij# 99 "277<< 4::#FW[WaWaHbc crLc|jry|j|j}t||j|j|j t |j|j|j|jdk(r/t |j|j|jyy)N)r@rAr5) rEdownload_url_prefixr@rr+rArrCr.rO)rIurls rKr1z _LFW.downloadGs  " ))*4==/:$S$))dmmQUQYQYZ001$2B2B1CDdiiP 99 D445djj\BDII N !rLidentitynoc xtjj|j||dt |ddS)N_04dz.jpg)r9r:r;rGint)rIrUrVs rK _get_pathz_LFW._get_pathPs1ww||DOOX(1SWSMQU7VWWrLc:d|jd|jS)Nz Alignment: z Split: )r-r,rIs rK extra_reprz_LFW.extra_reprSsT^^,Idjj\BBrLc,t|jS)N)lenrDr]s rK__len__z _LFW.__len__Vs499~rL)r3N)__name__ __module__ __qualname__r<rSr>rNrBrOr r strrrrboolrr8rEr1rZr[r^ra __classcell__rJs@rKrr s-K<K\hI 8>?8?@;IYGJ E)-/3'5CI    H% #8,#$ <$OX#X5c?XsXCCCrLrceZdZdZdddddefdededed eed eed ed eege fd dffd Z d e e ee e ffdZd eee ffdZde d e e e ffdZd effd ZxZS) LFWPeoplea`LFW `_ Dataset. Args: root (str or ``pathlib.Path``): Root directory of dataset where directory ``lfw-py`` exists or will be saved to if download is set to True. split (string, optional): The image split to use. Can be one of ``train``, ``test``, ``10fold`` (default). image_set (str, optional): Type of image funneling to use, ``original``, ``funneled`` or ``deepfunneled``. Defaults to ``funneled``. transform (callable, optional): A function/transform that takes in a PIL image or torch.Tensor, depends on the given loader, and returns a transformed version. E.g, ``transforms.RandomCrop`` target_transform (callable, optional): A function/transform that takes in the target and transforms it. download (bool, optional): If true, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again. loader (callable, optional): A function to load an image given its path. By default, it uses PIL as its image loader, but users could also pass in ``torchvision.io.decode_image`` for decoding image data into tensors directly. r(rNFr+r,r-r/r0r1r2r3c t||||d|||||j|_|j \|_|_y)Nr5r2)r7r8 _get_classes class_to_idx _get_peoplerDtargets rIr+r,r-r/r0r1r2rJs rKr8zLFWPeople.__init__psN ui9FVX`iop --/"&"2"2"4 4?TZZ\'')!,?E? @09/?@GAta@ @@ @ @As#B*(B%;B* B6%B**B3indexc|j|j|}|j|}|j|j|}|j|j |}||fS)z Args: index (int): Index Returns: tuple: Tuple (image, target) where target is the identity of the person. )rHrDrpr/r0)rIrrtargets rK __getitem__zLFWPeople.__getitem__sgll499U+,e$ >> %..%C  ,**62FF{rLcRt|dt|jzS)Nz Classes (identities): )r7r^r`rn)rIrJs rKr^zLFWPeople.extra_reprs*w!#(@TEVEVAW@X&YYYrL)rbrcrd__doc__r rerrrfrr8r rrZrorrmrr^rgrhs@rKrjrjZs0#(,/3'5 5 5 5 5 H% 5 #8, 5 5#$ 5  5U49d3i#78$d38nsCx&ZCZZrLrjceZdZdZdddddefdededed eed eed ed eege fd dffd Z ded e e e eefe e eefe e ffdZde d e e e e ffdZxZS)LFWPairsa`LFW `_ Dataset. Args: root (str or ``pathlib.Path``): Root directory of dataset where directory ``lfw-py`` exists or will be saved to if download is set to True. split (string, optional): The image split to use. Can be one of ``train``, ``test``, ``10fold``. Defaults to ``10fold``. image_set (str, optional): Type of image funneling to use, ``original``, ``funneled`` or ``deepfunneled``. Defaults to ``funneled``. transform (callable, optional): A function/transform that takes in a PIL image and returns a transformed version. E.g, ``transforms.RandomRotation`` target_transform (callable, optional): A function/transform that takes in the target and transforms it. download (bool, optional): If true, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again. loader (callable, optional): A function to load an image given its path. By default, it uses PIL as its image loader, but users could also pass in ``torchvision.io.decode_image`` for decoding image data into tensors directly. r(rNFr+r,r-r/r0r1r2r3c t||||d|||||j|j\|_|_|_y)Nr6rl)r7r8 _get_pairsrG pair_namesrDrprqs rKr8zLFWPairs.__init__sE ui)EUW_hno37??4??3S0DLrLrGc pggg}}}ttjj|j|j 5}|j }|jdk(r.|djd\}}t|t|}}ndt|d}}d} t|D]a} || | |zD cgc]!} | jjd#} } || |z| d|zzD cgc]!} | jjd#} } | d|zz } | D]q}|j|d|d|j|d|dd}}}|j|d|df|j||f|j|s| D]q}|j|d|d|j|d|dd}}}|j|d|df|j||f|j|sd ddd|||fScc} wcc} w#1swYxYw)Nr(rrsr ) rtr9r:r;r+rCrur,rZrvrwr[ry)rIrGrrDrprzr{r|n_pairsr}r~r matched_pairsunmatched_pairspairimg1img2sames rKrzLFWPairs._get_pairssF$&B'D "'',,tyy$*:*:; < )KKMEzzX%#(8>>$#7 #&w<W#$c%(mAg )FKAPQT[P[F\ ]d!3!3D!9 ] ]HMaRYk\]abelal\mHn"o4::<#5#5d#;"o"oQ[ ))D'+~~d1gtAw'GX\]^X_aefgahIikl$D%%tAwQ&89KKt -NN4( ) ,)D'+~~d1gtAw'GX\]^X_aefgahIikl$D%%tAwQ&89KKt -NN4( ) ) )04((!^"o ) )s,A9H,=&H"#H,5&H'C9H," H,,H5rc(|j|\}}|j||j|}}|j|}|j"|j||j|}}|j|j |}|||fS)z Args: index (int): Index Returns: tuple: (image1, image2, target) where target is `0` for different indentities and `1` for same identities. )rDrHrpr/r0)rIrrrrs rKrzLFWPairs.__getitem__sYYu% d\\$'d);de$ >> %-t~~d/C$D  ,**62FT6!!rL)rbrcrdrr rerrrfrr8r rrZrrrgrhs@rKrrs2#(,/3'5 T T T T H% T #8, T T#$ T  T)S)U4c3h3H$uUXZ]U]J_aefiaj3j-k)8""sC})="rLr)r9pathlibrtypingrrrrrr r folderr utilsr rrrvisionrrrjrrLrKrsN DDD"^^!M=M`RZRZjS"tS"rL