Eip r S SK r S SKJr S SKJrJrJrJr SSKJ r SSK
J r J r J
r
SSKJr " S S \5 rg)
N)Path)AnyCallableOptionalUnion )default_loader)check_integrity download_and_extract_archive download_url)
VisionDatasetc
^ \ rS rSrSrSrSrSrSSS\4S\ \
\ 4 S \ \
S
\ \
S \S \
\
/\4 S
S4 U 4S jjjrS\S
\\\4 4S jrS
\4S jrS
\4S jrSS jrSrU =r$ )SBU a `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 > > [ TU ] XUS9 XPl U( a U R 5 U R 5 ( d [ S5 e/ U l / U l [ R R U R SS5 n[ R R U R SS5 n[ [ U5 [ U5 5 H u pUR 5 n
[ R R U
5 n [ R R U R SU 5 n [ R R! U 5 ( d M U R 5 n
U R R# U 5 U R R# U
5 M g )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 U/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torchvision/datasets/sbu.pyr SBU.__init__! s EUV MMO$$&& ij j
TYY 3YZ TYY 3]^ U T%[ 9LE,,.CGG$$S)Eww||DIIy%@Hww~~h'' ,,. ""5)
$$W- : indexc 0 [ R R U R SU R U 5 nU R U5 nU R b U R
U5 nU R U nU R b U R U5 nX44$ )zl
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* r7 r1 imgtargets r4 __getitem__SBU.__getitem__B s 77<< 9dkk%6HIkk(# >> %..%C u% ,**62F{r6 c , [ U R 5 $ )z$The number of photos in the dataset.)lenr )r* s r4 __len__ SBU.__len__U s 4;; r6 c U R n[ R R XR 5 n[ X R 5 ( d gg)z1Check the md5 checksum of the downloaded tarball.FT)r r! r" r# r1 r
md5_checksum)r* r fpaths r4 r SBU._check_integrityY s8 yy T==1 u&7&788r6 c U R 5 ( a g[ U R U R U R U R U R
5 [
[ R R U R SS5 5 nU HH nUR 5 n [ U[ R R U R S5 5 MJ SSS5 g! [ a Mb f = f! , ( d f g= f)zEDownload and extract the tarball, and download each individual photo.Nr r )
r r r/ r r1 rB r% r! r" r# r& r OSError)r* fhliner/ s r4 r SBU.downloada s " " $TXXtyy$))T]]TXTeTef "'',,tyy)5[\
]ackkm bggll499i&HI ^
]
^
]s0 C5#4C$C5$
C2.C51C22C55
D)r r r )r N)__name__
__module__ __qualname____firstlineno____doc__r/ r1 rB r r strr r r boolr r inttupler; r? r r __static_attributes__
__classcell__)r3 s @r4 r r s " UC0H5L
)-/3 '5
.CI . H% . #8, .
. #$
.
. .B sCx & $ r6 r )r! pathlibr typingr r r r folderr utilsr
r r visionr
r r6 r4 r[ s+ 1 1 " N N !g- gr6