Yhp z d dl Z d dlmZ d dlmZmZmZmZ ddlm Z ddl
m Z m Z m
Z
ddlmZ G d d e ZdS )
N)Path)AnyCallableOptionalUnion )default_loader)check_integrity download_and_extract_archive download_url)
VisionDatasetc
e Zd ZdZdZdZdZdddefdee e
f de e d e e d
e
d e e gef d df fd
Zded eeef fdZd efdZd e
fdZddZ xZS )SBUa `SBU Captioned Photo `_ Dataset.
Args:
root (str or ``pathlib.Path``): Root directory of dataset where tarball
``SBUCaptionedPhotoDataset.tar.gz`` exists.
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.
zHhttps://www.cs.rice.edu/~vo9/sbucaptions/SBUCaptionedPhotoDataset.tar.gzz SBUCaptionedPhotoDataset.tar.gz 9aec147b3488753cf758b4d493422285NTroot transformtarget_transformdownloadloaderreturnc d t ||| || _ |r| | st d g | _ g | _ t j
| j dd }t j
| j dd }t t | t | D ]\ }} | }
t j |
} t j
| j d| } t j | rH| }
| j | | j |
d S )N)r r zHDataset not found or corrupted. You can use download=True to download itdataset$SBU_captioned_photo_dataset_urls.txtz(SBU_captioned_photo_dataset_captions.txt)super__init__r r _check_integrity RuntimeErrorphotoscaptionsospathjoinr zipopenrstripbasenameexistsappend)selfr r r r r file1file2line1line2urlphotofilenamecaption __class__s j/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/sbu.pyr z SBU.__init__! sg EUVVV MMOOO$$&& k ijj j
TY 3YZZ TY 3]^^ U T%[[ 9 9 . .LE5,,..CG$$S))Ew||DIy%@@Hw~~h''
. ,,.. ""5)))
$$W--- . . indexc t j | j d| j | }| | }| j | | }| j | }| j | | }||fS )z
Args:
index (int): Index
Returns:
tuple: (image, target) where target is a caption for the photo.
r ) r r! r" r r r r r r )r) r5 r0 imgtargets r3 __getitem__zSBU.__getitem__B s 7<< 9dk%6HIIkk(## > %..%%C u% ,**622FF{r4 c * t | j S )z$The number of photos in the dataset.)lenr )r) s r3 __len__z SBU.__len__U s 4; r4 c | j }t j || j }t || j sdS dS )z1Check the md5 checksum of the downloaded tarball.FT)r r r! r" r0 r
md5_checksum)r) r fpaths r3 r zSBU._check_integrityY s? y T4=11 ud&788 5tr4 c | rdS t | j | j | j | j | j t
t j | j dd 5 }|D ]Z}|
} t |t j | j d K# t $ r Y Ww xY w ddd dS # 1 swxY w Y dS )zEDownload and extract the tarball, and download each individual photo.Nr r )
r r r. r r0 r> r$ r r! r" r% r OSError)r) fhliner. s r3 r z SBU.downloada s3 " " F$TXty$)T]TXTefff "',,ty)5[\\
]
] ac
kkmm bgll49i&H&HIIII D
s6 6C"3CC"
CC"CC""C&)C&)r N)__name__
__module__ __qualname____doc__r. r0 r> r r strr r r boolr r inttupler9 r< r r
__classcell__)r2 s @r3 r r s9 " UC0H5L
)-/3 '5
. .CI . H% . #8, .
. #$
.
. . . . . .B sCx & $ r4 r )r pathlibr typingr r r r folderr utilsr
r r visionr
r r4 r3 rS s 1 1 1 1 1 1 1 1 1 1 1 1 " " " " " " N N N N N N N N N N ! ! ! ! ! !g g g g g- g g g g gr4