Dih b d dl mZ d dlmZmZmZmZ ddlmZ ddl m
Z
ddl m Z G d de Z
y )
)Path)AnyCallableOptionalUnion )default_loader) download_and_extract_archive)
VisionDatasetc e Zd ZdZdZdZdddefdeee f de
e de
e d e d
e eee f ge
f d df fd
Zd efd
Zded ee
e
f fdZd e fdZddZ xZS )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
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_labelsr )
selfr r r r r fcpath __class__s
]/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/torchvision/datasets/sun397.pyr zSUN397.__init__ sA 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% 6E<;E7E< A F7E<<Fc , t | j S N)r% r) r. s r3 __len__zSUN397.__len__< s 4$$%% idxc | j | | j | }}| j | }| j r| j | }| j r| j | }||fS r5 )r) r- r r r )r. r9
image_filelabelimages r3 __getitem__zSUN397.__getitem__? se - -c 2DLL4EE
J' >>NN5)E ))%0Ee|r8 c 6 | j j S r5 )r is_dirr6 s r3 r zSUN397._check_existsK s ~~$$&&r8 c ~ | j ry t | j | j | j y )N)
download_rootmd5)r r
_DATASET_URLr _DATASET_MD5r6 s r3 r zSUN397._downloadN s. $T%6%6diiUYUfUfgr8 )r N)__name__
__module__ __qualname____doc__rD rE r r strr r r boolr r intr7 tupler> r r
__classcell__)r2 s @r3 r
r
s $ QL5L
)-/3 4B
CI H% #8,
%T *+S01
8& &
s
uS#X
't 'hr8 r
N)pathlibr typingr r r r folderr utilsr
visionr r
r8 r3 rU s' 1 1 " / !Gh] Ghr8