a
PSic*( @ s d dl Z d dlmZmZmZmZmZ d dlmZ ddl m
Z
m Z m Z m
Z
ddlmZ G dd deZG d d
d
eZG d d d eZdS )
N)AnyCallableListOptionalTuple)Image )check_integrity download_and_extract_archive download_urlverify_str_arg)
VisionDatasetc s e Zd ZdZdZddddZddd d
d d d
dZddddZdZd'e e e e e
e e
e e d fdd
Z
e ejdddZd d Zd d Zd d! Ze d"d#d$Zd%d& Z ZS )(_LFWzlfw-pyz http://vis-www.cs.umass.edu/lfw/)lfwzlfw.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 NF)rootsplit image_setview transformtarget_transformdownloadc s t jtj|| j||d t| d| j | _
| j| j
\}| _ | _ t| dddg| _
t| dg d| _| j
| j| j d| _g | _|r| | std 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 r1 __class__ T/var/www/html/django/DPS/env/lib/python3.9/site-packages/torchvision/datasets/lfw.pyr" s
z
_LFW.__init__)r$ returnc C sB t |d$}t |}|dW d S 1 s40 Y d S )NrbRGB)openr convert)r2 r$ fimgr5 r5 r6 _loader< s
z _LFW._loaderc C st t tj| j| j| j}t tj| j| j| j| j }|rB|sFdS | j dkrpt tj| j| j
| j| j
S dS )NFr T) r r# r$ r% r r* r+ r- checksumsr names)r2 st1st2r5 r5 r6 r/ A s
z_LFW._check_integrityc C sx | rtd d S | j | j }t|| j| j| jd t| j | j | j | j dkrtt| j | j
| j d S )Nz%Files already downloaded and verified)r* r+ r ) r/ printdownload_url_prefixr* r
r r+ r r- r r@ )r2 urlr5 r5 r6 r J s
z
_LFW.downloadc C s$ t j| j|| dt|ddS )N_04dz.jpg)r# r$ r% r1 int)r2 identitynor5 r5 r6 _get_pathT s z_LFW._get_pathr7 c C s d| j d| j S )Nz Alignment: z
Split: )r r r2 r5 r5 r6
extra_reprW s z_LFW.extra_reprc C s
t | jS )N)lenr. rM r5 r5 r6 __len__Z s z _LFW.__len__)NNF)__name__
__module__ __qualname__r& rD r( r? r, r@ strr r boolr" r r> r/ r rK rN rP
__classcell__r5 r5 r3 r6 r
sD
r c st e Zd ZdZdeeeee ee ed fdd
Zd d
Z d d Z
e e e
e
f d
ddZed fdd Z ZS ) LFWPeoplea `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`` (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 c s6 t |||d||| | | _| \| _| _d S )Nr )r! r" _get_classes class_to_idx _get_peopler. targetsr2 r r r r r r r3 r5 r6 r" r s
zLFWPeople.__init__c C s g g }}t tj| j| j}| }| jdkrDt|d dfnd\}}t |D ]}t|| }dd ||d || d D } ||d 7 }t
| D ]J\}
\} } t dt| d D ]*}
| | |
}| | | | j
| qqqTW d n1 s0 Y ||fS )Nr r r )r r c S s g | ]}| d qS stripr .0liner5 r5 r6
z)LFWPeople._get_people..)r: r# r$ r% r r- readlinesr rH range enumeraterK appendrZ )r2 r. r\ r< linesn_foldssfoldZn_linesr irI Znum_imgsnumr= r5 r5 r6 r[ s
" "
4zLFWPeople._get_peoplec C sb t tj| j| j&}| }dd |D }W d n1 sB0 Y dd t|D }|S )Nc S s g | ]}| d qS )r r` rb r5 r5 r6 re rf z*LFWPeople._get_classes..c S s i | ]\}}||qS r5 r5 )rc ro namer5 r5 r6
rf z*LFWPeople._get_classes..)r: r# r$ r% r r@ rg ri )r2 r< rk r@ rZ r5 r5 r6 rY s
,zLFWPeople._get_classesindexr7 c C sJ | | j| }| j| }| jdur.| |}| jdurB| |}||fS )z
Args:
index (int): Index
Returns:
tuple: Tuple (image, target) where target is the identity of the person.
N)r> r. r\ r r )r2 rt r= targetr5 r5 r6 __getitem__ s
zLFWPeople.__getitem__rL c s t dt| j S )Nz
Classes (identities): )r! rN rO rZ rM r3 r5 r6 rN s zLFWPeople.extra_repr)r r NNF)rQ rR rS __doc__rT r r rU r" r[ rY rH r r rv rN rV r5 r5 r3 r6 rW ^ s$ rW c s\ e Zd ZdZdeeeee ee ed fdd
Zd d
Z e
e e e e
f d d d
Z
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 NFrX c s4 t |||d||| | | j\| _| _| _d S )Nr )r! r"
_get_pairsr1
pair_namesr. r\ r] r3 r5 r6 r" s zLFWPairs.__init__c C s g g g }}}t tj| j| j}| }| jdkrb|d d\}}t|t| }}ndt|d }}d} t |D ] }
dd || | | D } dd || | | d| D } | d| 7 } | D ]b}
|
|
d |
d |
|
d |
d d }}}| |
d |
d f | ||f | | q| D ]d}
|
|
d |
d |
|
d |
d d }}}| |
d |
d f | ||f | | q:qW d n1 s0 Y |||fS )
Nr r r_ r c S s g | ]}| d qS r^ r` rb r5 r5 r6 re rf z'LFWPairs._get_pairs..c S s g | ]}| d qS r^ r` rb r5 r5 r6 re rf ) r: r# r$ r% r r- rg r rH rh rK rj )r2 r1 rz r. r\ r< rk rl Zn_pairsrm rn Z
matched_pairsZunmatched_pairspairimg1img2samer5 r5 r6 ry s.
" 0 00zLFWPairs._get_pairsrs c C sl | j | \}}| || | }}| j| }| jdurN| || | }}| jdurb| |}|||fS )z
Args:
index (int): Index
Returns:
tuple: (image1, image2, target) where target is `0` for different indentities and `1` for same identities.
N)r. r> r\ r r )r2 rt r~ r ru r5 r5 r6 rv s
zLFWPairs.__getitem__)r r NNF)rQ rR rS rw rT r r rU r" ry rH r r rv rV r5 r5 r3 r6 rx s
rx )r# typingr r r r r PILr utilsr r
r r visionr
r rW rx r5 r5 r5 r6 s TR