QЦi v S SK r S SKJr S SKJrJrJrJrJr S SK J
r
SSK J r J
r
Jr SSKJr " S S \5 rg)
N)Path)AnyCallableOptionalTupleUnion)Image )check_integrity download_and_extract_archive download_url)
VisionDatasetc ^ \ rS rSrSrSrSrSr SS\\ \
4 S\ \ S \ \ S
\
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
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.
zHhttps://www.cs.rice.edu/~vo9/sbucaptions/SBUCaptionedPhotoDataset.tar.gzz SBUCaptionedPhotoDataset.tar.gz 9aec147b3488753cf758b4d493422285Nroot transformtarget_transformdownloadreturnc 2 > [ T
U ] XUS9 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 pxUR 5 n [ R R U 5 n
[ R R U R SU
5 n [ R R U 5 ( d M UR 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 _check_integrity RuntimeErrorphotoscaptionsospathjoinr zipopenrstripbasenameexistsappend)selfr r r r file1file2line1line2urlphotofilenamecaption __class__s W/var/www/html/ai-image-ml/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 X [ R R U R SU R U 5 n[
R " U5 R S5 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 RGB) r! r" r# r r r r% convertr r r )r* r7 r1 imgtargets r4 __getitem__SBU.__getitem__= s 77<< 9dkk%6HIjj"**51 >> %..%C u% ,**62F{r6 c , [ U R 5 $ )z$The number of photos in the dataset.)lenr )r* s r4 __len__ SBU.__len__P 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_integrityT 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 rD r% r! r" r# r& r
OSError)r* fhliner/ s r4 r SBU.download\ s " " $TXXtyy$))T]]TXTeTef "'',,tyy)5[\
]ackkm bggll499i&HI ^
]
^
]s0 C5#4C$C5$
C2.C51C22C55
D)r r )NNT)r N)__name__
__module__ __qualname____firstlineno____doc__r/ r1 rD r strr r r boolr intr r r= rA r r __static_attributes__
__classcell__)r3 s @r4 r r s UC0H5L
)-/3 .CI . H% . #8, .
.
. .> sCx & $ r6 r )r! pathlibr typingr r r r r PILr utilsr r r
visionr r r6 r4 r\ s+ 8 8 N N !b- br6