a
hh @ s\ d dl mZ d dlmZmZmZmZ ddlmZ ddl m
Z
ddl m Z G dd de Z
d S )
)Path)AnyCallableOptionalUnion )default_loader) download_and_extract_archive)
VisionDatasetc s e Zd ZdZdZdZdddefeee f e
e e
e e e eee f ge
f dd fdd
Zed d
d Zeee
e
f d d
dZe d ddZdd ddZ ZS )SUN397a4 `The SUN397 Data Set `_.
The SUN397 or Scene UNderstanding (SUN) is a dataset for scene recognition consisting of
397 categories with 108'754 images.
Args:
root (str or ``pathlib.Path``): Root directory of the dataset.
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.
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.
z;http://vision.princeton.edu/projects/2010/SUN/SUN397.tar.gzZ 8ca2778205c41d23104230ba66911c7aNF)root transformtarget_transformdownloadloaderreturnc s t j|||d t jd _|r. s>tdt jd }dd |D _ W d n1 sr0 Y t
t j t t
j _t jd _ fdd jD _| _d S ) N)r
r r z;Dataset not found. You can use download=True to download itz
ClassName.txtc S s g | ]}|d d qS ) N)strip).0c r I/var/www/auris/lib/python3.9/site-packages/torchvision/datasets/sun397.py
2 z#SUN397.__init__..z sun_*.jpgc s. g | ]&} j d | jjdd qS )/r ) class_to_idxjoin relative_to _data_dirparts)r pathselfr r r 7 s )super__init__r r r _download
_check_exists RuntimeErroropenclassesdictziprangelenr listrglob _image_files_labelsr )r# r r
r r r f __class__r" r r% s .
zSUN397.__init__)r c C s
t | jS N)r. r1 r" r r r __len__< s zSUN397.__len__)idxr c C sH | j | | j| }}| |}| jr0| |}| jr@| |}||fS r6 )r1 r2 r r
r )r# r8 Z
image_filelabelimager r r __getitem__? s
zSUN397.__getitem__c C s
| j S r6 )r is_dirr" r r r r' K s zSUN397._check_existsc C s$ | r d S t| j| j| jd d S )N)Z
download_rootmd5)r' r _DATASET_URLr _DATASET_MD5r" r r r r& N s zSUN397._download)__name__
__module__ __qualname____doc__r> r? r r strr r r boolr r% intr7 tupler; r' r&
__classcell__r r r4 r r
s&
r N)pathlibr typingr r r r folderr utilsr Zvisionr
r r r r r s