o
)i @ s` d dl Z d dlmZmZmZmZ d dlmZ ddlm Z m
Z
m Z ddl m
Z
G dd de
ZdS ) N)AnyCallableOptionalTuple)Image )check_integrity download_and_extract_archive download_url)
VisionDatasetc s e Zd ZdZdZdZdZ ddedee d ee d
e
d df
fd d
Z de d e
eef fddZd e fddZd e
fddZdddZ ZS )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.gzZ 9aec147b3488753cf758b4d493422285NTroot transformtarget_transformdownloadreturnc
s t j|||d |r| | stdg | _g | _tj | j
dd}tj | j
dd}t t |t |D ]-\}}|
} tj| }
tj | j
d|
} tj| re|
} | j|
| j| q8d 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 _check_integrity RuntimeErrorphotoscaptionsospathjoinr
zipopenrstripbasenameexistsappend)
selfr
r r r Zfile1Zfile2line1line2urlZphotofilenamecaption __class__ g/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/datasets/sbu.pyr s& z SBU.__init__indexc C sb t j| jd| j| }t|d}| jdur | |}| j | }| j
dur-|
|}||fS )z
Args:
index (int): Index
Returns:
tuple: (image, target) where target is a caption for the photo.
r RGBN) r r r r
r r r convertr r r )r# r- r' Zimgtargetr+ r+ r, __getitem__< s
zSBU.__getitem__c C s
t | jS )z$The number of photos in the dataset.)lenr )r# r+ r+ r, __len__O s
z SBU.__len__c C s* | j }tj|| j}t|| jsdS dS )z1Check the md5 checksum of the downloaded tarball.FT)r
r r r r' r md5_checksum)r# r
Zfpathr+ r+ r, r S s
zSBU._check_integrityc
C s | r
td dS t| j| j| j| j| j ttj
| jdd(}|D ] }| }z
t |tj
| jd W q% t
yA Y q%w W d dS 1 sMw Y dS )zEDownload and extract the tarball, and download each individual photo.z%Files already downloaded and verifiedNr r )r printr r& r
r' r4 r r r r r r
OSError)r# fhliner& r+ r+ r, r [ s "z SBU.download)NNT)r N)__name__
__module__ __qualname____doc__r& r' r4 strr r boolr intr r r1 r3 r r
__classcell__r+ r+ r) r, r
s. r )r typingr r r r ZPILr utilsr r r
Zvisionr r r+ r+ r+ r, s