YhddlZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZGd d eZGd d eZGd deZdS)N) defaultdict) HTMLParser)Path)AnyCallableOptionalUnion)default_loader) 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 m/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/flickr.pyrzFlickr8kParser.__init__sI  24 *.*.tagattrsc2||_|dkr d|_dSdS)NtableTrr)rrrs rhandle_starttagzFlickr8kParser.handle_starttags% '>> DMMM >rc2d|_|dkr d|_dSdS)NrFr )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 strrrlisttuplerr!r#r2 __classcell__rs@rrr sLL /U39- /$ / / / / / /!3!tE#x}:L4M/N!SW!!!! """""" > > > > > > > > > >rrc eZdZdZddefdeeefdedee dee de ege fddf fd Z d e de e e ffd Zde 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 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. 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. Nrann_file transformtarget_transformloaderrct|||tj||_t |j}t|j5}| | dddn #1swxYwY|j |_ tt|j |_||_dS)Nr?r@)rrr+r, expanduserr>rropenfeedreadrr8sortedkeysidsrA) rrr>r?r@rAparserfhrs rrzFlickr8k.__init__Fs EUVVV**844   ** $-  #B KK " " " # # # # # # # # # # # # # # #!-t/44667788 s2(B&&B*-B*indexc|j|}||}|j||}|j|}|j||}||fSz Args: index (int): Index Returns: tuple: Tuple (image, target). target is a list of captions for the image. )rJrAr?rr@)rrMr1imgtargets r __getitem__zFlickr8k.__getitem__Zsp%kk&!! > %..%%C!&)  ,**622FF{rc*t|jSNlenrJrs r__len__zFlickr8k.__len__p48}}r)r3r4r5r6r r r7rrrrrintr9rRrXr:r;s@rr=r=7s  $)-/3'5 CIH%  #8,  #$  (sCx,rr=c eZdZdZddefdededeedeedeegefddf fd Z d e de eeffd Z de fd Z xZS) Flickr30ka8`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 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. 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. Nrr>r?r@rArcPt|||tj||_t t|_t|j5}|D]T}| d\}} |j|dd | U dddn #1swxYwYtt|j|_||_dS)NrC r()rrr+r,rDr>rr8rrEr0r*r/rHrIrJrA) rrr>r?r@rArLliner1captionrs rrzFlickr30k.__init__s= EUVVV**844 't,, $-  >B > >"&**,,"4"4T":": -44W==== > > > > > > > > > > > > > > > > t/44667788 s7ACC #C rMc"|j|}tj|j|}||}|j||}|j|}|j||}||fSrO) rJr+r,r-rrAr?rr@)rrMr1filenamerPrQs rrRzFlickr30k.__getitem__s%7<< 622kk(## > %..%%C!&)  ,**622FF{rc*t|jSrTrUrWs rrXzFlickr30k.__len__rYr)r3r4r5r6r r7rrrrrZr9rRrXr:r;s@rr\r\ts  $)-/3'5 H%  #8,  #$  *sCx.rr\)r.r+ collectionsr html.parserrpathlibrtypingrrrr folderr visionr rr=r\rrrks ######""""""111111111111""""""!!!!!!(>(>(>(>(>Z(>(>(>V:::::}:::z<<<<< <<<<<r