j*ddlZddlmZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZmZmZmZddlmZGdd eZGd d eZGd d eZdS)N)Path)AnyCallableDictListOptionalTupleUnion)Image)check_integritydownload_and_extract_archive download_urlverify_str_arg) VisionDatasetceZdZdZdZddddZddd d d d d dZddddZdZ d(de e e fde de de de e de e deddffd Zde dejfd 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_transformdownloadreturncttj||j||t |d|j |_ |j|j \}|_ |_ t |dddg|_ t |dgd|_|j |j|jd|_g|_|r||st)d tj|j||_dS) 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) selfr+r,r-r.r/r0r1rF __class__s IC:\Projects\Historical\venv\Lib\site-packages\torchvision/datasets/lfw.pyr7z _LFW.__init__!s< dD,<==euvvv' (9(9;H[H[H]H]^^.2nT^.L+ DM48"4::<<(G9LMM #EKKMM7?8?@;IYGGJ E)-/3>>CI>> >  > H% >#8,>> >>>>>>8&C&EK&&&& $OOOOX#X5c?XsXXXXCCCCCCrJrceZdZdZ ddededed eed eed ed dffd Zd e e ee e ffdZ d e ee ffdZde d e eeffdZd 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 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. r(rNFr+r,r-r/r0r1r2c t|||d|||||_|\|_|_dS)Nr4)r6r7 _get_classes class_to_idx _get_peoplerCtargetsrGr+r,r-r/r0r1rHs rIr7zLFWPeople.__init__rs] ui9FVX`aaa --//"&"2"2"4"4 4<<z)LFWPeople._get_people..s.^^^t$**,,,,T22^^^rJ)rNr8r9r:r+rB readlinesr,r`range enumerateraappendrs)rGrCrurPlinesn_foldssfoldn_linesr4ir[num_imgsnumrQs rIrtzLFWPeople._get_peoplesBg "',,ty$*:;; < < DKKMME/3zX/E/E#eAh--++6JGQg D DeAh--^^uQUQQX[[\_E\?]^^^Wq[ /8/@/@DD+A+($QH (9::DD"nnXs;; C(((t'8'BCCCCDD  D D D D D D D D D D D D D D D DW}sDEEEcttj|j|j5}|}d|D}dddn #1swxYwYdt|D}|S)Ncfg|].}|d/S)rr{r}s rIrz*LFWPeople._get_classes..s1???TZZ\\''))!,???rJci|]\}}|| Sr)r~rnames rI z*LFWPeople._get_classes..s@@@GAta@@@rJ)rNr8r9r:r+rUrr)rGrPrrUrss rIrrzLFWPeople._get_classess "',,ty$*55 6 6 @!KKMME?????E @ @ @ @ @ @ @ @ @ @ @ @ @ @ @A@y/?/?@@@ s!A%%A),A)indexc||j|}|j|}|j||}|j||}||fS)z Args: index (int): Index Returns: tuple: Tuple (image, target) where target is the identity of the person. )rRrCrur/r0)rGrrQtargets rI __getitem__zLFWPeople.__getitem__sgll49U+,,e$ > %..%%C  ,**622FF{rJcttdt|jzS)Nz Classes (identities): )r6rdrfrs)rGrHs rIrdzLFWPeople.extra_reprs2ww!!##&YTEVAWAW&Y&YYYrJr(rNNF)rhrirj__doc__rkrrrlr7r rr`rtrrrrrrdrmrns@rIrprp^sU,#(,/3 5 5 5 5 5 H% 5 #8, 5 5  5 5 5 5 5 5U49d3i#78$d38nsCx&ZCZZZZZZZZZZrJrpceZdZdZ ddededed eed eed ed dffd Zded e e e eefe e eefe e ffdZ de 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. r(rNFr+r,r-r/r0r1r2c t|||d|||||j\|_|_|_dS)Nr5)r6r7 _get_pairsrF pair_namesrCrurvs rIr7zLFWPairs.__init__sO ui)EUW_```37??4?3S3S0DLLLrJrFc ggg}}}ttj|j|j5}|}|jdkr=|dd\}}t|t|}}ndt|d}}d} t|D]s} d|| | |zD} d|| |z| d|zzD} | d|zz } | D]} | | d| d| | d| dd}}}| | d| df| ||f| || D]} | | d| d| | d| dd}}}| | d| df| ||f| |u dddn #1swxYwY|||fS) Nr(rrzr c\g|])}|d*Sryr{r}s rIrz'LFWPairs._get_pairs..s. ] ] ]d!3!3D!9!9 ] ] ]rJc\g|])}|d*Sryr{r}s rIrz'LFWPairs._get_pairs..s."o"o"o4::<<#5#5d#;#;"o"o"orJ) rNr8r9r:r+rBrr,r`rrar)rGrFrrCrurPrrn_pairsrr matched_pairsunmatched_pairspairimg1img2sames rIrzLFWPairs._get_pairss$&B'D "',,ty$*:;; < < )KKMMEzX%%#(8>>$#7#7 #&w<<W#$c%(mmAg ) ) ] ]eAPQT[P[OF\ ] ] ] "o"oaRYk\]abelal\mNmHn"o"o"oQ[ )))D'+~~d1gtAw'G'GX\]^X_aefgahIiIikl$D%%tAwQ&8999KKt ---NN4((((+))D'+~~d1gtAw'G'GX\]^X_aefgahIiIikl$D%%tAwQ&8999KKt ---NN4(((( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )04((sG:II  I rc4|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. )rCrRrur/r0)rGrrrrs rIrzLFWPairs.__getitem__sYu% d\\$''d););de$ > %--t~~d/C/C$D  ,**622FT6!!rJr)rhrirjrrkrrrlr7r rr`rrrrmrns@rIrrs8,#(,/3 T T T T T H% T #8, T T  T T T T T T)S)U4c3h3H$uUXZ]U]J_aefiaj3j-k))))8""sC})=""""""""rJr)r8pathlibrtypingrrrrrr r PILr utilsr rrrvisionrrrprrrJrIrsJ DDDDDDDDDDDDDDDDDD^^^^^^^^^^^^!!!!!!PPPPP=PPPfOZOZOZOZOZOZOZOZdO"O"O"O"O"tO"O"O"O"O"rJ