o
)i @ s` d dl Z d dlZd dlmZmZmZmZ d dlZddl m
Z
m Z ddl m
Z
G dd de
ZdS ) N)AnyCallableOptionalTuple ) download_and_extract_archiveverify_str_arg)
VisionDatasetc s e Zd ZdZdZdZ d ded ed
ed ee d ee d
e
ddf fdd
Z defddZ dede
eef fddZdefddZde
fddZd ddZ ZS ) 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.gzZ fff73e5086ae6bdbea199a49dfb8a4c1trainr NFrootsplit partition transformtarget_transformdownloadreturnc sv t |dd _t|ts d| krdks n td| d| _t j|||d t j
t j
_ jd _ jd _ jd
_|rQ sYtd g _g }t j j j d &}|D ]} | d
\}
} j j|
| ||
qoW d n1 sw Y tt | _ t t j t t! j _" fdd|D _#d S )Nr
)r valtestr
zPParameter 'partition' should be an integer with `1 <= partition <= 10`, but got z instead)r r dtdlabelsZimagesz;Dataset not found. You can use download=True to download itz.txt/c s g | ]} j | qS ) class_to_idx).0clsselfr g/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/datasets/dtd.py
J s z DTD.__init__..)$r _split
isinstanceint
ValueError
_partitionsuper__init__pathlibPathr type__name__lower _base_folder _data_folderZ _meta_folderZ_images_folder _download
_check_exists RuntimeError _image_filesopenstripr
appendjoinpathsortedsetclassesdictziprangelenr _labels) r r r
r r r r r9 fileliner name __class__r r r' # s: z DTD.__init__c C s
t | jS N)r= r2 r r r r __len__L s
z DTD.__len__idxc C sP | j | | j| }}tj|d}| jr | |}| jr$| |}||fS )NRGB)r2 r> PILZImager3 convertr r )r rF Z
image_filelabelimager r r __getitem__O s
zDTD.__getitem__c C s d| j d| j S )Nzsplit=z , partition=)r! r% r r r r
extra_repr[ s zDTD.extra_reprc C s t j| jo
t j| jS rD )ospathexistsr. isdirr r r r r0 ^ s zDTD._check_existsc C s( | rd S t| jt| j| jd d S )N)Z
download_rootmd5)r0 r _URLstrr- _MD5r r r r r/ a s z
DTD._download)r r NNF)r N)r+
__module__ __qualname____doc__rS rU rT r# r r boolr' rE r r rL rM r0 r/
__classcell__r r rB r r
s: ) r
)rN r( typingr r r r Z PIL.ImagerH utilsr r Zvisionr r
r r r r s