U h" @s|ddlZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl m ZzddlmZWn ek r|ddl mZYnXddlmZddlmZmZdd 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.tarZ 6cd6e144f989b92b3379bac3b3de84fdZ VOCdevkitZVOC2012)urlfilenamemd5base_dirzJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2011/VOCtrainval_25-May-2011.tarzVOCtrainval_25-May-2011.tarZ 6c3384ef61512963050cb5d687e5bf1eZTrainValZVOC2011zJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2010/VOCtrainval_03-May-2010.tarzVOCtrainval_03-May-2010.tarZ da459979d0c395079b5c75ee67908abbZVOC2010zJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2009/VOCtrainval_11-May-2009.tarzVOCtrainval_11-May-2009.tarZ a3e00b113cfcfebf17e343f59da3caa1ZVOC2009zJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2008/VOCtrainval_14-Jul-2008.tarZ 2629fa636546599198acfcfbfcf1904aZVOC2008zJhttp://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tarzVOCtrainval_06-Nov-2007.tarZ c52e279531787c972589f7e41ab4ae64ZVOC2007zFhttp://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tarzVOCtest_06-Nov-2007.tarZ b6e924de25625d8de591ea690078ad9f)2012Z2011Z2010Z2009Z20082007 2007-testc sleZdZUeed<eed<eed<deeefeeeee ee ee dfd d Z e d d d Z Z S)_VOCBase _SPLITS_DIR _TARGET_DIR_TARGET_FILE_EXTrtrainFN)rootyear image_setdownload transformtarget_transform transformsc st||||t|dddtddDd_ddd g}|d krN|d t|d |_|d krp|d krpd n|} t| } | d_| d_ | d_ | d} t j j| } |rtjjj j dt j | stdt j | dj} t j | |dd}tt j |}dd|D}W5QRXt j | dfdd|D_t j | jfdd|D_tjtjkstdS)NrcSsg|] }t|qS)str).0yrr$r$J/var/www/html/venv/lib/python3.8/site-packages/torchvision/datasets/voc.py Psz%_VOCBase.__init__..ii)Z valid_valuesrZtrainvalvalrtestrrrrrr)rrzHDataset not found or corrupted. You can use download=True to download itZ ImageSets z.txtcSsg|] }|qSr$)stripr&xr$r$r(r)jsZ JPEGImagescsg|]}tj|dqS)z.jpg)ospathjoinr.) image_dirr$r(r)mscs g|]}tj|jqSr$)r0r1r2rr.)self target_dirr$r(r)ps)super__init__rrangerappendrDATASET_YEAR_DICTrrrr0r1r2rrisdir RuntimeErrorrrstripopen readlinesimagesrtargetslenAssertionError)r4rrrr r!r"r#Zvalid_image_setskeyZdataset_year_dictrZvoc_rootZ splits_dirZsplit_ffZ file_names __class__)r3r4r5r(r7Ds4        z_VOCBase.__init__returncCs t|jSN)rBr@r4r$r$r(__len__tsz_VOCBase.__len__)rrFNNN)__name__ __module__ __qualname__r%__annotations__r rboolrrr7intrL __classcell__r$r$rFr(r?s&  0rc@sJeZdZdZdZdZdZeee dddZ e e e e fdd 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. Z SegmentationZSegmentationClassz.pngrHcCs|jSrJrArKr$r$r(masksszVOCSegmentation.masksindexrIcCsHt|j|d}t|j|}|jdk r@|||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is the image segmentation. RGBN)r r>r@convertrVr#r4rXZimgtargetr$r$r( __getitem__s  zVOCSegmentation.__getitem__N)rMrNrO__doc__rrrpropertyrr%rVrRrrr]r$r$r$r(rTxsrTc@sfeZdZdZdZdZdZeee dddZ e e e e fdd d Zeeee e fd d 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. ZMainZ Annotationsz.xmlrHcCs|jSrJrUrKr$r$r( annotationsszVOCDetection.annotationsrWcCsPt|j|d}|t|j|}|jdk rH|||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is a dictionary of the XML tree. rYN) r r>r@rZ parse_voc_xmlET_parseraZgetrootr#r[r$r$r(r]s  zVOCDetection.__getitem__)noderIcCsi}t|}|r|tt}ttj|D]$}|D]\}}|||q2q&|jdkrd|dg|d<|jdd|Di}|j r|j }|s|||j<|S)N annotationobjectcSs*i|]"\}}|t|dkr"|dn|qS)r r)rB)r&indvr$r$r( sz.VOCDetection.parse_voc_xml..) list collections defaultdictmapr`rbitemsr9tagtextr-)rdZvoc_dictchildrenZdef_dicZdcrgrhrpr$r$r(rbs    zVOCDetection.parse_voc_xmlN)rMrNrOr^rrrr_rr%rarRrrr] staticmethod ET_Elementrrbr$r$r$r(r`sr`) rkr0pathlibrtypingrrrrrrr Zxml.etree.ElementTreer rsZdefusedxml.ElementTreer rc ImportErrorZPILr utilsrrZvisionrr1r2r:rrTr`r$r$r$r(sd $         .9,