Di; ~ d dl Zd dlmZ d dlmZmZmZmZm Z d dl
Z d dl m
Z
ddlmZmZmZ ddlmZ G d d e Zy)
N)Path)AnyCallablecastOptionalUnion)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 \/var/www/pod-logistic/pod-ai/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)rA r msgs rG 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 ) r/ r. rK r fromarrayr0 transposer r )rA rR imgtargets rG __getitem__zSTL10.__getitem__g s ;; " ))E*C E0B,CC ))E*DC oobll3 :; >> %..%C ,**62FF{rQ c 4 | j j d S )Nr )r. r5 rA s rG __len__z
STL10.__len__ s yyq!!rQ 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% ` 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 s WW\\$))T5E5E{ SNnd+
" $TXXtyy4==VZVbVbc rQ c : dj di | j S )NzSplit: {split} )rN __dict__rZ s rG
extra_reprzSTL10.extra_repr s & &&7 77rQ 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 )r` sep)r7 r8 r9 r r: folds_list_filer= r> r? r0
fromstringint64r. r/ )rA r
path_to_foldsrE str_idxlist_idxs rG __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: 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 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
4rQ r )os.pathr7 pathlibr typingr r r r r numpyr0 PILr utilsr r r
visionr r rs rQ rG r s. 7 7 P P !b4M b4rQ