YhD r d dl Z d dlZd dlmZmZmZmZ ddlmZ ddl m
Z
m Z ddl m
Z
G d de
ZdS ) 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
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 j/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/dtd.py
z DTD.__init__..N 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" _labelsr )r% r r r r r r r rC 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)rG r< r% s r& __len__z DTD.__len__Q s 4$%%%r( idxc | j | | j | }}| | }| j r| | }| j r| | }||fS rN )r< rH r r r )r% rQ
image_filelabelimages r& __getitem__zDTD.__getitem__T sn -c 2DL4EE
J'' > *NN5))E 1))%00Ee|r( c & d| j d| j S )Nzsplit=z , partition=)r) r- rO s r&
extra_reprzDTD.extra_repr` s B BB BBBr( c t j | j o#t j | j S rN )ospathexistsr6 isdirrO s r& r: zDTD._check_existsc s/ w~~d /00URW]]4CT5U5UUr( c | rd S t | j t | j | j d S )N)
download_rootmd5)r: r _URLstrr5 _MD5rO s r& r9 z
DTD._downloadf sG F$TYc$BS>T>TZ^Zcddddddr( )r N)r3
__module__ __qualname____doc__ra rc r r rb r0 r1 r+ r r boolr r/ rP tuplerV rX r: r9
__classcell__)rL s @r& r
r
s . RD -D
(,/3 rn s 1 1 1 1 1 1 1 1 1 1 1 1 " " " " " " ? ? ? ? ? ? ? ? ! ! ! ! ! !^e ^e ^e ^e ^e- ^e ^e ^e ^e ^er(