o )i8" @sjddlZddlZddlmZddlmZzddlmZ Wne y+ddlmZ Ynwddl m Z m Z mZmZmZmZddlmZddlmZmZd 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)AnyCallableDictListOptionalTuple)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-testcs|eZdZUeed<eed<eed<      dded ed ed ed eed eedeeffdd Zde fddZ Z S)_VOCBase _SPLITS_DIR _TARGET_DIR_TARGET_FILE_EXTrtrainFNrootyear 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)NrcSsg|]}t|qS)str).0yrr"r"g/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/datasets/voc.py Pz%_VOCBase.__init__..ii)Z valid_values)rZtrainvalvalrtestrrrrrr)rrzHDataset not found or corrupted. You can use download=True to download itZ ImageSets z.txtcSsg|]}|qSr")stripr$xr"r"r&r'jr(Z JPEGImagescsg|] }tj|dqS)z.jpg)ospathjoinr-) image_dirr"r&r'mscs g|] }tj|jqSr")r/r0r1rr-)self target_dirr"r&r'ps )super__init__rrangerappendrDATASET_YEAR_DICTrrrr/r0r1rr isdir RuntimeErrorrrstripopen readlinesimagesrtargetslen)r3rrrrrr r!Zvalid_image_setskeyZdataset_year_dictrZvoc_rootZ splits_dirZsplit_ffZ file_names __class__)r2r3r4r&r6Ds6      z_VOCBase.__init__returncCs t|jSN)rAr?r3r"r"r&__len__ts z_VOCBase.__len__)rrFNNN) __name__ __module__ __qualname__r#__annotations__boolr rr6intrI __classcell__r"r"rDr&r?s4 0rc@sLeZdZdZdZdZdZedee fddZ de de e e ffd d Zd S) VOCSegmentationa"`Pascal VOC `_ 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. Z SegmentationZSegmentationClassz.pngrFcC|jSrGr@rHr"r"r&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 r=r?convertrTr!r3rVZimgtargetr"r"r& __getitem__s  zVOCSegmentation.__getitem__N)rJrKrL__doc__rrrpropertyr r#rTrOr rr[r"r"r"r&rQxsrQc@sjeZdZdZdZdZdZedee fddZ de de e e ffd d Zed edee e ffd 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. ZMainZ Annotationsz.xmlrFcCrRrGrSrHr"r"r& annotationsrUzVOCDetection.annotationsrVcCsPt|j|d}|t|j|}|jdur$|||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is a dictionary of the XML tree. rWN) r r=r?rX parse_voc_xmlET_parser_Zgetrootr!rYr"r"r&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)rr)rA)r$indvr"r"r& s*z.VOCDetection.parse_voc_xml..) list collections defaultdictmapr^r`itemsr8tagtextr,)rbZvoc_dictchildrenZdef_dicZdcrerfrnr"r"r&r`s     zVOCDetection.parse_voc_xmlN)rJrKrLr\rrrr]r r#r_rOr rr[ staticmethod ET_Elementrr`r"r"r"r&r^s r^)rir/Zxml.etree.ElementTreerrqZvisionrZdefusedxml.ElementTreerra ImportErrortypingrrrr r r ZPILr utilsr rr0r1r9rrQr^r"r"r"r&sf           .9,