j r d dl Z d dlZd dlmZmZmZmZmZ d dlZ ddl
m Z m Z ddl
mZ G d de ZdS ) N)AnyCallableOptionalTupleUnion ) download_and_extract_archiveverify_str_arg)
VisionDatasetc e Zd ZdZdZdZ ddeeej f d ed
e
d e e d e e d
e
ddf fd
Zde
fdZde
deeef fdZdefdZde
fdZddZ xZS )DTDaN `Describable Textures Dataset (DTD) `_.
Args:
root (str or ``pathlib.Path``): 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 fff73e5086ae6bdbea199a49dfb8a4c1trainr NFrootsplit partition transformtarget_transformdownloadreturnc
t |dd _ t |t s"d|cxk rdk sn t d| d | _ t
||| t j j
t j
z _ j dz _ j d z _ j d
z _ |r st) d g _ g }t- j j j d z 5 }|D ]t} | d
\ }
} j j |
| | |
u d d d n # 1 swxY w Y t7 t9 | _ t= t? j tA tC j _" fd|D _# d S )Nr )r valtestr
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/c * g | ]}j | S ) class_to_idx).0clsselfs IC:\Projects\Historical\venv\Lib\site-packages\torchvision/datasets/dtd.py
z DTD.__init__..J s! BBB3)#.BBB )$r
_split
isinstanceint
ValueError
_partitionsuper__init__pathlibPathr type__name__lower _base_folder _data_folder _meta_folder_images_folder _download
_check_exists RuntimeError _image_filesopenstripr appendjoinpathsortedsetclassesdictziprangelenr! _labels)
r$ r r r r r r rB fileliner# name __class__s
` r% r. z DTD.__init__# s %UG5MNN )S)) 1 3G3G3G3GR3G3G3G3G /$/ / /
$
EUVVV #L 3 3d4jj6I6O6O6Q6Q Q - 5 - 8 " /( : NN !!## ^ \]] ]
$#&Kdo&K&K&KK
L
L $PT
$
$ JJLL . .s 3 3 T!(()<)E)Ec4)P)PQQQs####
$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ c'll++ T\5T\9J9J3K3K!L!L M MBBBB'BBB s A8GG !G c * t | j S N)rF r; r$ s r% __len__z DTD.__len__L s 4$%%%r' idxc | j | | j | }}t j | d }| j r| | }| j r| | }||fS )NRGB)r; rG PILImager< convertr r )r$ rP
image_filelabelimages r% __getitem__zDTD.__getitem__O s~ -c 2DL4EE
z**22599 > *NN5))E 1))%00Ee|r' c & d| j d| j S )Nzsplit=z , partition=)r( r, rN s r%
extra_reprzDTD.extra_repr[ s B BB BBBr' c t j | j o#t j | j S rM )ospathexistsr5 isdirrN s r% r9 zDTD._check_exists^ s/ w~~d /00URW]]4CT5U5UUr' c | rd S t | j t | j | j d S )N)
download_rootmd5)r9 r _URLstrr4 _MD5rN s r% r8 z
DTD._downloada sG F$TYc$BS>T>TZ^Zcddddddr' )r r NNF)r N)r2
__module__ __qualname____doc__rd rf r re r/ r0 r* r r boolr. rO r r rY r[ r9 r8
__classcell__)rK s @r% r
r
st ( RD -D
(,/3 'C 'CC %&'C 'C 'C
H% 'C #8,
'C 'C
'C 'C 'C 'C 'C 'CR& & & & &
s
uS#X
CC C C C CVt V V V Ve e e e e e e er' r
)r] r/ typingr r r r r PIL.ImagerS utilsr r
visionr r
r r' r% rp s 8 8 8 8 8 8 8 8 8 8 8 8 8 8 ? ? ? ? ? ? ? ? ! ! ! ! ! !Ye Ye Ye Ye Ye- Ye Ye Ye Ye Yer'