kho f d dl mZ d dlmZmZmZmZmZ ddlm Z ddl
m Z ddl m
Z
G d de
Zy )
)Path)AnyCallableOptionalTupleUnion )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
W/var/www/teggl/fontify/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 r4 __len__zSUN397.__len__< s 4$$%% idxc | j | | j | }}| j | }| j r| j | }| j r| j | }||fS r6 )r* r. r r r )r/ r:
image_filelabelimages r4 __getitem__zSUN397.__getitem__? se - -c 2DLL4EE
J' >>NN5)E ))%0Ee|r9 c 6 | j j S r6 )r is_dirr7 s r4 r zSUN397._check_existsK s ~~$$&&r9 c ~ | j ry t | j | j | j y )N)
download_rootmd5)r r _DATASET_URLr _DATASET_MD5r7 s r4 r zSUN397._downloadN s. $T%6%6diiUYUfUfgr9 )r N)__name__
__module__ __qualname____doc__rE rF r
r strr r r boolr r intr8 r r? r r
__classcell__)r3 s @r4 r r
s $ QL5L
)-/3 4B
CI H% #8,
%T *+S01
8& &
s
uS#X
't 'hr9 r N)pathlibr typingr r r r r folderr
utilsr visionr r r9 r4 rU s' 8 8 " / !Gh] Ghr9