o )i@sddlZddlZddlmZddlmZddlmZmZm Z m Z m Z m Z ddl mZddlmZGdd d eZGd d d eZGd d d eZdS)N) defaultdict) HTMLParser)AnyCallableDictListOptionalTuple)Image) VisionDatasetcsteZdZdZdeddffdd Zdedeeeeefddfd d Z deddfd d Z d eddfddZ Z S)Flickr8kParserzBParser for extracting captions from the Flickr8k dataset web page.rootreturnNcs,t||_i|_d|_d|_d|_dS)NF)super__init__r annotationsin_table current_tag current_img)selfr __class__j/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/datasets/flickr.pyrs  zFlickr8kParser.__init__tagattrscCs||_|dkr d|_dSdS)NtableTrr)rrrrrrhandle_starttag zFlickr8kParser.handle_starttagcCsd|_|dkr d|_dSdS)NrFr)rrrrr handle_endtag"r zFlickr8kParser.handle_endtagdatacCs|jrJ|dkr d|_dS|jdkr3|dd}tj|j|d}t|d}||_g|j |<dS|jdkrL|jrN|j}|j | | dSdSdSdS)NzImage Not Founda/z_*.jpgrZli) rrrsplitospathjoinrglobrappendstrip)rr"img_idrrr handle_data(s   zFlickr8kParser.handle_data) __name__ __module__ __qualname____doc__strrrr rrr!r. __classcell__rrrrr s & r c leZdZdZ  ddededeedeeddf fdd Zd ede e e ffd d Z defd dZ Z S)Flickr8ka`Flickr8k Entities `_ Dataset. Args: root (string): Root directory where images are downloaded to. ann_file (string): Path to annotation file. transform (callable, optional): A function/transform that takes in a PIL image and returns a transformed version. E.g, ``transforms.PILToTensor`` target_transform (callable, optional): A function/transform that takes in the target and transforms it. Nrann_file transformtarget_transformrcstj|||dtj||_t|j}t|j}| | Wdn1s,wY|j |_ t t |j |_dS)Nr8r9)rrr'r( expanduserr7r ropenfeedreadrlistsortedkeysids)rrr7r8r9parserfhrrrrCs  zFlickr8k.__init__indexcCsT|j|}t|d}|jdur||}|j|}|jdur&||}||fSz Args: index (int): Index Returns: tuple: Tuple (image, target). target is a list of captions for the image. RGBN)rBr r<convertr8rr9)rrEr-imgtargetrrr __getitem__Us      zFlickr8k.__getitem__cC t|jSNlenrBrrrr__len__k zFlickr8k.__len__NNr/r0r1r2r3rrrintr rrKrQr4rrrrr67s"r6c r5) Flickr30ka`Flickr30k Entities `_ Dataset. Args: root (string): Root directory where images are downloaded to. ann_file (string): Path to annotation file. transform (callable, optional): A function/transform that takes in a PIL image and returns a transformed version. E.g, ``transforms.PILToTensor`` target_transform (callable, optional): A function/transform that takes in the target and transforms it. Nrr7r8r9rc stj|||dtj||_tt|_t |j"}|D]}| d\}}|j|dd |qWdn1s?wYtt |j|_dS)Nr: r%)rrr'r(r;r7rr?rr<r,r&r+r@rArB) rrr7r8r9rDliner-captionrrrr{s  zFlickr30k.__init__rEcCsd|j|}tj|j|}t|d}|jdur||}|j |}|j dur.| |}||fSrF) rBr'r(r)rr r<rHr8rr9)rrEr-filenamerIrJrrrrKs      zFlickr30k.__getitem__cCrLrMrNrPrrrrQrRzFlickr30k.__len__rSrTrrrrrVos"rV)r*r' collectionsr html.parserrtypingrrrrrr ZPILr Zvisionr r r6rVrrrrs     +8