Ph f d dl Z d dlZd dlmZmZmZmZ d dlZddl m
Z
m Z ddl m
Z
G d de
Zy) N)AnyCallableOptionalTuple ) download_and_extract_archiveverify_str_arg)
VisionDatasetc e Zd ZdZdZdZ ddedededee d ee d
e
d df fd
Z d efd
Z ded e
eef fdZd efdZd e
fdZddZ xZS )DTDa> `Describable Textures Dataset (DTD) `_.
Args:
root (string): Root directory of the dataset.
split (string, optional): The dataset split, supports ``"train"`` (default), ``"val"``, or ``"test"``.
partition (int, optional): The dataset partition. Should be ``1 <= partition <= 10``. Defaults to ``1``.
.. note::
The partition only changes which split each image belongs to. Thus, regardless of the selected
partition, combining all splits will result in all images.
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. Default is False.
zDhttps://www.robots.ox.ac.uk/~vgg/data/dtd/download/dtd-r1.0.1.tar.gz fff73e5086ae6bdbea199a49dfb8a4c1Nrootsplit partition transformtarget_transformdownloadreturnc
t |dd | _ t |t s d|cxk rdk sn t d| d || _ t | ||| t j | j t | j j z | _ | j dz | _ | j d z | _ | j d
z | _ |r| j% | j' s t) d g | _ g }t- | j | j | j
d z 5 }|D ]j } | j/ j1 d
\ }
} | j* j3 | j" j5 |
| |j3 |
l d d d t7 t9 | | _ t= t? | j: tA tC | j: | _" |D
cg c] }
| jD |
c}
| _# y # 1 sw Y xY wc c}
w )Nr )trainvaltestr
zPParameter 'partition' should be an integer with `1 <= partition <= 10`, but got z instead)r r dtdlabelsimagesz;Dataset not found. You can use download=True to download itz.txt/)$r _split
isinstanceint
ValueError
_partitionsuper__init__pathlibPathr type__name__lower _base_folder _data_folder _meta_folder_images_folder _download
_check_exists RuntimeError _image_filesopenstripr appendjoinpathsortedsetclassesdictziprangelen class_to_idx_labels)
selfr r r r r r r8 filelineclsname __class__s
cC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/datasets/dtd.pyr$ z DTD.__init__# s %UG5MN )S)1 3GR3G $+X/
$
EUV #LL 3d4j6I6I6O6O6Q Q - - 5 - - 8 " / /( : NN !!# \] ]
$##
doo5Fd&KK
LPT JJL . .s 3 T!!(()<)<)E)Ec4)PQs# M c'l+ T\\5T\\9J3K!L M:AB'3))#.'B M
L Cs A0H9I9Ic , t | j S N)r< r1 r? s rE __len__z DTD.__len__L s 4$$%% idxc | j | | j | }}t j j | j d }| j r| j
| }| j r| j | }||fS )NRGB)r1 r> PILImager2 convertr r )r? rK
image_filelabelimages rE __getitem__zDTD.__getitem__O st - -c 2DLL4EE
z*2259 >>NN5)E ))%0Ee|rJ c : d| j d| j S )Nzsplit=z , partition=)r r" rH s rE
extra_reprzDTD.extra_repr[ s }L 0ABBrJ c t j j | j xr) t j j | j S rG )ospathexistsr+ isdirrH s rE r/ zDTD._check_exists^ s3 ww~~d / /0URWW]]4CTCT5UUrJ c | j ry t | j t | j | j
y )N)
download_rootmd5)r/ r _URLstrr* _MD5rH s rE r. z
DTD._downloada s3 $TYYc$BSBS>TZ^ZcZcdrJ )r r NNF)r N)r(
__module__ __qualname____doc__r_ ra r` r r r boolr$ rI r r rT rV r/ r.
__classcell__)rD s @rE r r s ( RD -D
(,/3 'C'C 'C 'C
H% 'C #8,
'C 'C
'CR& &
s
uS#X
CC CVt VerJ r )rX r% typingr r r r PIL.ImagerN utilsr r visionr
r rJ rE rl s* 1 1 ? !Ye- YerJ