jddlZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z m Z mZmZddlmZddlmZGd d eZGd d eZGd deZdS)N) defaultdict) HTMLParser)Path)AnyCallableDictListOptionalTupleUnion)Image) VisionDatasetceZdZdZdeeefddffd Zdedee ee efddfdZ deddfd Z d eddfd Z xZS) Flickr8kParserzBParser for extracting captions from the Flickr8k dataset web page.rootreturnNct||_i|_d|_d|_d|_dS)NF)super__init__r annotationsin_table current_tag current_img)selfr __class__s LC:\Projects\Historical\venv\Lib\site-packages\torchvision/datasets/flickr.pyrzFlickr8kParser.__init__sI  24 *.*.tagattrsc2||_|dkr d|_dSdS)NtableTrr)rrr s rhandle_starttagzFlickr8kParser.handle_starttags% '>> DMMM >rc2d|_|dkr d|_dSdS)Nr"Fr#)rrs r handle_endtagzFlickr8kParser.handle_endtag#s% '>>!DMMM >rdatac|jr|dkr d|_dS|jdkrp|dd}tj|j|dz}tj|d}||_g|j |<dS|jdkrD|jr?|j}|j | | dSdSdSdS)NzImage Not Founda/z_*.jpgrli) rrrsplitospathjoinrglobrappendstrip)rr'img_ids r handle_datazFlickr8kParser.handle_data)s = >(((#'   !S((C,di(1BCC6**1-#) +- (((!T))d.>)) (// ===== > >*)))r)__name__ __module__ __qualname____doc__r strrrr r r r$r&r5 __classcell__rs@rrr sLL /U39- /$ / / / / / /!3!tE#x}:L4M/N!SW!!!! """""" > > > > > > > > > >rrc eZdZdZ d deeefdedeedeeddf fd Z d e de e e ffd Z de fd ZxZS) Flickr8ka`Flickr8k Entities `_ Dataset. Args: root (str or ``pathlib.Path``): 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_transformrct|||tj||_t |j}t|j5}| | dddn #1swxYwY|j |_ tt|j |_dS)Nr@rA)rrr.r/ expanduserr?rropenfeedreadrlistsortedkeysids)rrr?r@rAparserfhrs rrzFlickr8k.__init__Ds EUVVV**844   ** $-  #B KK " " " # # # # # # # # # # # # # # #!-t/44667788s2(B&&B*-B*indexc|j|}tj|d}|j||}|j|}|j||}||fSz Args: index (int): Index Returns: tuple: Tuple (image, target). target is a list of captions for the image. RGB)rKr rEconvertr@rrA)rrNr4imgtargets r __getitem__zFlickr8k.__getitem__Vs|%j  ((// > %..%%C!&)  ,**622FF{rc*t|jSNlenrKrs r__len__zFlickr8k.__len__l48}}rNN)r6r7r8r9r r:rr rrintr rrUr[r;r<s@rr>r>8s  )-/3 99CI99H% 9 #8, 9  999999$sCx,rr>c eZdZdZ d dededeedeeddf fd Zd ede e e ffd Z defd Z xZ S) Flickr30ka `Flickr30k Entities `_ Dataset. Args: root (str or ``pathlib.Path``): 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. Nrr?r@rArcBt|||tj||_t t|_t|j5}|D]T}| d\}}|j|dd |U dddn #1swxYwYtt|j|_dS)NrC r+)rrr.r/rDr?rrHrrEr3r-r2rIrJrK) rrr?r@rArMliner4captionrs rrzFlickr30k.__init__|s6 EUVVV**844 't,, $-  >B > >"&**,,"4"4T":": -44W==== > > > > > > > > > > > > > > > > t/44667788s7ACC #C rNcF|j|}tj|j|}t j|d}|j||}|j |}|j | |}||fSrP) rKr.r/r0rr rErRr@rrA)rrNr4filenamerSrTs rrUzFlickr30k.__getitem__s%7<< 622j""**511 > %..%%C!&)  ,**622FF{rc*t|jSrWrXrZs rr[zFlickr30k.__len__r\rr])r6r7r8r9r:r rrr^r rrUr[r;r<s@rr`r`ps  )-/3 9999H% 9 #8, 9  999999&sCx.rr`)r1r. collectionsr html.parserrpathlibrtypingrrrr r r r PILr visionrrr>r`rrros1 ######""""""DDDDDDDDDDDDDDDDDD!!!!!!(>(>(>(>(>Z(>(>(>V55555}555p77777 77777r