jc# ddlZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl m Z ddlmZn#e$r ddl mZYnwxYwddlmZddlmZmZdd 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)AnyCallableDictListOptionalTupleUnion)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)Nr3c,g|]}t|S)str).0yrs IC:\Projects\Historical\venv\Lib\site-packages\torchvision/datasets/voc.py z%_VOCBase.__init__..Ps>c>c>c2s2ww>c>c>cii) valid_values)r1trainvalvalr*testr4r+rrrr)rrzHDataset not found or corrupted. You can use download=True to download it ImageSets z.txtc6g|]}|Sr;)strip)r=xs r?r@z%_VOCBase.__init__..js ;;;!''));;;rA JPEGImagescVg|]%}tj|dz&S)z.jpg)ospathjoin)r=rJ image_dirs r?r@z%_VOCBase.__init__..ms-OOOqrw||Iq6z::OOOrAc`g|]*}tj|jz+Sr;)rMrNrOr0)r=rJself target_dirs r?r@z%_VOCBase.__init__..ps2```PQ ZT5J1JKK```rA)super__init__rranger3appendr4DATASET_YEAR_DICTrrrrMrNrOr2risdir RuntimeErrorr.rstripopen readlinesimagesr/targetslen)rRr2r3r4r5r6r7r8valid_image_setskeydataset_year_dictrvoc_root splits_dirsplit_ff file_namesrPrS __class__s` @@r?rUz_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'' ( ( `_ 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.pngrjc|jSrlr_rms r?maskszVOCSegmentation.masks |rAindexctj|j|d}tj|j|}|j|||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is the image segmentation. RGB)r r\r^convertr|r8rRr~imgtargets r? __getitem__zVOCSegmentation.__getitem__sgjU+,,44U;;DJu-.. ? &//#v66KCF{rAN)rorprq__doc__r.r/r0propertyrr<r|rtr rrr;rAr?rwrwxs$!K%K tCyXsCxrArwceZdZdZdZdZdZedee 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.xmlrjc|jSrlr{rms r? annotationszVOCDetection.annotationsr}rAr~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 r\r^r parse_voc_xmlET_parsergetrootr8rs r?rzVOCDetection.__getitem__sjU+,,44U;;##HT-=e-D$E$E$M$M$O$OPP ? &//#v66KCF{rAnodeci}t|}|rtjt}ttj|D]7}|D] \}}|||!8|jdkr |dg|d<|jd|Di}|j r%|j }|s |||j<|S)N annotationobjectcPi|]#\}}|t|dkr|dn|$S)rr)r`)r=indvs r? z.VOCDetection.parse_voc_xml..s4"a"a"avsA3A! !"a"a"arA) list collections defaultdictmaprritemsrWtagtextrI)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 *%)"rAN)rorprqrr.r/r0rrr<rrtr rr staticmethod ET_Elementrrr;rAr?rrs&KK T#YXsCx J4S>\rAr) rrMpathlibrtypingrrrrrr r xml.etree.ElementTreer rdefusedxml.ElementTreer r ImportErrorPILr utilsrrvisionrrNrOrXr-rwrr;rAr?rs DDDDDDDDDDDDDDDDDD77777788888888888777777778????????!!!!!!\11GLLi88   \11GLL[)DD   \11GLLi88   \11GLLi88   \11GLLi88   \11GLLi88   X-1GLLi88 K++\6 6 6 6 6 }6 6 6 r)))))h)))X<<<<<8<<<<<s / ==