khB d dl Zd dlmZ d dlmZmZmZmZm Z m
Z
d dl Z d dl
mZ ddlmZmZmZ ddlmZ G d d e Zy)
N)Path)AnyCallablecastOptionalTupleUnion)Image )check_integrity download_and_extract_archiveverify_str_arg)
VisionDatasetc v e Zd ZdZdZdZdZdZdZdZ dd gd
d gd d
ggZ
ddgddggZ dZ d'de
eef dedee dee dee deddf fd
Zdee dee fd Zd edeeef fd Zdefd Zd(d ed!ee deej6 eej6 f fd"Z defd#Z d)d$Z defd%Z dee ddfd&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 unlabeledtestNrootsplitfolds transformtarget_transformdownloadreturnc t | ||| t |d| j | _ | j | | _ |r| j n| j s t d | | j dk( r{| j | j d d | j d d \ | _ | _
t t j | j | _
| j# | n| j dk( r| j | j d d | j d d \ | _ | _
t t j | j | _
| 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 | j d d \ | _ }t j& d g| j j( d z | _
n?| j | j* d d | j* d d \ | _ | _
t, j. j1 | j2 | j4 | j6 } t, j. j9 | r8t; | 5 }
|
j= j? | _ d d d y y # 1 sw Y y xY w) 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 V/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/datasets/stl10.pyr( zSTL10.__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 NA >'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!
5Q vvx 2 2 4
5
5 &
5
5s 4$M""M+c ||S t |t r+|t d v r|S d}t |j | d}t |j 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)rB r msgs rH r* zSTL10._verify_folds[ s] =L
s
#b ! XC SZZ ./ /RC SZZU 45 5 indexc Z | j (| j | t | j | }}n| j | d}}t j t j |d }| j | j | }| j | j | }||fS )z
Args:
index (int): Index
Returns:
tuple: (image, target) where target is index of the target class.
N)r r% r ) r0 r/ rL r
fromarrayr1 transposer r! )rB rS imgtargets rH __getitem__zSTL10.__getitem__g s ;; " ))E*C E0B,CC ))E*DC oobll3 :; >> %..%C ,**62FF{rR c 4 | j j d S )Nr )r/ r6 rB s rH __len__z
STL10.__len__ s yyq!!rR data_file labels_filec b d }|rrt j j | j | j | }t |d 5 }t
j |t j dz
}d d d t j j | j | j | }t |d 5 }t
j |t j }t
j |d }t
j |d }d d d ||fS # 1 sw Y xY w# 1 sw Y |fS xY w)Nrb)dtyper )r& ` rc )r r rb r% ) r8 r9 r: r r; r> r1 fromfileuint8reshaperV ) rB r] r^ r0 path_to_labelsrF path_to_data
everythingimagess rH
__loadfilezSTL10.__loadfile s WW\\$))T5E5E{ SNnd+
" $TXXtyy4==VZVbVbc rR c : dj di | j S )NzSplit: {split} )rO __dict__r[ s rH
extra_reprzSTL10.extra_repr s & &&7 77rR c |y t j j | j | j | j
}t
| 5 }|j j | }t j |t j d }| j |d d d d d d f | _ | j | j | | _
d d d y # 1 sw Y y xY w)N )ra sep)r8 r9 r: r r; folds_list_filer> r? r@ r1
fromstringint64r/ r0 )rB r
path_to_foldsrF str_idxlist_idxs rH __load_foldszSTL10.__load_folds s = TYY0@0@$BVBVW
-
4Affh))+E2G}}WBHH#FH (Aq!"34DI{{& "kk( 3 4 4 4s
BC C&)r NNNF)N)r# N)"__name__
__module__ __qualname____doc__r; rq rm rr r< rz r. r7 r) r strr r rL r boolr( r* r r rY r\ r1 r2 r- r+ r" rv r3
__classcell__)rG s @rH r r s & !K
EC$H0G((O :; :; >?J BClTvEwxI
>F
#(,/3 *5CI *5 *5 } *5
H% *5 #8,
*5 *5
*5X
68C=
6Xc]
6 sCx 4" " C hsm uUWU_U_aijljtjtauUuOv $ 8C 8
4(3-
4D
4rR r )os.pathr8 pathlibr typingr r r r r r numpyr1 PILr
utilsr r
r visionr r rt rR rH r s. > > P P !b4M b4rR