,}h
b d dl mZ d dlmZmZmZmZmZ d dlZ ddl
m Z ddl m
Z
G d de
Zy) )Path)AnyCallableOptionalTupleUnionN ) download_and_extract_archive)
VisionDatasetc e Zd ZdZdZdZ ddeeef de e
de e
de d df
fd
Z d e
fd Zd e
d eeef fd
Zd 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 8ca2778205c41d23104230ba66911c7aNroot transformtarget_transformdownloadreturnc
t | ||| t | j dz | _ |r| j | j
s t d t | j dz 5 }|D cg c] }|dd j c}| _
d d d t t | j t t | j | _ t! | j j# d | _ | j$ D cg c]F }| j dj' |j) | j j* dd H c}| _ y c c}w # 1 sw Y xY wc c}w )
N)r r r
z;Dataset not found. You can use download=True to download itz
ClassName.txt z sun_*.jpg/r )super__init__r r _data_dir _download
_check_exists RuntimeErroropenstripclassesdictziprangelen class_to_idxlistrglob _image_filesjoin relative_toparts_labels) selfr r r r fcpath __class__s Z/var/www/html/test/engine/venv/lib/python3.12/site-packages/torchvision/datasets/sun397.pyr zSUN397.__init__ s8 EUV dii 83 NN !!# \] ]
$..?2
3 6q345aAabEKKM5DL 6 !T\\5T\\9J3K!L M !5!5k!B C cgbsbs
Z^D chht'7'7'G'M'MaPR'S T U
6 6 6
s% 6E5;E0E5 A F0E55E>c , t | j S N)r$ r( r- s r2 __len__zSUN397.__len__7 s 4$$%% idxc | j | | j | }}t j j | j d }| j r| j
| }| j r| j | }||fS )NRGB)r( r, PILImager convertr r )r- r8
image_filelabelimages r2 __getitem__zSUN397.__getitem__: st - -c 2DLL4EE
z*2259 >>NN5)E ))%0Ee|r7 c 6 | j j S r4 )r is_dirr5 s r2 r zSUN397._check_existsF s ~~$$&&r7 c ~ | j ry t | j | j | j y )N)
download_rootmd5)r r
_DATASET_URLr _DATASET_MD5r5 s r2 r zSUN397._downloadI s. $T%6%6diiUYUfUfgr7 )NNF)r N)__name__
__module__ __qualname____doc__rG rH r strr r r boolr intr6 r r rA r r
__classcell__)r1 s @r2 r
r
s
QL5L
)-/3
CI
H%
#8,
4& &
s
uS#X
't 'hr7 r
)pathlibr typingr r r r r PIL.Imager; utilsr
visionr r
r7 r2 rW s' 8 8 / !Bh] Bhr7