j d dl Zd dlmZ d dlmZmZmZmZm Z m
Z
d dl Z d dl
mZ ddlmZmZmZ ddlmZ G d d e ZdS )
N)Path)AnyCallablecastOptionalTupleUnion)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 KC:\Projects\Historical\venv\Lib\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)rB r msgs rH 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 ) r0 r/ rL r
fromarrayr1 transposer r! )rB rS imgtargets rH __getitem__zSTL10.__getitem__g s ; " )E*C E0B,C,CCC )E*DC obl3 : :;; > %..%%C ,**622FF{rR c & | j j d S )Nr )r/ r6 rB s rH __len__z
STL10.__len__ s yq!!rR 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& ` 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 sl < W\\$)T5E{ S SNnd++
r? r@ r1
fromstringint64r/ r0 )rB r
path_to_foldsrF str_idxlist_idxs rH __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; 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 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
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 P P P P P P P P ! ! ! ! ! !b4 b4 b4 b4 b4M b4 b4 b4 b4 b4rR