a PSic$ @srddlZddlZddlmZddlmZzddlmZ Wne yVddlmZ Yn0ddl Z ddl m Z mZmZmZmZmZddlmZddlmZmZd d d ejd d ddddejdd dddddejd dddddejd dddd dejd dddd d!ejd d"dd#d$d%ejd d"dd&ZGd'd(d(eZGd)d*d*eZGd+d,d,eZdS)-N)Element) VisionDataset)parse)AnyCallableDictOptionalTupleList)Image)download_and_extract_archiveverify_str_argzJhttp://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 sdeZdZUeed<eed<eed<deeeeeeeeeedfd d Ze d d d Z Z S)_VOCBase _SPLITS_DIR _TARGET_DIR_TARGET_FILE_EXTrtrainFN)rootyear image_setdownload transformtarget_transform transformscst|||||dkr:|dkr2tdd}ntd|_gd}|dkrZ|dt|d|_|dkr||dkr|dn|} t | } | d_ | d _ | d _ | d } t jj| } |rtj jj j d t j| std t j| dj} t j| |dd}tt j|"}dd|D}Wdn1sR0Yt j| dfdd|D_t j| jfdd|D_tjtjksJdS)NrtestzAccessing the test image set of the year 2007 with year='2007-test' is deprecated since 0.12 and will be removed in 0.14. Please use the combination year='2007' and image_set='test' instead.rzzIn the test image set of the year 2007 only image_set='test' is allowed. For all other image sets use year='2007' instead.)rtrainvalvalrrrrr)rrzHDataset not found or corrupted. You can use download=True to download itZ ImageSets z.txtcSsg|] }|qS)strip.0xr&r&T/var/www/html/django/DPS/env/lib/python3.9/site-packages/torchvision/datasets/voc.py wz%_VOCBase.__init__..Z JPEGImagescsg|]}tj|dqS)z.jpg)ospathjoinr() image_dirr&r+r,zr-cs g|]}tj|jqSr&)r.r/r0rr()self target_dirr&r+r,}r-)super__init__warningswarn ValueErrorrappendrrDATASET_YEAR_DICTrrrr.r/r0rr isdir RuntimeErrorrrstripopen readlinesimagesrtargetslen)r2rrrrrr r!Zvalid_image_setskeyZdataset_year_dictrZvoc_rootZ splits_dirsplit_ff file_names __class__)r1r2r3r+r5EsF      2z_VOCBase.__init__returncCs t|jSN)rBr@r2r&r&r+__len__sz_VOCBase.__len__)rrFNNN) __name__ __module__ __qualname__str__annotations__boolr rr5intrM __classcell__r&r&rGr+r@s& `_ Segmentation Dataset. Args: root (string): 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 an 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. SegmentationZSegmentationClassz.pngrIcCs|jSrKrArLr&r&r+masksszVOCSegmentation.masksindexrJcCsHt|j|d}t|j|}|jdur@|||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is the image segmentation. RGBN)r r>r@convertrYr!r2r[imgtargetr&r&r+ __getitem__s  zVOCSegmentation.__getitem__N)rNrOrP__doc__rrrpropertyr rQrYrTr rrar&r&r&r+rVsrVc@sfeZdZdZdZdZdZeee dddZ e e e e fdd d Zeeee e fd d d ZdS) VOCDetectiona_`Pascal VOC `_ Detection Dataset. Args: root (string): 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 an 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. ZMain Annotationsz.xmlrIcCs|jSrKrXrLr&r&r+ annotationsszVOCDetection.annotationsrZcCsPt|j|d}|t|j|}|jdurH|||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is a dictionary of the XML tree. r\N) r r>r@r] parse_voc_xmlET_parserfgetrootr!r^r&r&r+ras  zVOCDetection.__getitem__)noderJcCsi}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)rr)rB)r)indvr&r&r+ r-z.VOCDetection.parse_voc_xml..) list collections defaultdictmaprdrgitemsr9tagtextr')rjZvoc_dictchildrenZdef_dicdcrmrnrvr&r&r+rgs    zVOCDetection.parse_voc_xmlN)rNrOrPrbrrrrcr rQrfrTr rra staticmethod ET_Elementrrgr&r&r&r+rdsrd)rqr.xml.etree.ElementTreerrzvisionrZdefusedxml.ElementTreerrh ImportErrorr6typingrrrr r r PILr utilsr rr/r0r:rrVrdr&r&r&r+sd           .E,