o
0i
@ s\ d dl mZ d dlmZmZmZmZmZ d dlZ ddl
m Z ddl m
Z
G dd de
ZdS ) )Path)AnyCallableOptionalTupleUnionN ) download_and_extract_archive)
VisionDatasetc s e Zd ZdZdZdZ ddeeef de e
de e
d e d
df
fd d
Z d
e
fd
dZde
d
eeef fddZd
e fddZdddZ ZS )SUN397a `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 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.
z;http://vision.princeton.edu/projects/2010/SUN/SUN397.tar.gz 8ca2778205c41d23104230ba66911c7aNFroot transformtarget_transformdownloadreturnc s t j|||d t jd _|r s tdt jd }dd |D _ W d n1 s9w 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 Y/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torchvision/datasets/sun397.py
. s z#SUN397.__init__..z sun_*.jpgc s. g | ]} j d | jjdd qS )/r ) class_to_idxjoin relative_to _data_dirparts)r pathselfr r r 3 s ")super__init__r r
r _download
_check_exists RuntimeErroropenclassesdictziprangelenr listrglob _image_files_labels)r" r
r r r f __class__r! r r$ s
zSUN397.__init__c C s
t | jS N)r- r0 r! r r r __len__7
zSUN397.__len__idxc C sP | j | | j| }}tj|d}| jr | |}| jr$| |}||fS )NRGB)r0 r1 PILImager( convertr r )r" r8
image_filelabelimager r r __getitem__: s
zSUN397.__getitem__c C s
| j S r5 )r is_dirr! r r r r& F r7 zSUN397._check_existsc C s$ | rd S t| j| j| jd d S )N)
download_rootmd5)r& r _DATASET_URLr
_DATASET_MD5r! r r r r% I s zSUN397._download)NNF)r N)__name__
__module__ __qualname____doc__rD rE r strr r r boolr$ intr6 r r r@ r& r%
__classcell__r r r3 r r
s,
r )pathlibr typingr r r r r PIL.Imager: utilsr visionr
r r r r r s