Di,ddlZddlmZddlmZmZmZmZddlm Z ddl m Z m Z m Z mZddlmZGdd eZGd d eZGd d eZy)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 t |tjj ||j ||t |jd|jj|_ |j|j\} |_ |_ t |jdddg|_ t |jdgd|_|j|j|jd|_g|_|r t%d |j)s t+d tjj |j,| |_||_y) 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 Z/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/torchvision/datasets/lfw.pyr5z _LFW.__init__ s dD,<,<=euv' (9;H[H[H]^.2nnT^^.L+ DM48"4::<(G9LM #EKKM7r@ checksumsr+names)rGst1st2s rIrCz_LFW._check_integrityBsbggll499dmmDdhhObggll499d6F6FGX\XhXhIij# 99 "277<< 4::#FW[WaWaHbc crJc|jry|j|j}t||j|j|j t |j|j|j|jdk(r/t |j|j|jyy)N)r=r>r2) rCdownload_url_prefixr=r r(r>r r@r+rM)rGurls rIr.z _LFW.downloadKs  " ))*4==/:$S$))dmmQUQYQYZ001$2B2B1CDdiiP 99 D445djj\BDII N !rJidentitynoc xtjj|j||dt |ddS)N_04dz.jpg)r6r7r8rEint)rGrSrTs rI _get_pathz_LFW._get_pathTs1ww||DOOX(1SWSMQU7VWWrJc:d|jd|jS)Nz Alignment: z Split: )r*r)rGs rI extra_reprz_LFW.extra_reprWsT^^,Idjj\BBrJc,t|jS)N)lenrAr[s rI__len__z _LFW.__len__Zs499~rJ)r0N)__name__ __module__ __qualname__r9rQr;rLr?rMr rstrrrrboolrr5rCr.rXrYr\r_ __classcell__rHs@rIrr s.K<K\hI 8>?8?@;IYGJ E)-/3'5 CI      H% #8,  #$   D$OX#X5c?XsXCCCrJrceZdZdZdddddefdededed 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|||||j|_|j \|_|_y)Nr2r/)r4r5 _get_classes class_to_idx _get_peoplerAtargets rGr(r)r*r,r-r.r/rHs rIr5zLFWPeople.__init__wsN 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. )rFrArnr,r-)rGrrtargets rI __getitem__zLFWPeople.__getitem__sgll499U+,e$ >> %..%C  ,**62FF{rJcRt|dt|jzS)Nz Classes (identities): )r4r\r^rl)rGrHs rIr\zLFWPeople.extra_reprs*w!#(@TEVEVAW@X&YYYrJ)r`rarb__doc__r rcrrrdrr5tuplelistrXrmdictrkrr\rerfs@rIrhrh^s6#(,/3'5 5 5 5 5 H% 5 #8, 5 5#$ 5  5U49d3i#78$d38nsCx&ZCZZrJrhceZdZdZdddddefdededed 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|j\|_|_|_y)Nr3rj)r4r5 _get_pairsrE pair_namesrArnros rIr5zLFWPairs.__init__sE ui)EUW_hno37??4??3S0DLrJrEc 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%rrqr) rrr6r7r8r(r@rsr)rXrtrurYrw)rGrErrArnrxryrzn_pairsr{r|r~ matched_pairsunmatched_pairspairimg1img2sames rIrzLFWPairs._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. )rArFrnr,r-)rGrrrrs rIrzLFWPairs.__getitem__sYYu% d\\$'d);de$ >> %-t~~d/C$D  ,**62FT6!!rJ)r`rarbrr rcrrrdrr5rrrXrrrerfs@rIrrs8#(,/3'5 T T T T H% T #8, T T#$ T  T)S)U4c3h3H$uUXZ]U]J_aefiaj3j-k)8""sC})="rJr)r6pathlibrtypingrrrrfolderr utilsr r r r visionrrrhrrJrIrsK 11"^^!Q=QhUZUZpV"tV"rJ