Yh d dl Zd dlmZ d dlmZmZmZmZ d dl Z
d dl m Z ddl
mZmZmZ ddlmZ G d d e ZdS )
N)Path)AnyCallableOptionalUnion)Image )check_integrity download_urlverify_str_arg)
VisionDatasetc
e Zd ZdZg dg dg ddZ dd eeef d
ed ee d ee d
e
ddf fd
Z de de
eef fdZde fdZde
fdZddZdefdZ xZS )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.mat e26dedcc434d2e4c54c9b2d4a06d8373)z5http://ufldl.stanford.edu/housenumbers/test_32x32.matztest_32x32.mat eb5a983be6a315427106f1b164d9cef3)z6http://ufldl.stanford.edu/housenumbers/extra_32x32.matzextra_32x32.mat a93ce644f1a588dc4d68dda5feec44a7)traintestextrar NFrootsplit transformtarget_transformdownloadreturnc d t ||| t |dt | j | _ | j | d | _ | j | d | _ | j | d | _ |r|
| st d dd l
m} | t j | j | j }|d | _ |d t, j | _ t- j | j | j d k d t- j | j d
| _ 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 urlfilenamefile_md5r _check_integrity RuntimeErrorscipy.ioioloadmatospathjoinr dataastypenpint64squeezelabelsplace transpose) selfr r r r r sio
loaded_mat __class__s k/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/svhn.pyr# z
SVHN.__init__6 sg EUVVV#E7E$/:N:N:P:P4Q4QRR
?5)!, .q1
.q1
MMOOO$$&& k ijj j [[ di !G!GHH
sO
!o,,RX66>>@@
dkR /333LL99 indexc | j | t | j | }}t j t j |d }| j | | }| j | | }||fS )z
Args:
index (int): Index
Returns:
tuple: (image, target) where target is index of the target class.
)r r r ) r2 intr7 r fromarrayr4 r9 r r )r: r@ imgtargets r> __getitem__zSVHN.__getitem__^ s i&DK,>(?(?V obl3 : :;; > %..%%C ,**622FF{r? c * t | j S )N)lenr2 r: s r> __len__z SVHN.__len__t s 49~~ r? c | j }| j | j d }t j || j }t || S Nr )r r% r r/ r0 r1 r( r
)r: r md5fpaths r> r* zSVHN._check_integrityw sC yodj)!, T4=11 uc***r? c x | j | j d }t | j | j | j | d S rL )r% r r r' r r( )r: rM s r> r z
SVHN.download} s5 odj)!,TXty$-=====r? c & dj di | j S )NzSplit: {split} )format__dict__rI s r>
extra_reprzSVHN.extra_repr s & &77 777r? )r NNF)r N)__name__
__module__ __qualname____doc__r% r strr r r boolr# rB r$ r rF rJ r* r rT
__classcell__)r= s @r> r r sp 0
J* (,/3
&: &:CI &: &: H% &:
#8, &:
&:
&: &: &: &: &: &:P sCx , +$ + + + + > > > >8C 8 8 8 8 8 8 8 8r? r )os.pathr/ pathlibr typingr r r r numpyr4 PILr utilsr
r r visionr
r rQ r? r> rc s 1 1 1 1 1 1 1 1 1 1 1 1 @ @ @ @ @ @ @ @ @ @ ! ! ! ! ! !v8 v8 v8 v8 v8= v8 v8 v8 v8 v8r?