Ph^ f d dl Z d dlmZmZmZmZ d dlmZ ddlm Z m
Z
m Z ddl m
Z
G d de
Zy) N)AnyCallableOptionalTuple)Image )check_integrity download_and_extract_archive download_url)
VisionDatasetc e Zd ZdZdZdZdZ ddedee dee d e
d
df
fd
Z d e d
e
eef fd
Zd
e fdZd
e
fdZddZ xZS )SBUa `SBU Captioned Photo `_ Dataset.
Args:
root (string): 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 t
| ||| |r| j | j s t d g | _ g | _ t j j | j dd }t j j | j dd }t t | t | D ] \ }}|j } t j j | }
t j j | j d|
} t j j | s|j } | j
j! |
| j j! | y )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 cC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/datasets/sbu.pyr z SBU.__init__ s EUV MMO$$& ij j
TYY 3YZ TYY 3]^ U T%[ 9LE5,,.CGG$$S)Eww||DIIy%@Hww~~h' ,,. ""5)
$$W- : indexc T t j j | j d| j | }t j | j d }| j | j | }| j | }| j | j | }||fS )z
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' r3 r. imgtargets r1 __getitem__zSBU.__getitem__< s 77<< 9dkk%6HIjj"**51 >> %..%C u% ,**62FF{r2 c , t | j S )z$The number of photos in the dataset.)lenr )r' s r1 __len__z SBU.__len__O s 4;; r2 c | j }t j j || j }t || j syy)z1Check the md5 checksum of the downloaded tarball.FT)r r r r r. r md5_checksum)r' r fpaths r1 r zSBU._check_integrityS s9 yy T4==1 ud&7&78r2 c | j r t d yt | j | j | j | j
| j t t j j | j dd 5 }|D ]G }|j } t |t j j | j d I ddd y# t $ r Y _w xY w# 1 sw Y yxY w)zEDownload and extract the tarball, and download each individual photo.z%Files already downloaded and verifiedNr r )r printr
r, r r. r> r" r r r r# r OSError)r' fhliner, s r1 r z SBU.download[ s " 9 : $TXXtyy$))T]]TXTeTef "'',,tyy)5[\
]ackkm bggll499i&HI ^
]
^
]s0 C8)4C) C8) C52C84C55C88D)NNT)r N)__name__
__module__ __qualname____doc__r, r. r> strr r boolr intr r r9 r< r r
__classcell__)r0 s @r1 r r
s UC0H5L
)-/3 . . H% . #8, .
.
.> sCx & $ r2 r )r typingr r r r PILr utilsr r
r visionr r r2 r1 rR s( 1 1 N N !c- cr2