+
jp t ^ RI t ^ RIHt ^ RIHtHtHtHt ^RIH t ^RI
H t H t H
t
^RIHt ! R R]4 tR# ) N)Path)AnyCallableOptionalUnion)default_loader)check_integrity download_and_extract_archive download_url)
VisionDatasetc a a ] tR t^ t oRtRtRtRtRRR]3V3R lV 3R lllt V3R lR
lt
V3R lR lt V3R
lR lt V3R lR lt
RtVtV ;t# )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 9aec147b3488753cf758b4d493422285NTc < V ^8 d QhRS[ S[S[3, RS[S[, RS[S[, RS[RS[S[.S[3, RR/# ) root transformtarget_transformdownloadloaderreturnN)r strr r r boolr )format
__classdict__s "Y/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchvision/datasets/sbu.py __annotate__SBU.__annotate__! se . .CI . H% . #8, .
. #$
.
. c L < \ SV ` WVR 7 WPn V' d V P 4 V P 4 ' g \ R4 h. V n . V n \ P P V P RR4 p\ P P V P RR4 p\ \ V4 \ V4 4 F w rVP 4 p
\ P P V
4 p \ P P V P RV 4 p \ P P! V 4 ' g K V P 4 p
V P P# V 4 V P P# V
4 K R# ))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.txtN)super__init__r r _check_integrity RuntimeErrorphotoscaptionsospathjoinr zipopenrstripbasenameexistsappend)selfr r r r r file1file2line1line2urlphotofilenamecaption __class__s &&&&&& r r$ SBU.__init__! s EUV MMO$$&& ij j
TYY 3YZ TYY 3]^ U T%[ 9LE,,.CGG$$S)Eww||DIIy%@Hww~~h'' ,,. ""5)
$$W- :r c < < V ^8 d QhRS[ RS[S[S[3, /# )r indexr )inttupler )r r s "r r r B s# sCx r c D \ P P V P RV P V, 4 pV P V4 pV P e V P
V4 pV P V, pV P e V P V4 pW43# )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 )r2 r> r9 imgtargets && r __getitem__SBU.__getitem__B s 77<< 9dkk%6HIkk(# >> %..%C u% ,**62F{r c < V ^8 d QhRS[ /# r r )r? )r r s "r r r U s r c , \ V P 4 # )z$The number of photos in the dataset.)lenr' )r2 s &r __len__ SBU.__len__U s 4;; r c < V ^8 d QhRS[ /# rG )r )r r s "r r r Y s $ r c V P p\ P P WP 4 p\ W P 4 ' g R# R# )z1Check the md5 checksum of the downloaded tarball.FT)r r) r* r+ r9 r md5_checksum)r2 r fpaths & r r% SBU._check_integrityY s8 yy T==1 u&7&788r c < V ^8 d QhRR/# )r r N )r r s "r r r a s $ r c
* V P 4 ' d R# \ V P V P V P V P V P
4 \
\ P P V P RR4 4 ;_uu_ 4 pV FH pVP 4 p \ V\ P P V P R4 4 KJ RRR4 R# \ d Kd i ; i + ' g i R# ; i)zEDownload and extract the tarball, and download each individual photo.Nr! r" )
r% r
r7 r r9 rN r- r) r* r+ r. r OSError)r2 fhliner7 s & r r SBU.downloada s " " $TXXtyy$))T]]TXTeTef "'',,tyy)5[\
]
]ackkm bggll499i&HI ^
]
^
]
]s0 D-4C/!D/ C> :D=C> >D D )r( r r' )__name__
__module__ __qualname____firstlineno____doc__r7 r9 rN r r$ rD rJ r% r __static_attributes____classdictcell__
__classcell__)r; r s @@r r r sb " UC0H5L
)-/3 '5
. .B & r r )r) pathlibr typingr r r r folderr utilsr r
r visionr r rR r r re s+ 1 1 " N N !g- gr