Yhh j d dl mZ d dlmZmZmZmZ ddlmZ ddl m
Z
ddl m Z G d de Z
d S )
)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 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 m/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/sun397.py
z#SUN397.__init__..2 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__..7 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_labelsr )r' r r r r r f __class__s ` r r) zSUN397.__init__ s| 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)r2 r5 r' s r __len__zSUN397.__len__< s 4$%%%r idxc | j | | j | }}| | }| j r| | }| j r| | }||fS r: )r5 r6 r r r )r' r=
image_filelabelimages r __getitem__zSUN397.__getitem__? sn -c 2DL4EE
J'' > *NN5))E 1))%00Ee|r c 4 | j S r: )r$ is_dirr; s r r+ zSUN397._check_existsK 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._downloadN s? F$T%6diUYUfggggggr )r N)__name__
__module__ __qualname____doc__rH rI r r strr r r boolr r) intr< tuplerB r+ r*
__classcell__)r8 s @r r
r
sF $ QL5L
)-/3 4B
CI H% #8,
%T *+S01
8& & & & &
s
uS#X
't ' ' ' 'h h h h h h h hr r
N)pathlibr typingr r r r folderr utilsr
visionr r
r r rY s 1 1 1 1 1 1 1 1 1 1 1 1 " " " " " " / / / / / / ! ! ! ! ! !Gh Gh Gh Gh Gh] Gh Gh Gh Gh Ghr