QЦiB S SK rS SKJr S SKJrJrJrJrJ r J
r
S SK r S SK
Jr SSKJrJrJr SSKJr " S S \5 rg)
N)Path)AnyCallablecastOptionalTupleUnion)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 Y/var/www/html/ai-image-ml/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)rC r msgs rI 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 ) r1 r0 rN r
fromarrayr2 transposer! r" )rC rV imgtargets rI __getitem__STL10.__getitem__g s ;; " ))E*C E0B,C ))E*D oobll3 :; >> %..%C ,**62F{rU c 4 U R R S $ )Nr )r0 r7 rC s rI __len__
STL10.__len__ s yyq!!rU 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' ` rh )r r rg r& ) r9 r: r; r r< r? r2 fromfileuint8reshaperY ) rC rb rc r1 path_to_labelsrG path_to_data
everythingimagess rI
__loadfileSTL10.__loadfile s WW\\$))T5E5E{ SNnd+qQbhh7!; , ww||DIIt/?/?K
,
%Qbhh7JZZ
O rU c U R 5 ( a g [ U R U R U R U R
S9 U R 5 g )N)rs rt )r, r
urlr rs tgz_md5r_ s rI r# STL10.download sA " " $TXXtyy4==VZVbVbc rU c : SR " S0 U R D6$ )NzSplit: {split} )rQ __dict__r_ s rI
extra_reprSTL10.extra_repr s &&7 77rU 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 )rf sep)r9 r: r; r r< folds_list_filer? r@ rA r2
fromstringint64r0 r1 )rC r
path_to_foldsrG str_idxlist_idxs rI __load_foldsSTL10.__load_folds s = TYY0@0@$BVBVW
-
Affh))+E2G}}WBHH#FH (Aq!"34DI{{& "kk( 3 !
s
BC
C+)rB r0 r r1 r )r NNNF)N)r$ N)$__name__
__module__ __qualname____firstlineno____doc__r< rx rs ry r= r r/ r8 r* r strr r rN r boolr) r+ r r r\ r` r2 r3 r. r, r# r~ r4 __static_attributes__
__classcell__)rH s @rI 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
4rU r )os.pathr9 pathlibr typingr r r r r r numpyr2 PILr
utilsr r
r visionr r r| rU rI r s. > > P P !b4M b4rU