Yh; d dl Zd dlmZ d dlmZmZmZmZm Z d dl
Z d dl m
Z
ddlmZmZmZ ddlmZ G d d e ZdS )
N)Path)AnyCallablecastOptionalUnion)Image )check_integrity download_and_extract_archiveverify_str_arg)
VisionDatasetc e Zd ZdZdZdZdZdZdZdZ dd gd
d gd d
ggZ
ddgddggZ dZ d)de
eef dedee dee dee ded df fd
Zdee d ee fd Zd ed eeef fd Zd efd!Zd*d"ed#ee d eej eej f fd$Z d efd%Z d+d&Z d efd'Z dee d dfd(Z xZ!S ),STL10a `STL10 `_ Dataset.
Args:
root (str or ``pathlib.Path``): Root directory of dataset where directory
``stl10_binary`` exists.
split (string): One of {'train', 'test', 'unlabeled', 'train+unlabeled'}.
Accordingly, dataset is selected.
folds (int, optional): One of {0-9} or None.
For training, loads one of the 10 pre-defined folds of 1k samples for the
standard evaluation procedure. If no value is passed, loads the 5k samples.
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.
stl10_binaryz9http://ai.stanford.edu/~acoates/stl10/stl10_binary.tar.gzzstl10_binary.tar.gz 91f7769df0f17e558f3565bffb0c7dfbzclass_names.txtzfold_indices.txtz train_X.bin 918c2871b30a85fa023e0c44e0bee87fz train_y.bin 5a34089d4802c674881badbb80307741zunlabeled_X.bin 5242ba1fed5e4be9e1e742405eb56ca4z
test_X.bin 7f263ba9f9e0b06b93213547f721ac82z
test_y.bin 36f9794fa4beb8a2c72628de14fa638e)traintrain+unlabeled unlabeledtestr NFrootsplitfolds transformtarget_transformdownloadreturnc t ||| t |d| j | _ | | | _ |r| n#| st d | | j dk r|
| j d d | j d d \ | _ | _
t t j | j
| _
| | n| j dk r |
| j d d | j d d \ | _ | _
t t j | j
| _
| | |
| j d d \ }}t j | j |f | _ t j | j
t j d g|j d z f | _
n| j d
k r\|
| j d d \ | _ }t j d g| j j d z | _
nE|
| j d d | j d d \ | _ | _
t, j | j | j | j } t, j | rUt; | 5 }
|
| _ d d d d S # 1 swxY w Y d S d S ) N)r r r zHDataset not found or corrupted. You can use download=True to download itr r r
r r )!super__init__r
splitsr
_verify_foldsr r! _check_integrity RuntimeError_STL10__loadfile
train_listdatalabelsr npndarray_STL10__load_folds concatenateasarrayshape test_listospathjoinr base_folderclass_names_fileisfileopenread
splitlinesclasses) selfr r r r r r! unlabeled_data_
class_filef __class__s l/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/stl10.pyr' zSTL10.__init__/ s/ EUVVV#E7DK@@
''..
k MMOOOO&&(( k ijj j
) : %)__T_Q5G5JDO\]L^_`La%b%b "DIt{ rz4;77DK e $ $ $ $
Z,
,
,%)__T_Q5G5JDO\]L^_`La%b%b "DIt{ rz4;77DK e $ $ $ $0B10E F F NA >'BCCDI .$+rz2$ I]^_I`B`7a7a)bccDKK
Z;
&
& ??4?1+=a+@AALDIq *bTDIOA,>%>??DKK%)__T^A5Fq5I4>Z[K\]^K_%`%` "DIt{W\\$)T-=t?TUU
7>>* % % 5j!!
5Q vvxx 2 2 4 4
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5 5 5s ,M::M>M>c ||S t |t r7|t d v r|S d}t | | d}t | t | )N
zDValue for argument folds should be in the range [0, 10), but got {}.z>Expected type None or int for argument folds, but got type {}.)
isinstanceintrange
ValueErrorformattype)rA r msgs rG r) zSTL10._verify_folds[ sx =L
s
#
# 6b !! XC SZZ . .// /RC SZZU 4 455 5 indexc D | j (| j | t | j | }}n| j | d}}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.
N)r
r$ r ) r/ r. rK r fromarrayr0 transposer r )rA rR imgtargets rG __getitem__zSTL10.__getitem__g s ; " )E*C E0B,C,CCC )E*DC obl3 : :;; > %..%%C ,**622FF{rQ c & | j j d S )Nr )r. r5 rA s rG __len__z
STL10.__len__ s yq!!rQ data_file labels_filec 8 d }|rvt j | j | j | }t |d 5 }t
j |t j dz
}d d d n # 1 swxY w Y t j | j | j | }t |d 5 }t
j |t j }t
j |d }t
j
|d }d d d n # 1 swxY w Y ||fS )Nrb)dtyper
)r% ` rb )r r
ra r$ ) r7 r8 r9 r r: r= r0 fromfileuint8reshaperU ) rA r\ r] r/ path_to_labelsrE path_to_data
everythingimagess rG
__loadfilezSTL10.__loadfile sl < W\\$)T5E{ S SNnd++
r? r0
fromstringint64r. r/ )rA r
path_to_foldsrE str_idxlist_idxs rG __load_foldszSTL10.__load_folds s = F TY0@$BVWW
-
4Affhh))++E2G}WBH#FFFH (AAAqqq!!!"34DI{& "k( 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s BCCC)r NNNF)N)r" N)"__name__
__module__ __qualname____doc__r: rp rl rq r; ry r- r6 r( r strr r rK r boolr' r) tupler rX r[ r0 r1 r, r* r! ru r2
__classcell__)rF s @rG r r sB & !K
EC$H0G((O :; :; >?J BClTvEwxI
>F
#(,/3 *5 *5CI *5 *5 } *5
H% *5 #8,
*5 *5
*5 *5 *5 *5 *5 *5X
68C=
6Xc]
6
6
6
6 sCx 4" " " " " C hsm uUWU_aijljtauUuOv $ 8C 8 8 8 8
4(3-
4D
4
4
4
4
4
4
4
4rQ r )os.pathr7 pathlibr typingr r r r r numpyr0 PILr utilsr r r
visionr r rs rQ rG r s 7 7 7 7 7 7 7 7 7 7 7 7 7 7 P P P P P P P P P P ! ! ! ! ! !b4 b4 b4 b4 b4M b4 b4 b4 b4 b4rQ