o
;)iK @ sh d dl Z d dlZd dlmZmZmZmZmZ ddlm Z ddl
m Z m Z ddl
mZ G dd deZdS ) N)AnyCallableOptionalTupleUnion )default_loader) download_and_extract_archiveverify_str_arg)
VisionDatasetc s e Zd ZdZdZdZdddddefdeee j
f d ed
e d e e
d e e
d
ede
eee j
f gef ddf fdd
Zde fddZde deeef fddZdefddZdefddZd dd Z ZS ) 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.gzZ fff73e5086ae6bdbea199a49dfb8a4c1trainr NFrootsplit partition transformtarget_transformdownloadloaderreturnc
s| 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 S/home/app/PyTorch/.pytorch/lib/python3.10/site-packages/torchvision/datasets/dtd.py
N 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 _labelsr )
r! r r r r r r r r< fileliner name __class__r r" r* & s<
z DTD.__init__c C s
t | jS N)r@ r5 r r r r" __len__Q s
z DTD.__len__idxc C sH | j | | j| }}| |}| jr| |}| jr | |}||fS rG )r5 rA r r r )r! rI Z
image_filelabelimager r r" __getitem__T 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 rG )ospathexistsr1 isdirr r r r" r3 c s zDTD._check_existsc C s( | rd S t| jt| j| jd d S )N)Z
download_rootmd5)r3 r _URLstrr0 _MD5r r r r" r2 f s z
DTD._download)r N)r.
__module__ __qualname____doc__rS rU r r rT r+ r, r& r r boolr r* rH r rL rM r3 r2
__classcell__r r rE r" r s@ + r )rN r+ typingr r r r r folderr utilsr r
Zvisionr r r r r r" s