a
h, @ s d dl Z d dlmZ d dlmZmZmZmZ ddlm Z ddl
m Z m Z m
Z
mZ ddlmZ G dd d eZG d
d d eZG d d
d
eZdS ) N)Path)AnyCallableOptionalUnion )default_loader)check_integrity download_and_extract_archive download_urlverify_str_arg)
VisionDatasetc s e Zd ZdZdZddddZddd d
d d d
dZddddZdZddde fe
e e f e e e e
e e
e eee gef dd fdd
ZedddZddd d Ze e
ee f e d d d Ze dd!d"Zedd#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 ZDevTrainZDevTest10foldZtraintestr NF) rootsplit image_setview transformtarget_transformdownloadloaderreturnc
s t jtj|| j||d t| d| j | _
| j| j
\} | _ | _ t| dddg| _
t| dg d| _| j
| j| j d| _g | _|rtd | | std
tj| j| | _|| _d S ) 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 r2 __class__ F/var/www/auris/lib/python3.9/site-packages/torchvision/datasets/lfw.pyr" s z
_LFW.__init__r 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 r0 B s
z_LFW._check_integrityc C sp | r d S | j | j }t|| j| j| jd t| j | j | j | jdkrlt| j | j | j d S )N)r* r+ r )
r0 download_url_prefixr* r
r r+ r r- r r; )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% r2 int)r4 r> r? r7 r7 r8 _get_pathT s z_LFW._get_pathc C s d| j d| j S )Nz Alignment: z
Split: )r r r4 r7 r7 r8
extra_reprW s z_LFW.extra_reprc C s
t | jS )N)lenr. rC r7 r7 r8 __len__Z s z _LFW.__len__)__name__
__module__ __qualname__r& r< r( r: r, r; r r strr r r boolr r" r0 r rA rB rD rF
__classcell__r7 r7 r5 r8 r
sH
" r c
s e Zd ZdZdddddefeeeee ee eeege f dd fdd
Z
e e e e e
f d d
d Zeee
f d d d
Ze
e e e f dddZed fdd Z ZS ) 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 r NFr r r r r r r r c
s: t j|||d||||d | | _| \| _| _d S )Nr r )r! r" _get_classes class_to_idx _get_peopler. targetsr4 r r r r r r r r5 r7 r8 r" w s
zLFWPeople.__init__r9 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 .0liner7 r7 r8
z)LFWPeople._get_people..)openr# r$ r% r r- readlinesr rA range enumeraterB appendrQ )r4 r. rS flinesn_foldssfoldZn_linesr ir> Znum_imgsnumimgr7 r7 r8 rR 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 rW rY r7 r7 r8 r\ r] z*LFWPeople._get_classes..c S s i | ]\}}||qS r7 r7 )rZ rh namer7 r7 r8
r] z*LFWPeople._get_classes..)r^ r# r$ r% r r; r_ ra )r4 rc rd r; rQ r7 r7 r8 rP s
,zLFWPeople._get_classesindexr 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)r3 r. rS r r )r4 rn rj targetr7 r7 r8 __getitem__ s
zLFWPeople.__getitem__c s t dt| j S )Nz
Classes (identities): )r! rD rE rQ rC r5 r7 r8 rD s zLFWPeople.extra_repr)rG rH rI __doc__r rJ r r rK r r" tuplelistrA rR dictrP rp rD rL r7 r7 r5 r8 rM ^ s* rM c
s e Zd ZdZdddddefeeeee ee eeege f dd fdd
Z
ee e e eef e e eef e e
f d d
d Ze
e e e e
f d d
dZ ZS )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 r NFrN c
s8 t j|||d||||d | | j\| _| _| _d S )Nr rO )r! r"
_get_pairsr2
pair_namesr. rS rT r5 r7 r8 r" s
zLFWPairs.__init__)r2 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 d n1 s0 Y |||fS )
Nr r rV r c S s g | ]}| d qS rU rW rY r7 r7 r8 r\ r] z'LFWPairs._get_pairs..c S s g | ]}| d qS rU rW rY r7 r7 r8 r\ r] ) r^ r# r$ r% r r- r_ r rA r` rB rb )r4 r2 rw r. rS rc rd re Zn_pairsrf rg Z
matched_pairsZunmatched_pairspairimg1img2Zsamer7 r7 r8 rv s.
" 0 00zLFWPairs._get_pairsrm 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. r3 rS r r )r4 rn r{ r| ro r7 r7 r8 rp s
zLFWPairs.__getitem__)rG rH rI rq r rJ r r rK r r" rr rs rA rv rp rL r7 r7 r5 r8 ru s& 6 ru )r# pathlibr typingr r r r folderr utilsr r
r r Zvisionr
r rM ru r7 r7 r7 r8 s TX