o V h" @szddlZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl m ZzddlmZWney=ddl mZYnwddlmZddlmZmZdd lmZd d d ejd dddddejdd dddddejd dddddejd dddd dejd ddd d!d"ejd d#dd$d%d&ejd d#dd'ZGd(d)d)eZGd*d+d+eZGd,d-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-testcseZdZUeed<eed<eed<      ddeeefd ed ed ed ee d ee dee ffdd Z de fddZ Z S)_VOCBase _SPLITS_DIR _TARGET_DIR_TARGET_FILE_EXTr$trainFNrootyear image_setdownload transformtarget_transform transformscst||||t|dddtddDd_gd}|dkr&|d t|d |_|dkr7|d kr7d n|} t| } | d _| d _ | d_ | d} t j j| } |rftjjj j dt j | sptdt j | dj} t j | |dd}tt j |}dd|D}Wdn1swYt j | dfdd|D_t j | jfdd|D_tjtjksJdS)Nr1cSsg|]}t|qS)str).0yrr7r7L/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/datasets/voc.py Pz%_VOCBase.__init__..ii) valid_values)r/trainvalvalr)testr2r*rrrr)rrzHDataset not found or corrupted. You can use download=True to download it ImageSets z.txtcSsg|]}|qSr7)stripr9xr7r7r;r<jr= JPEGImagescsg|] }tj|dqS)z.jpg)ospathjoinrE) image_dirr7r;r<mscs g|] }tj|jqSr7)rHrIrJr.rE)self target_dirr7r;r<ps )super__init__rranger1appendr2DATASET_YEAR_DICTrrrrHrIrJr0risdir RuntimeErrorr,rstripopen readlinesimagesr-targetslen)rLr0r1r2r3r4r5r6valid_image_setskeydataset_year_dictrvoc_root splits_dirsplit_ff file_names __class__)rKrLrMr;rODs6      z_VOCBase.__init__returncCs t|jSN)rZrXrLr7r7r;__len__ts z_VOCBase.__len__)r$r/FNNN)__name__ __module__ __qualname__r8__annotations__r rboolrrrOintrh __classcell__r7r7rcr;r+?s4  0r+c@sLeZdZdZdZdZdZedee fddZ de de e e ffd d Zd 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.pngrecC|jSrfrYrgr7r7r;maskszVOCSegmentation.masksindexcCsHt|j|d}t|j|}|jdur |||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is the image segmentation. RGBN)r rVrXconvertrur6rLrwimgtargetr7r7r; __getitem__s  zVOCSegmentation.__getitem__N)rirjrk__doc__r,r-r.propertyrr8rurnrrr}r7r7r7r;rpxsrpc@sjeZdZdZdZdZdZedee fddZ de de e e ffd d Zed edee e ffd d ZdS) 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.xmlrecCrsrfrtrgr7r7r; annotationsrvzVOCDetection.annotationsrwcCsPt|j|d}|t|j|}|jdur$|||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is a dictionary of the XML tree. rxN) r rVrXry parse_voc_xmlET_parsergetrootr6rzr7r7r;r}s  zVOCDetection.__getitem__nodecCsi}t|}|r>tt}ttj|D]}|D] \}}|||qq|jdkr2|dg|d<|jdd|Di}|j rM|j }|sM|||j<|S)N annotationobjectcSs*i|]\}}|t|dkr|dn|qS)r r)rZ)r9indvr7r7r; s*z.VOCDetection.parse_voc_xml..) list collections defaultdictmaprritemsrQtagtextrD)rvoc_dictchildrendef_dicdcrrrr7r7r;rs     zVOCDetection.parse_voc_xmlN)rirjrkr~r,r-r.rrr8rrnrrr} staticmethod ET_Elementrrr7r7r7r;rs r) rrHpathlibrtypingrrrrrrr xml.etree.ElementTreer rdefusedxml.ElementTreer r ImportErrorPILr utilsrrvisionrrIrJrRr+rprr7r7r7r;sh $          .9,