Eih b S SK Jr S SKJrJrJrJr SSKJr SSK J
r
SSK J r " S S\ 5 r
g )
)Path)AnyCallableOptionalUnion )default_loader) download_and_extract_archive)
VisionDatasetc ^ \ rS rSrSrSrSrSSS\4S\\ \
4 S\ \ S \ \ S
\
S \ \\ \
4 /\4 S S4 U 4S
jjjrS \4S jrS\S \\\4 4S jrS \
4S jrSS jrSrU =r$ )SUN397
a `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
6 > [ T U ] XUS9 [ U R 5 S- U l U( a U R 5 U R
5 ( d [ S5 e[ U R S- 5 nU Vs/ s H owSS R 5 PM snU l
S S S 5 [ [ U R [ [ U R 5 5 5 5 U l [! U R R# S5 5 U l U R$ Vs/ s HF oR SR' UR) U R 5 R* SS 5 PMH snU l XPl g s snf ! , ( d f N= fs snf )
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
X/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torchvision/datasets/sun397.pyr SUN397.__init__ s? EUV dii 83 NN !!## \] ]
$..?2
3q3451aabEKKM15DL 4 !T\\5T\\9J3K!L M !5!5k!B C cgbsbs
bsZ^ chht'7'7'G'M'MaPR'S T Ubs
6 4
3
s% <F F F$A
F F
Fc , [ U R 5 $ N)r& r* r/ s r4 __len__SUN397.__len__< s 4$$%% idxc U R U U R U p2U R U5 nU R ( a U R U5 nU R ( a U R U5 nXC4$ r7 )r* r. r r r )r/ r<
image_filelabelimages r4 __getitem__SUN397.__getitem__? sa - -c 2DLL4EE J' >>NN5)E ))%0E|r; c 6 U R R 5 $ r7 )r is_dirr8 s r4 r SUN397._check_existsK s ~~$$&&r; c U R 5 ( a g [ U R U R U R S9 g )N)
download_rootmd5)r r
_DATASET_URLr _DATASET_MD5r8 s r4 r SUN397._downloadN s1 $T%6%6diiUYUfUfgr; )r r* r. r' r" r )r N)__name__
__module__ __qualname____firstlineno____doc__rI rJ r r strr r r boolr r intr9 tuplerA r r __static_attributes__
__classcell__)r3 s @r4 r
r
s $ QL5L
)-/3 4B
CI H% #8,
%T *+S01
8& &
s
uS#X
't 'h hr; r
N)pathlibr typingr r r r folderr utilsr
visionr r
r; r4 r] s' 1 1 " / !Gh] Ghr;