o
)i
@ 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 ZdZdZdZ ddedee dee d e d
df
fd d
Z
d
e fd
dZ de d
e
eef 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 (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 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 j/var/www/html/eduruby.in/lip-sync/lip-sync-env/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+ r. r r r r __len__7
zSUN397.__len__idxc C sP | j | | j| }}tj|d}| jr | |}| jr$| |}||fS )NRGB)r. r/ PILZImager& convertr r
)r r6 Z
image_filelabelimager r r __getitem__: s
zSUN397.__getitem__c C s
| j S r3 )r is_dirr r r r r$ F r5 zSUN397._check_existsc C s$ | rd 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)r N)__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.Imager8 utilsr Zvisionr r
r r r r s