o
)i( @ s d dl Z d dlmZmZmZmZmZmZmZ d dl m
Z
ddl m Z m
Z
mZmZ ddlmZ G dd deZG d d
d
eZG d d d eZdS )
N)AnyCallableDictListOptionalTupleUnion)Image )check_integrity download_and_extract_archive download_urlverify_str_arg)
VisionDatasetc s e Zd ZdZdZddddZddd d
d d d
dZddddZdZ d/de de de de de
e de
e d e d df fd d
Z
d e d ejfd!d"Zd e fd#d$Zd0d%d&Zd'e d(eee f d e fd)d*Zd e fd+d,Zd efd-d.Z ZS )1_LFWzlfw-pyz http://vis-www.cs.umass.edu/lfw/)Zlfwzlfw.tgzZ a17d05bd522c52d84eca14327a23d494)Z lfw_funneledzlfw-funneled.tgzZ 1b42dfed7d15c9b2dd63d5e5840c86ad)zlfw-deepfunneledzlfw-deepfunneled.tgzZ 68331da3eb755a505a502b5aacb3c201)originalfunneledZ deepfunneledZ 9f1ba174e4e1c508ff7cdf10ac338a7dZ 5132f7440eb68cf58910c8a45a2ac10bZ 4f27cbf15b2da4a85c1907eb4181ad21Z 450f0863dd89e85e73936a6d71a3474bZ e4bf5be0a43b5dcd9dc5ccfcb8fb19c5Z 54eaac34beb6d042ed3a7d883e247a21Z a6d0a479bd074669f656265a6e693f6d)z pairs.txtzpairsDevTest.txtzpairsDevTrain.txtz
people.txtzpeopleDevTest.txtzpeopleDevTrain.txt
lfw-names.txt ZDevTrainZDevTest10foldtraintestr NFrootsplit image_setview transformtarget_transformdownloadreturnc s t jtj|| j||d t| d| j | _
| j| j
\}| _ | _ t| dddg| _
t| dg d| _| j
| j| j d| _g | _|rR| | sZtd tj| j|| _d S )
N)r r r r peoplepairsr r z.txtzHDataset 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_filedatar _check_integrity RuntimeErrorr
images_dir) selfr r r r r r r r3 __class__ g/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/datasets/lfw.pyr$ s
z
_LFW.__init__r& c C sB t |d}t |}|dW d S 1 sw Y d S )NrbRGB)openr convert)r4 r& fimgr7 r7 r8 _loader< s
$z _LFW._loaderc C st t tj| j| j| j}t tj| j| j| j| j }|r!|s#dS | j dkr8t tj| j| j
| j| j
S dS )NFr! T) r r% r& r' r r, r- r/ checksumsr names)r4 Zst1Zst2r7 r7 r8 r1 A s
z_LFW._check_integrityc C s| | r
td d S | j | j }t|| j| j| jd t| j | j | j | j dkr`_ Dataset.
Args:
root (string): 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 an 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 r NFr r r r r r r c s6 t |||d||| | | _| \| _| _d S )Nr! )r# r$ _get_classes class_to_idx _get_peopler0 targetsr4 r r r r r r r5 r7 r8 r$ r s
zLFWPeople.__init__c C s g g }}t tj| j| jn}| }| jdkr"t|d dfnd\}}t |D ]I}t|| }dd ||d || d D } ||d 7 }t
| D ]%\}
\} } t dt| d D ]}
| | |
}| | | | j
| q\qMq*W d ||fS 1 sw Y ||fS )Nr r r
)r
r c S g | ] }| d qS stripr .0liner7 r7 r8
z)LFWPeople._get_people..)r; r% r& r' r r/ readlinesr rH range enumeraterI appendrV )r4 r0 rX r= linesn_foldssfoldZn_linesr! irE Znum_imgsnumr> r7 r7 r8 rW s(
" "
zLFWPeople._get_peoplec C sb t tj| j| j}| }dd |D }W d n1 s!w Y dd t|D }|S )Nc S s g | ]
}| d qS )r r] r_ r7 r7 r8 rb s z*LFWPeople._get_classes..c S s i | ]\}}||qS r7 r7 )r` rl namer7 r7 r8
s z*LFWPeople._get_classes..)r; r% r& r' r rA rd rf )r4 r= rh rA rV r7 r7 r8 rU s zLFWPeople._get_classesindexc C sJ | | j| }| j| }| jdur| |}| jdur!| |}||fS )z
Args:
index (int): Index
Returns:
tuple: Tuple (image, target) where target is the identity of the person.
N)r? r0 rX r r )r4 rp r> targetr7 r7 r8 __getitem__ s
zLFWPeople.__getitem__c s t dt| j S )Nz
Classes (identities): )r# rK rL rV rJ r5 r7 r8 rK s zLFWPeople.extra_reprr r NNF)rN rO rP __doc__rQ r r rR r$ r r rH rW r rU r rr rK rS r7 r7 r5 r8 rT ^ s4 rT c s e Zd ZdZ ddededed ee d
ee d ed df fd
d
Zded e e
e eef e
e eef e
e f fddZ de d e e
e
e f fddZ ZS )LFWPairsa `LFW `_ Dataset.
Args:
root (string): 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 an 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 r NFr r r r r r r c s4 t |||d||| | | j\| _| _| _d S )Nr" )r# r$
_get_pairsr3
pair_namesr0 rX rY r5 r7 r8 r$ s zLFWPairs.__init__r3 c C s g g g }}}t tj| j| j}| }| jdkr0|d d\}}t|t|}}n dt|d }}d} t |D ]}
dd || | | D } dd || | | d| D } | d| 7 } | D ]1}
|
|
d |
d |
|
d |
d d}}}| |
d |
d f | ||f | | qg| D ]1}
|
|
d |
d |
|
d |
d d}}}| |
d |
d f | ||f | | qq?W d n1 sw Y |||fS )
Nr r r\ r
c S rZ r[ r] r_ r7 r7 r8 rb rc z'LFWPairs._get_pairs..c S rZ r[ r] r_ r7 r7 r8 rb rc ) r; r% r& r' r r/ rd r rH re rI rg )r4 r3 rw r0 rX r= rh ri Zn_pairsrj rk Z
matched_pairsZunmatched_pairspairimg1img2Zsamer7 r7 r8 rv s4
" 0 0
zLFWPairs._get_pairsrp c C sl | j | \}}| || |}}| j| }| jdur'| || |}}| jdur1| |}|||fS )z
Args:
index (int): Index
Returns:
tuple: (image1, image2, target) where target is `0` for different indentities and `1` for same identities.
N)r0 r? rX r r )r4 rp r{ r| rq r7 r7 r8 rr s
zLFWPairs.__getitem__rs )rN rO rP rt rQ r r rR r$ r r rH rv r rr rS r7 r7 r5 r8 ru s0 8
$ ru )r% typingr r r r r r r ZPILr utilsr r r
r Zvisionr r rT ru r7 r7 r7 r8 s $ TR