a
PSic
@ sX d dl mZ d dlmZmZmZmZ d dlZddl m
Z
ddl m Z G dd de Z
dS ) )Path)AnyTupleCallableOptionalN ) download_and_extract_archive)
VisionDatasetc s e Zd ZdZdZdZdeee ee e dd fdd
Z
e d d
d Z e
eef d d d
Ze d ddZdd ddZ 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 (string): Root directory of the dataset.
transform (callable, optional): A function/transform that takes in an 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.gzZ 8ca2778205c41d23104230ba66911c7aNF)root transformtarget_transformdownloadreturnc 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 W/var/www/html/django/DPS/env/lib/python3.9/site-packages/torchvision/datasets/sun397.py
. 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__)r c C s
t | jS N)r, r/ r r r r __len__7 s zSUN397.__len__c C sP | j | | j| }}tj|d}| jr8| |}| jrH| |}||fS )NRGB)r/ r0 PILImager' convertr r
)r! idx
image_filelabelimager r r __getitem__: s
zSUN397.__getitem__c C s
| j S r4 )r is_dirr r r r r% F s zSUN397._check_existsc C s$ | r d 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)__name__
__module__ __qualname____doc__rB rC strr r boolr# intr5 r r r> r% r$
__classcell__r r r2 r r
s" r
)pathlibr typingr r r r PIL.Imager7 utilsr visionr r
r r r r s