Ei; ~ S SK rS SKJr S SKJrJrJrJrJ r S SK
r S SK J
r
SSKJrJrJr SSKJr " S S \5 rg)
N)Path)AnyCallablecastOptionalUnion)Image )check_integrity download_and_extract_archiveverify_str_arg)
VisionDatasetc ^ \ rS rSrSrSrSrSrSrSr Sr
S S
/S S /S
S//r SS/SS//r Sr
S)S\\\4 S\S\\ S\\ S\\ S\SS4U 4S jjjrS\\ S\\ 4S jrS \S\\\4 4S jrS\4S jrS*S!\S"\\ S\\R8 \\R8 4 4S# jjr S\4S$ jr S+S% jr S\4S& jr S\\ SS4S' jr!S(r"U =r#$ ),STL10 a `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 unlabeledtestNrootsplitfolds transformtarget_transformdownloadreturnc > [ T U ] XUS9 [ USU R 5 U l U R U5 U l U( a U R 5 O U R 5 ( d [ S5 eU U R S:X a{ U R U R S S U R S S 5 u U l U l
[ [ R U R 5 U l
U R# U5 GOU R S:X Ga U R U R S S U R S S 5 u U l U l
[ [ R U R 5 U l
U R# U5 U R U R S S 5 u px[ R$ " U R U45 U l [ R$ " U R [ R& " S /UR( S - 5 45 U l
OU R S
:X a` U R U R S S 5 u U l n[ R& " S /U R R( S - 5 U l
O?U R U R* S S U R* S S 5 u U l U l
[, R. R1 U R2 U R4 U R6 5 n [, R. R9 U 5 ( a8 [; U 5 n
U
R= 5 R? 5 U l S S S 5 g g ! , ( d f g = f) 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 W/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torchvision/datasets/stl10.pyr( STL10.__init__/ s EUV#E7DKK@
''.
MMO&&(( ij j
:: %)__T__Q5G5JDOO\]L^_`La%b "DIt{ rzz4;;7DK e $
ZZ,
,%)__T__Q5G5JDOO\]L^_`La%b "DIt{ rzz4;;7DK e $ $0B10E F N >'BCDI ..$++rzz2$ I]I]^_I`B`7a)bcDK
ZZ;
& ??4??1+=a+@ALDIq **bTDIIOOA,>%>?DK%)__T^^A5Fq5I4>>Z[K\]^K_%` "DIt{WW\\$))T-=-=t?T?TU
77>>* % %j!Q vvx 2 2 4 "! &!!s $M33
Nc Uc U$ [ U[ 5 ( a- U[ S5 ; a U$ Sn[ UR U5 5 eSn[ UR [ U5 5 5 e)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)rB r msgs rH r* STL10._verify_folds[ s` =L
s
#
#b ! XC SZZ ./ /RC SZZU 45 5 indexc ^ U R b' U R U [ U R U 5 p2OU R U Sp2[ R " [
R " US5 5 nU R b U R U5 nU R b U R U5 nX#4$ )zn
Args:
index (int): Index
Returns:
tuple: (image, target) where target is index of the target class.
N)r
r% r ) r0 r/ rM r fromarrayr1 transposer r! )rB rU imgtargets rH __getitem__STL10.__getitem__g s ;; " ))E*C E0B,C ))E*D oobll3 :; >> %..%C ,**62F{rT c 4 U R R S $ )Nr )r/ r6 rB s rH __len__
STL10.__len__ s yyq!!rT data_file labels_filec z S nU( ap [ R R U R U R U5 n[ US5 n[ R " U[ R S9S-
nS S S 5 [ R R U R U R U5 n[ US5 n[ R " U[ R S9n[ R " US5 n[ R " US5 nS S S 5 X4$ ! , ( d f N= f! , ( d f WU4$ = f)Nrb)dtyper
)r& ` rg )r r
rf r% ) r8 r9 r: r r; r> r1 fromfileuint8reshaperX ) rB ra rb r0 path_to_labelsrF path_to_data
everythingimagess rH
__loadfileSTL10.__loadfile s WW\\$))T5E5E{ SNnd+qQbhh7!; , ww||DIIt/?/?K
,
%Qbhh7JZZ
O rT c U R 5 ( a g [ U R U R U R U R
S9 U R 5 g )N)rr rs )r+ r urlr rr tgz_md5r^ s rH r" STL10.download sA " " $TXXtyy4==VZVbVbc rT c : SR " S0 U R D6$ )NzSplit: {split} )rP __dict__r^ s rH
extra_reprSTL10.extra_repr s &&7 77rT c Uc g [ R R U R U R U R
5 n[
U5 nUR 5 R 5 U n[ R " U[ R SS9nU R US S 2S S 2S S 24 U l U R b U R U U l
S S S 5 g ! , ( d f g = f)N )re sep)r8 r9 r: r r; folds_list_filer> r? r@ r1
fromstringint64r/ r0 )rB r
path_to_foldsrF str_idxlist_idxs rH __load_foldsSTL10.__load_folds s = TYY0@0@$BVBVW
-
Affh))+E2G}}WBHH#FH (Aq!"34DI{{& "kk( 3 !
s
BC
C+)rA r/ r r0 r )r NNNF)N)r# N)$__name__
__module__ __qualname____firstlineno____doc__r; rw rr rx r< r r. r7 r) r strr r rM r boolr( r* tupler r[ r_ r1 r2 r- r+ r" r} r3 __static_attributes__
__classcell__)rG s @rH r r s & !K
EC$H0G((O :; :; >?J BClTvEwxI
>F
#(,/3 *5CI *5 *5 } *5
H% *5 #8,
*5 *5
*5 *5X
68C=
6Xc]
6 sCx 4" " C hsm uUWU_U_aijljtjtauUuOv $ 8C 8
4(3-
4D
4
4rT r )os.pathr8 pathlibr typingr r r r r numpyr1 PILr utilsr r r
visionr r r{ rT rH r s. 7 7 P P !b4M b4rT