DiD j d dl Z d dlZd dlmZmZmZmZ ddlmZ ddl m
Z
m Z ddl m
Z
G d de
Zy) N)AnyCallableOptionalUnion )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 Z/var/www/pod-logistic/pod-ai/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> r3 rA s rG __len__z DTD.__len__Q s 4$$%% idxc | j | | j | }}| j | }| j r| j | }| j r| j | }||fS rI )r3 r@ r r r )rA rM
image_filelabelimages rG __getitem__zDTD.__getitem__T se - -c 2DLL4EE
J' >>NN5)E ))%0Ee|rL c : d| j d| j S )Nzsplit=z , partition=)r r$ rJ s rG
extra_reprzDTD.extra_repr` s }L 0ABBrL c t j j | j xr) t j j | j S rI )ospathexistsr- isdirrJ s rG r1 zDTD._check_existsc s3 ww~~d / /0URWW]]4CTCT5UUrL c | j ry t | j t | j | j
y )N)
download_rootmd5)r1 r _URLstrr, _MD5rJ s rG r0 z
DTD._downloadf s3 $TYYc$BSBS>TZ^ZcZcdrL )r N)r*
__module__ __qualname____doc__r] r_ r r r^ r' r( r" r r boolr r& rK tuplerR rT r1 r0
__classcell__)rF s @rG r
r
s . RD -D
(,/3 rk s* 1 1 " ? !^e- ^erL