ڧg
j d dl mZ d dlmZmZmZmZmZ d dlZ ddl
m Z ddl m
Z
G d de
ZdS ) )Path)AnyCallableOptionalTupleUnionN ) download_and_extract_archive)
VisionDatasetc e Zd ZdZdZdZ ddeeef de e
de e
d e d
df
fd
Z d
e
fd Zd
e
d
eeef fdZd
e fdZddZ xZS )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
t ||| t j dz _ |r st d t j dz 5 }d |D _ d d d n # 1 swxY w Y t t j t t j _ t j d _ fd j D _ d S )N)r r r
z;Dataset not found. You can use download=True to download itz
ClassName.txtc F g | ] }|d d S ) N)strip).0cs W/var/www/html/ai-engine/env/lib/python3.11/site-packages/torchvision/datasets/sun397.py
z#SUN397.__init__... s( 555aAabbEKKMM555 z sun_*.jpgc g | ]G}j d | j j dd HS )/r ) class_to_idxjoin relative_to _data_dirparts)r pathselfs r r z#SUN397.__init__..3 sW
Z^D chht'7'7'G'G'MaPRd'S T T U
r )super__init__r r r# _download
_check_exists RuntimeErroropenclassesdictziprangelenr listrglob _image_files_labels)r& r r r r f __class__s ` r r( zSUN397.__init__ ss EUVVV di 83 NN !!## ^ \]] ]
$.?2
3
3 6q551555DL 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 !T\5T\9J9J3K3K!L!L M M !5!5k!B!B C C
bfbs
s B11B58B5c * t | j S N)r1 r4 r& s r __len__zSUN397.__len__7 s 4$%%%r idxc | j | | j | }}t j | d }| j r| | }| j r| | }||fS )NRGB)r4 r5 PILImager, convertr r )r& r<
image_filelabelimages r __getitem__zSUN397.__getitem__: s~ -c 2DL4EE
z**22599 > *NN5))E 1))%00Ee|r c 4 | j S r9 )r# is_dirr: s r r* zSUN397._check_existsF s ~$$&&&r c t | rd S t | j | j | j d S )N)
download_rootmd5)r* r
_DATASET_URLr _DATASET_MD5r: s r r) zSUN397._downloadI s? F$T%6diUYUfggggggr )NNF)r N)__name__
__module__ __qualname____doc__rK rL r strr r r boolr( intr; r r rE r* r)
__classcell__)r7 s @r r
r
s#
QL5L
)-/3
CI
H%
#8,
4& & & & &
s
uS#X
't ' ' ' 'h h h h h h h hr r
)pathlibr typingr r r r r PIL.Imager? utilsr
visionr r
r r r[ s 8 8 8 8 8 8 8 8 8 8 8 8 8 8 / / / / / / ! ! ! ! ! !Bh Bh Bh Bh Bh] Bh Bh Bh Bh Bhr