Yhp" ddlZddlZddlmZddlmZmZmZmZddl m Z  ddl m Zn#e$r ddl m ZYnwxYwddlmZddlmZmZdd lmZd d d ejd dddddejdd dddddejd dddddejd dddd dejd ddd d!d"ejd d#dd$d%d&ejd d#dd'ZGd(d)eZGd*d+eZGd,d-eZdS).N)Path)AnyCallableOptionalUnion)Element)parse)Image)download_and_extract_archiveverify_str_arg) VisionDatasetzJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tarzVOCtrainval_11-May-2012.tar 6cd6e144f989b92b3379bac3b3de84fd VOCdevkitVOC2012)urlfilenamemd5base_dirzJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2011/VOCtrainval_25-May-2011.tarzVOCtrainval_25-May-2011.tar 6c3384ef61512963050cb5d687e5bf1eTrainValVOC2011zJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2010/VOCtrainval_03-May-2010.tarzVOCtrainval_03-May-2010.tar da459979d0c395079b5c75ee67908abbVOC2010zJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2009/VOCtrainval_11-May-2009.tarzVOCtrainval_11-May-2009.tar a3e00b113cfcfebf17e343f59da3caa1VOC2009zJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2008/VOCtrainval_14-Jul-2008.tar 2629fa636546599198acfcfbfcf1904aVOC2008zJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tarzVOCtrainval_06-Nov-2007.tar c52e279531787c972589f7e41ab4ae64VOC2007zFhttp://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tarzVOCtest_06-Nov-2007.tar b6e924de25625d8de591ea690078ad9f)201220112010200920082007 2007-testceZdZUeed<eed<eed< ddeeefd ed ed ed ee d ee dee ffd Z de fdZ xZ S)_VOCBase _SPLITS_DIR _TARGET_DIR_TARGET_FILE_EXTr"trainFNrootyear image_setdownload transformtarget_transform transformsc Zt||||t|ddtddD_gd}|dkr|dt|d |_|dkr|dkrd n|} t| } | d _| d _ | d _ | d} tj j| } |r'tjjj j tj | st#dtj | dj} tj | |ddz}t)tj |5}d|D}dddn #1swxYwYtj | dfd|D_tj | jfd|D_t3jt3jksJdS)Nr0c,g|]}t|S)str).0yrs j/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/voc.py z%_VOCBase.__init__..Ps>c>c>c2s2ww>c>c>cii) valid_values)r.trainvalvalr'testr1r(rrrr)rrzHDataset not found or corrupted. You can use download=True to download it ImageSets z.txtc6g|]}|Sr8)strip)r:xs r<r=z%_VOCBase.__init__..js ;;;!''));;;r> JPEGImagescVg|]%}tj|dz&S)z.jpg)ospathjoin)r:rG image_dirs r<r=z%_VOCBase.__init__..ms-OOOqrw||Iq6z::OOOr>c`g|]*}tj|jz+Sr8)rJrKrLr-)r:rGself target_dirs r<r=z%_VOCBase.__init__..ps2```PQ ZT5J1JKK```r>)super__init__r ranger0appendr1DATASET_YEAR_DICTrrrrJrKrLr/r isdir RuntimeErrorr+rstripopen readlinesimagesr,targetslen)rOr/r0r1r2r3r4r5valid_image_setskeydataset_year_dictrvoc_root splits_dirsplit_ff file_namesrMrP __class__s` @@r<rRz_VOCBase.__init__Ds z96FGGG"4>c>cQVW[]aQbQb>c>c>cddd 777 6>>  # #F + + +' ;@PQQ!V^^ V0C0Ckk-c2$U+)*5 $U+$Z07<< 844  d (49t}Z^Zb c c c cw}}X&& kijj jW\\(K9IJJ ',,z9+;+;D+A+AF+JKK "',,w'' ( ( )r"r.FNNN)__name__ __module__ __qualname__r9__annotations__rrboolrrrRintrk __classcell__)rfs@r<r*r*?s  (,/3)-.5.5CI.5.5 .5  .5 H% .5#8,.5X&.5.5.5.5.5.5`         r>r*cfeZdZdZdZdZdZedee fdZ de de e e ffdZd S) VOCSegmentationa2`Pascal VOC `_ Segmentation Dataset. Args: root (str or ``pathlib.Path``): Root directory of the VOC Dataset. year (string, optional): The dataset year, supports years ``"2007"`` to ``"2012"``. image_set (string, optional): Select the image_set to use, ``"train"``, ``"trainval"`` or ``"val"``. If ``year=="2007"``, can also be ``"test"``. 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. 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. transforms (callable, optional): A function/transform that takes input sample and its target as entry and returns a transformed version. SegmentationSegmentationClassz.pngrgc|jSrir\rjs r<maskszVOCSegmentation.masks |r>indexctj|j|d}tj|j|}|j|||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is the image segmentation. RGB)r rYr[convertryr5rOr{imgtargets r< __getitem__zVOCSegmentation.__getitem__sgjU+,,44U;;DJu-.. ? &//#v66KCF{r>N)rlrmrn__doc__r+r,r-propertylistr9ryrqtuplerrr8r>r<rtrtxs$!K%K tCyXsCxr>rtceZdZdZdZdZdZedee fdZ de de e e ffdZed edee e ffd Zd S) VOCDetectionan`Pascal VOC `_ Detection Dataset. Args: root (str or ``pathlib.Path``): Root directory of the VOC Dataset. year (string, optional): The dataset year, supports years ``"2007"`` to ``"2012"``. image_set (string, optional): Select the image_set to use, ``"train"``, ``"trainval"`` or ``"val"``. If ``year=="2007"``, can also be ``"test"``. 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. (default: alphabetic indexing of VOC's 20 classes). transform (callable, optional): A function/transform that takes in a PIL image and returns a transformed version. E.g, ``transforms.RandomCrop`` target_transform (callable, required): A function/transform that takes in the target and transforms it. transforms (callable, optional): A function/transform that takes input sample and its target as entry and returns a transformed version. Main Annotationsz.xmlrgc|jSrirxrjs r< annotationszVOCDetection.annotationsrzr>r{c,tj|j|d}|t |j|}|j|||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is a dictionary of the XML tree. r}) r rYr[r~ parse_voc_xmlET_parsergetrootr5rs r<rzVOCDetection.__getitem__sjU+,,44U;;##HT-=e-D$E$E$M$M$O$OPP ? &//#v66KCF{r>nodeci}t|}|rtjt}ttj|D]7}|D] \}}|||!8|jdkr |dg|d<|jd|Di}|j r%|j }|s |||j<|S)N annotationobjectcPi|]#\}}|t|dkr|dn|$S)r r)r])r:indvs r< z.VOCDetection.parse_voc_xml..s4"a"a"avsA3A! !"a"a"ar>) r collections defaultdictmaprritemsrTtagtextrF)rvoc_dictchildrendef_dicdcrrrs r<rzVOCDetection.parse_voc_xmls#%::  c&1&=d&C&CG,4h?? + + hhjj++FCCL''****+x<''%,X%6$7!"a"aQXQ^Q^Q`Q`"a"a"abH 9 *9??$$D *%)"r>N)rlrmrnrr+r,r-rrr9rrqrrr staticmethod ET_Elementdictrr8r>r<rrs&KK T#YXsCx J4S>\r>r)rrJpathlibrtypingrrrrxml.etree.ElementTreerrdefusedxml.ElementTreer r ImportErrorPILr utilsr r visionrrKrLrUr*rtrr8r>r<rs 11111111111177777788888888888777777778????????!!!!!!\11GLLi88   \11GLL[)DD   \11GLLi88   \11GLLi88   \11GLLi88   \11GLLi88   X-1GLLi88 K++\6 6 6 6 6 }6 6 6 r)))))h)))X<<<<<8<<<<<s ) 77