U
h@) @ s d dl Z d dlmZ d dlmZmZmZmZmZm Z m
Z
d dl m Z ddl
mZmZmZmZ ddlmZ G dd d eZG d
d d eZG d d
d
eZdS ) N)Path)AnyCallableDictListOptionalTupleUnion)Image )check_integrity download_and_extract_archive download_urlverify_str_arg)
VisionDatasetc
s e Zd ZdZdZddddZddd d
d d d
dZddddZdZd(e e
e f e
e
e
e e
e e
edd fdd
Ze
ejdddZed d d Zdd d d Ze
e ee
f e
d!d"d#Ze
d d$d%Zed d&d'Z ZS ))_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 ZDevTrainZDevTest)10foldtraintestr NF)rootsplit image_setview transformtarget_transformdownloadreturnc s t jtj|| j||d t| d| j | _
| j| j
\}| _ | _ t| dddg| _
t| dddd g| _| j
| j| j d
| _g | _|r| | std tj| j|| _d S ) N)r r r r peoplepairsr r r 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__ J/var/www/html/venv/lib/python3.8/site-packages/torchvision/datasets/lfw.pyr$ ! s
z
_LFW.__init__)r& r c
C s6 t |d"}t |}|dW 5 Q R S Q R X d S )NrbRGB)openr
convert)r4 r& fimgr7 r7 r8 _loader= s
z _LFW._loaderr c 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)r4 Zst1Zst2r7 r7 r8 r1 B 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! ) r1 printdownload_url_prefixr, r
r r- r r/ r rB )r4 urlr7 r7 r8 r K s
z
_LFW.download)identitynor c C s$ t j| j|| dt|ddS )N_Z04dz.jpg)r% r& r' r3 int)r4 rF rG r7 r7 r8 _get_pathU s z_LFW._get_pathc C s d| j d| j S )Nz Alignment: z
Split: )r r r4 r7 r7 r8
extra_reprX s z_LFW.extra_reprc C s
t | jS )N)lenr0 rK r7 r7 r8 __len__[ s z _LFW.__len__)NNF)__name__
__module__ __qualname__r( rD r* rA r. rB r strr r r boolr$ r
r? r1 r rI rJ rL rN
__classcell__r7 r7 r5 r8 r sF
r c s e Zd ZdZdeeeee ee edd fdd
Ze e
e e
e f d d
d Z e
ee f d d d
Ze e eef dddZed fdd Z ZS ) 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 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$ s s
zLFWPeople.__init__r@ 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 5 Q R X ||fS )Nr r r )r r c S s g | ]}| d qS stripr .0liner7 r7 r8
s z)LFWPeople._get_people..)r; r% r& r' r r/ readlinesr rI range enumeraterJ appendrX )r4 r0 rZ r= linesn_foldssfoldZn_linesr! irF Znum_imgsnumr> r7 r7 r8 rY s
" "
zLFWPeople._get_peoplec C sN t tj| j| j }| }dd |D }W 5 Q R X dd t|D }|S )Nc S s g | ]}| d qS )r r^ r` r7 r7 r8 rc s z*LFWPeople._get_classes..c S s i | ]\}}||qS r7 r7 )ra rl namer7 r7 r8
s z*LFWPeople._get_classes..)r; r% r& r' r rB rd rf )r4 r= rh rB rX r7 r7 r8 rW s
zLFWPeople._get_classesindexr c C sJ | | j| }| j| }| jdk r.| |}| jdk rB| |}||fS )z
Args:
index (int): Index
Returns:
tuple: Tuple (image, target) where target is the identity of the person.
N)r? r0 rZ r r )r4 rq r> targetr7 r7 r8 __getitem__ s
zLFWPeople.__getitem__c s t dt| j S )Nz
Classes (identities): )r# rL rM rX rK r5 r7 r8 rL s zLFWPeople.extra_repr)r r NNF)rO rP rQ __doc__rR r r rS r$ r r rI rY r rW r rs rL rT r7 r7 r5 r8 rU _ s& rU c s e Zd ZdZdeeeee ee edd fdd
Zee e
e eef e
e eef e
e f d d
d Z e e e
e
e f d d
dZ ZS )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 r NFrV c s4 t |||d||| | | j\| _| _| _d S )Nr" )r# r$
_get_pairsr3
pair_namesr0 rZ r[ r5 r7 r8 r$ s zLFWPairs.__init__)r3 r 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 5 Q R X |||fS )
Nr r r] r c S s g | ]}| d qS r\ r^ r` r7 r7 r8 rc s z'LFWPairs._get_pairs..c S s g | ]}| d qS r\ r^ r` r7 r7 r8 rc s ) r; r% r& r' r r/ rd r rI re rJ rg )r4 r3 rw r0 rZ r= rh ri Zn_pairsrj rk Z
matched_pairsZunmatched_pairspairimg1img2Zsamer7 r7 r8 rv s.
" 0 0zLFWPairs._get_pairsrp c C sl | j | \}}| || | }}| j| }| jdk rN| || | }}| jdk rb| |}|||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? rZ r r )r4 rq r{ r| rr r7 r7 r8 rs s
zLFWPairs.__getitem__)r r NNF)rO rP rQ rt rR r r rS r$ r r rI rv r rs rT r7 r7 r5 r8 ru s"
6 ru )r% pathlibr typingr r r r r r r ZPILr
utilsr r
r r Zvisionr r rU ru r7 r7 r7 r8 s $ TR