U
ʹj
@ sX d dl mZ d dlmZmZmZmZ d dlZddl m
Z
ddl m Z G dd de Z
dS ) )Path)AnyCallableOptionalTupleN ) download_and_extract_archive)
VisionDatasetc s e Zd ZdZdZdZdeee ee e dd fdd
Z
e d d
d Z e e
eef d d
dZe 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 5 Q R X 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 Z/var/www/html/bwave/python/venv/lib/python3.8/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__)r c C s
t | jS N)r+ r. r r r r __len__7 s zSUN397.__len__)idxr c C sP | j | | j| }}tj|d}| jr8| |}| jrH| |}||fS )NRGB)r. r/ PILZImager& convertr r
)r r5 Z
image_filelabelimager r r __getitem__: s
zSUN397.__getitem__c C s
| j S r3 )r is_dirr r r r r$ F 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# I s zSUN397._download)NNF)__name__
__module__ __qualname____doc__r> r? strr r boolr" intr4 r r r; r$ r#
__classcell__r r r1 r r
s" r
)pathlibr typingr r r r Z PIL.Imager7 utilsr Zvisionr r
r r r r s