o
V hB @ s| d dl Zd dlmZ d dlmZmZmZmZm Z m
Z
d dl Z d dl
mZ ddlmZmZmZ ddlmZ G dd d eZdS )
N)Path)AnyCallablecastOptionalTupleUnion)Image )check_integrity download_and_extract_archiveverify_str_arg)
VisionDatasetc sJ e Zd ZdZdZdZdZdZdZdZ dd gd
d gd d
ggZ
ddgddggZ dZ d2de
eef dedee dee dee ded df fd d
Zdee d ee fd d Zd!ed eeef fd"d#Zd efd$d%Zd3d&ed'ee d eejeej f fd(d)Z d efd*d+Z d4d,d-Z d efd.d/Z dee d dfd0d1Z Z!S )5STL10a `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 s t j|||d t|d| j| _| || _|r | n| s&t d| | jdkrO|
| j d d | j d d \| _ | _
ttj| j
| _
| | n| jdkr|
| j d d | j d d \| _ | _
ttj| j
| _
| | |
| j d d \}}t| j |f| _ t| j
td g|jd f| _
n4| jd
kr|
| j d d \| _ }td g| j jd | _
n|
| jd d | jd d \| _ | _
tj| j| j| j} tj | rt | }
|
| _ W d d S 1 sw 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__ N/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/datasets/stl10.pyr% / s:
(
(
&
(
"zSTL10.__init__c C sN |d u r|S t |tr |tdv 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)r? r msgrF rF rG r' [ s
zSTL10._verify_foldsindexc C sx | j dur| j| t| j | }}n| j| d}}tt|d}| jdur.| |}| jdur8| |}||fS )z
Args:
index (int): Index
Returns:
tuple: (image, target) where target is index of the target class.
N)r
r" r ) r- r, rJ r fromarrayr. transposer r )r? rP imgtargetrF rF rG __getitem__g s
zSTL10.__getitem__c C s | j jd S )Nr )r, r3 r? rF rF rG __len__ s z
STL10.__len__ data_file labels_filec C s d }|r-t j| j| j|}t|d}tj|tjdd }W d n1 s(w Y t j| j| j|}t|d }tj|tjd}t |d}t
|d}W d ||fS 1 s^w Y ||fS )Nrb)dtyper
)r# ` r] )r r
r\ r" ) r5 r6 r7 r r8 r; r. fromfileuint8reshaperR ) r? rX rY r- path_to_labelsrC path_to_data
everythingimagesrF rF rG
__loadfile s
zSTL10.__loadfilec C s> | j | j D ]\}}tj| j| j|}t||s dS qdS )NFT)r+ r4 r5 r6 r7 r r8 r )r? filenamemd5fpathrF rF rG r( s
zSTL10._check_integrityc C s0 | rd S t| j| j| j| jd | d S )N)rf rg )r( r urlr rf tgz_md5rV rF rF rG r s zSTL10.downloadc C s dj di | jS )NzSplit: {split}rF )rM __dict__rV rF rF rG
extra_repr s zSTL10.extra_reprc C s |d u rd S t j| j| j| j}t|=}| | }t j
|t j dd}| j |d d d d d d f | _ | j
d urJ| j
| | _
W d d S W d d S 1 sUw Y d S )N )r[ sep)r5 r6 r7 r r8 folds_list_filer; r< r= r.
fromstringint64r, r- )r? r
path_to_foldsrC str_idxlist_idxrF rF rG __load_folds s
"zSTL10.__load_folds)r NNNF)N)r! N)"__name__
__module__ __qualname____doc__r8 ri rf rj r9 ro r+ r4 r& r strr r rJ r boolr% r' r r rU rW r. r/ r* r( r rl r0
__classcell__rF rF rD rG r sT
, ,
r )os.pathr5 pathlibr typingr r r r r r numpyr. PILr utilsr r r
visionr r rF rF rF rG s