khK n d dl Z d dlZd dlmZmZmZmZmZ ddlm Z ddl
m Z m Z ddl
mZ G d de Zy) N)AnyCallableOptionalTupleUnion )default_loader) download_and_extract_archiveverify_str_arg)
VisionDatasetc e Zd ZdZdZdZdddddefdeee j f d ed
e d e e
d e e
d
ede
eee j f gef ddf fd
Zde fdZde deeef fdZdefdZdefdZddZ xZS )DTDa} `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 or torch.Tensor, depends on the given loader,
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.
loader (callable, optional): A function to load an image given its path.
By default, it uses PIL as its image loader, but users could also pass in
``torchvision.io.decode_image`` for decoding image data into tensors directly.
zDhttps://www.robots.ox.ac.uk/~vgg/data/dtd/download/dtd-r1.0.1.tar.gz fff73e5086ae6bdbea199a49dfb8a4c1trainr NFrootsplit partition transformtarget_transformdownloadloaderreturnc
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 )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/)%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_labelsr )selfr r r r r r r r; filelineclsname __class__s T/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/datasets/dtd.pyr' z DTD.__init__& s %UG5MN )S)1 3GR3G $+X/
$
EUV #LL 3d4j6I6I6O6O6Q Q - - 5 - - 8 " / /( : NN !!# \] ]
$##
doo5Fd&KK
L $PT
$ JJL . .s 3 T!!(()<)<)E)Ec4)PQs#
$ $ c'l+ T\\5T\\9J3K!L M:AB3))#.B $ $ Cs A0I I I c , t | j S N)r? r4 rB s rH __len__z DTD.__len__Q s 4$$%% idxc | j | | j | }}| j | }| j r| j | }| j r| j | }||fS rJ )r4 rA r r r )rB rN
image_filelabelimages rH __getitem__zDTD.__getitem__T se - -c 2DLL4EE
J' >>NN5)E ))%0Ee|rM c : d| j d| j S )Nzsplit=z , partition=)r! r% rK s rH
extra_reprzDTD.extra_repr` s }L 0ABBrM c t j j | j xr) t j j | j S rJ )ospathexistsr. isdirrK s rH r2 zDTD._check_existsc s3 ww~~d / /0URWW]]4CTCT5UUrM c | j ry t | j t | j | j
y )N)
download_rootmd5)r2 r
_URLstrr- _MD5rK s rH r1 z
DTD._downloadf s3 $TYYc$BSBS>TZ^ZcZcdrM )r N)r+
__module__ __qualname____doc__r^ r` r r r_ r( r) r# r r boolr r' rL r rS rU r2 r1
__classcell__)rG s @rH r r s . RD -D
(,/3 rk s* 8 8 " ? !^e- ^erM