o
h @ st d dl Zd dlmZ d dlmZmZmZmZ d dl Z
d dl m Z ddl
mZmZmZ ddlmZ G dd d eZdS )
N)Path)AnyCallableOptionalUnion)Image )check_integrity download_urlverify_str_arg)
VisionDatasetc
s e Zd ZdZg dg dg ddZ d d eeef d
ed ee d ee d
e
ddf fdd
Z de de
eef fddZde fddZde
fddZd ddZdefddZ ZS ) SVHNa `SVHN `_ Dataset.
Note: The SVHN dataset assigns the label `10` to the digit `0`. However, in this Dataset,
we assign the label `0` to the digit `0` to be compatible with PyTorch loss functions which
expect the class labels to be in the range `[0, C-1]`
.. warning::
This class needs `scipy `_ to load data from `.mat` format.
Args:
root (str or ``pathlib.Path``): Root directory of the dataset where the data is stored.
split (string): One of {'train', 'test', 'extra'}.
Accordingly dataset is selected. 'extra' is Extra training set.
transform (callable, optional): A function/transform that takes in a PIL image
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): If true, downloads the dataset from the internet and
puts it in root directory. If dataset is already downloaded, it is not
downloaded again.
)z6http://ufldl.stanford.edu/housenumbers/train_32x32.matztrain_32x32.matZ e26dedcc434d2e4c54c9b2d4a06d8373)z5http://ufldl.stanford.edu/housenumbers/test_32x32.matztest_32x32.matZ eb5a983be6a315427106f1b164d9cef3)z6http://ufldl.stanford.edu/housenumbers/extra_32x32.matzextra_32x32.matZ a93ce644f1a588dc4d68dda5feec44a7)traintestextrar NFrootsplit transformtarget_transformdownloadreturnc s t j|||d t|dt| j | _| j| d | _| j| d | _| j| d | _ |r3|
| s;t ddd l
m} |tj| j| j}|d | _|d tj | _t| j| jd kd t| jd
| _d S ) N)r r r r r zHDataset not found or corrupted. You can use download=True to download itXy
) r r r ) super__init__r tuple
split_listkeysr urlfilenameZfile_md5r _check_integrity RuntimeErrorZscipy.ioioZloadmatospathjoinr dataZastypenpZint64ZsqueezelabelsZplace transpose)selfr r r r r sioZ
loaded_mat __class__ _/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/torchvision/datasets/svhn.pyr 6 s
z
SVHN.__init__indexc C s\ | j | t| j| }}tt|d}| jdur | |}| jdur*| |}||fS )z
Args:
index (int): Index
Returns:
tuple: (image, target) where target is index of the target class.
)r r r N) r) intr+ r Z fromarrayr* r, r r )r- r3 Zimgtargetr1 r1 r2 __getitem__^ s
zSVHN.__getitem__c C s
t | jS )N)lenr) r- r1 r1 r2 __len__t s
z SVHN.__len__c C s0 | j }| j| j d }tj|| j}t||S Nr )r r r r& r' r( r" r )r- r md5Zfpathr1 r1 r2 r# w s
zSVHN._check_integrityc C s( | j | j d }t| j| j| j| d S r: )r r r
r! r r" )r- r; r1 r1 r2 r } s z
SVHN.downloadc C s dj di | jS )NzSplit: {split}r1 )format__dict__r8 r1 r1 r2
extra_repr s zSVHN.extra_repr)r NNF)r N)__name__
__module__ __qualname____doc__r r strr r r boolr r4 r r r6 r9 r# r r>
__classcell__r1 r1 r/ r2 r
s8
(
r
)Zos.pathr& pathlibr typingr r r r numpyr* ZPILr utilsr r
r Zvisionr r
r1 r1 r1 r2 s