Ph ^ d dl mZ d dlmZmZmZmZ d dlZddl m
Z
ddl m Z G d de Z
y) )Path)AnyCallableOptionalTupleN ) download_and_extract_archive)
VisionDatasetc e Zd ZdZdZdZ ddedee dee de d df
fd
Z
d e fd Z d e d e
eef 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 (string): Root directory of the dataset.
transform (callable, optional): A function/transform that takes in an 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 fC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/datasets/sun397.pyr zSUN397.__init__ s; EUV dii 83 NN !!# \] ]
$..?2
3q3451aAabEKKM15DL 4 !T\\5T\\9J3K!L M !5!5k!B C cgbsbs
bsZ^D chht'7'7'G'M'MaPR'S T Ubs
6 4
3
s% 6E5;E0E5 A F0E55E>c , t | j S N)r# r' r, s r1 __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, r7
image_filelabelimages r1 __getitem__zSUN397.__getitem__: st - -c 2DLL4EE
z*2259 >>NN5)E ))%0Ee|r6 c 6 | j j S r3 )r is_dirr4 s r1 r zSUN397._check_existsF s ~~$$&&r6 c ~ | j ry t | j | j | j y )N)
download_rootmd5)r r _DATASET_URLr _DATASET_MD5r4 s r1 r zSUN397._downloadI s. $T%6%6diiUYUfUfgr6 )NNF)r N)__name__
__module__ __qualname____doc__rF rG strr r boolr intr5 r r r@ r r
__classcell__)r0 s @r1 r r
s
QL5L
)-/3
H%
#8,
4& &
s
uS#X
't 'hr6 r )pathlibr typingr r r r PIL.Imager: utilsr visionr
r r6 r1 rV s' 1 1 / !Bh] Bhr6