o
V ho @ s` d dl mZ d dlmZmZmZmZmZ ddlm Z ddl
m Z ddl m
Z
G dd de
Zd S )
)Path)AnyCallableOptionalTupleUnion )default_loader) download_and_extract_archive)
VisionDatasetc s e Zd ZdZdZdZdddefdeee f de
e de
e d e d
e eee f ge
f d df fd d
Zd efddZded ee
e
f fddZd e fddZdddZ 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.gz 8ca2778205c41d23104230ba66911c7aNFroot transformtarget_transformdownloadloaderreturnc 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 O/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/datasets/sun397.py
2 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 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__c C s
t | jS N)r/ r2 r# r r r __len__<
zSUN397.__len__idxc C sH | j | | j| }}| |}| jr| |}| jr | |}||fS r7 )r2 r3 r r r )r$ r:
image_filelabelimager r r __getitem__? s
zSUN397.__getitem__c C s
| j S r7 )r is_dirr# r r r r( K r9 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' N s zSUN397._download)r N)__name__
__module__ __qualname____doc__rB rC r r strr r r boolr r& intr8 r r> r( r'
__classcell__r r r5 r r
s2
r N)pathlibr typingr r r r r folderr utilsr
visionr r r r r r s