Yh,ddlZddlmZddlmZmZmZmZddlm Z ddl m Z m Z m Z mZddlmZGdd eZGd d eZGd d eZdS)N)Path)AnyCallableOptionalUnion)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 ttj||j||t |d|j |_ |j|j \} |_ |_ t |dddg|_ t |dgd|_|j |j|jd|_g|_|rt%d |st+d tj|j| |_||_dS) N)r,r-r*r+peoplepairsr)r$z.txtz|LFW dataset is no longer available for download.Please download the dataset manually and place it in the specified directoryzHDataset not found or corrupted. You can use download=True to download it)super__init__ospathjoin base_folderr lower file_dictkeysr*filenamemd5r+r) annot_file labels_filedata ValueErrorr._check_integrity RuntimeErrorr( images_dir_loader) selfr(r)r*r+r,r-r.r/rE __class__s j/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/lfw.pyr5z _LFW.__init__ sP dD,<==euvvv' (9(9;H[H[H]H]^^.2nT^.L+ DM48"4::<<(G9LMM #EKKMM7r@ checksumsr+names)rGst1st2s rIrCz_LFW._check_integrityBsbgll49dmDDdhOObgll49d6FGGX\XhIijj # 5 9 "27<< 4:#F#FW[WaHbcc ctrJc<|rdS|j|j}t||j|j|jt |j|j|j|jdkr%t |j|j |jdSdS)N)r=r>r2) rCdownload_url_prefixr=r r(r>r r@r+rM)rGurls rIr.z _LFW.downloadKs  " "  F):4=::$S$)dmQUQYZZZZ0D$2BDDdiPPP 9 D4BdjBBDI N N N N N ! rJidentitynoc vtj|j||dt |ddS)N_04dz.jpg)r6r7r8rEint)rGrSrTs rI _get_pathz_LFW._get_pathTs6w||DOX(7V7VSWW7V7V7V7VWWWrJc&d|jd|jS)Nz Alignment: z Split: )r*r)rGs rI extra_reprz_LFW.extra_reprWsBT^BBdjBBBrJc*t|jS)N)lenrAr[s rI__len__z _LFW.__len__Zs49~~rJ)r0N)__name__ __module__ __qualname__r9rQr;rLr?rMr rstrrrrboolrr5rCr.rXrYr\r_ __classcell__rHs@rIrr sK<K\hI 8>?8?@;IYGGJ E)-/3'5  CI      H% #8,  #$        D$OOOOX#X5c?XsXXXXCCCCCCrJrceZdZdZdddddefdededed 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. .. warning: The LFW dataset is no longer available for automatic download. Please download it manually and place it in the specified directory. 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): NOT SUPPORTED ANYMORE, leave to False. 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,r-r.r/r0c t|||d||||||_|\|_|_dS)Nr2r/)r4r5 _get_classes class_to_idx _get_peoplerAtargets rGr(r)r*r,r-r.r/rHs rIr5zLFWPeople.__init__wsc ui9FVX`ioppp --//"&"2"2"4"4 4<<z)LFWPeople._get_people..s.^^^t$**,,,,T22^^^rJ)openr6r7r8r(r@ readlinesr)rXrange enumeraterYappendrl)rGrArnflinesn_foldssfoldn_linesr2irSnum_imgsnumimgs rIrmzLFWPeople._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)rrtrvs rIryz*LFWPeople._get_classes..s1???TZZ\\''))!,???rJci|]\}}|| Sr)rwrnames rI z*LFWPeople._get_classes..s@@@GAta@@@rJ)rzr6r7r8r(rMr{r})rGrrrMrls rIrkzLFWPeople._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. )rFrArnr,r-)rGrrtargets rI __getitem__zLFWPeople.__getitem__sgll49U+,,e$ > %..%%C  ,**622FF{rJcttdt|jzS)Nz Classes (identities): )r4r\r^rl)rGrHs rIr\zLFWPeople.extra_reprs2ww!!##&YTEVAWAW&Y&YYYrJ)r`rarb__doc__r rcrrrdrr5tuplelistrXrmdictrkrr\rerfs@rIrhrh^sm6#(,/3'5 5 5 5 5 5 H% 5 #8, 5 5#$ 5  5 5 5 5 5 5U49d3i#78$d38nsCx&ZCZZZZZZZZZZrJrhceZdZdZdddddefdededed 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. .. warning: The LFW dataset is no longer available for automatic download. Please download it manually and place it in the specified directory. 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): NOT SUPPORTED ANYMORE, leave to False. 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,r-r.r/r0c t|||d||||||j\|_|_|_dS)Nr3rj)r4r5 _get_pairsrE pair_namesrArnros rIr5zLFWPairs.__init__sU ui)EUW_hnooo37??4?3S3S0DLLLrJrEc 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%rrsrc\g|])}|d*Srrrtrvs rIryz'LFWPairs._get_pairs..s. ] ] ]d!3!3D!9!9 ] ] ]rJc\g|])}|d*Srrrtrvs rIryz'LFWPairs._get_pairs..s."o"o"o4::<<#5#5d#;#;"o"o"orJ) rzr6r7r8r(r@r{r)rXr|rYr~)rGrErrArnrrrn_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. )rArFrnr,r-)rGrrrrs rIrzLFWPairs.__getitem__sYu% d\\$''d););de$ > %--t~~d/C/C$D  ,**622FT6!!rJ)r`rarbrr rcrrrdrr5rrrXrrrerfs@rIrrsQ8#(,/3'5 T T T T T H% T #8, T T#$ T  T T T T T T)S)U4c3h3H$uUXZ]U]J_aefiaj3j-k))))8""sC})=""""""""rJr)r6pathlibrtypingrrrrfolderr utilsr r r r visionrrrhrrrJrIrs8 111111111111""""""^^^^^^^^^^^^!!!!!!QQQQQ=QQQhUZUZUZUZUZUZUZUZpV"V"V"V"V"tV"V"V"V"V"rJ