Dip" ddlZddlZddlmZddlmZmZmZmZddl m Z  ddl m ZddlmZddlmZmZdd lmZd d d ej.j1d dddddej.j1dd dddddej.j1d dddddej.j1d dddd dej.j1d ddd d!d"ej.j1d d#dd$d%d&ej.j1d d#dd'ZGd(d)eZGd*d+eZGd,d-eZy#e$r ddl m ZY(wxYw).N)Path)AnyCallableOptionalUnion)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< ddeeefdedededee d ee d ee ffd Z d e fd Z xZ S)_VOCBase _SPLITS_DIR _TARGET_DIR_TARGET_FILE_EXTrootyear image_setdownload transformtarget_transform transformsc t|||||t|dtddDcgc] }t |c}|_gd} |dk(r| j dt|d| |_|dk(r|dk(rd n|} t| } | d |_ | d |_ | d |_ | d } tjj|j| } |r7t!|j|j|j|jtjj#| s t%dtjj| d|j&}tjj||j)ddz}t+tjj|5}|j-Dcgc]}|j/}}dddtjj| d}Dcgc]%}tjj||dz'c}|_tjj| |j2}|Dcgc]/}tjj|||j4z1c}|_t9|j0t9|j6k(sJycc}wcc}w#1swYxYwcc}wcc}w)Nr/ii) valid_values)traintrainvalvalr'testr0r(rrrr)rrzHDataset not found or corrupted. You can use download=True to download it ImageSets z.txt JPEGImagesz.jpg)super__init__r rangestrr/appendr0DATASET_YEAR_DICTrrrospathjoinr.r isdir RuntimeErrorr+rstripopen readlinesstripimagesr,r-targetslen)selfr.r/r0r1r2r3r4yrvalid_image_setskeydataset_year_dictrvoc_root splits_dirsplit_ffx file_names image_dir target_dir __class__s Z/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/torchvision/datasets/voc.pyr?z_VOCBase.__init__DsW z96FG"4QVW[]aQb>c2s2w>cd 7 6>  # #F +' ;@PQ!V^ V0Ck-c2$U+)*5 $U+$Z077<< 84  (499t}}Z^ZbZb cww}}X&ij jWW\\(K9I9IJ '',,z9+;+;D+AF+JK "'',,w' ( 'K,,K6returnc,t|jSN)rOrMrPs r^__len__z_VOCBase.__len__ts4;;)r"r7FNNN)__name__ __module__ __qualname__rA__annotations__rrboolrrr?intrc __classcell__)r]s@r^r*r*?s  (,/3)-.5CI.5.5 .5  .5 H% .5#8,.5X&.5`  rdr*cReZdZdZdZdZdZedee fdZ de de e e ffdZy ) 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.pngr_c|jSrarNrbs r^maskszVOCSegmentation.masks ||rdindexctj|j|jd}tj|j|}|j |j ||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is the image segmentation. RGB)r rJrMconvertrrr4rPrtimgtargets r^ __getitem__zVOCSegmentation.__getitem__sejjU+,44U;DJJu-. ?? &//#v6KCF{rdN)rerfrg__doc__r+r,r-propertylistrArrrjtuplerr{rdr^rmrmxsQ$!K%K tCysCxrdrmcveZdZdZdZdZdZedee fdZ de de e e ffdZed edee e ffd Zy ) 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.xmlr_c|jSrarqrbs r^ annotationszVOCDetection.annotationsrsrdrtctj|j|jd}|j t |j |j}|j|j||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is a dictionary of the XML tree. rv) r rJrMrw parse_voc_xmlET_parsergetrootr4rxs r^r{zVOCDetection.__getitem__swjjU+,44U;##HT-=-=e-D$E$M$M$OP ?? &//#v6KCF{rdnodec i}t|}|rtjt}ttj |D].}|j D]\}}||j|0|jdk(r |dg|d<|j|j Dcic]\}}|t|dk(r|dn|c}}i}|jr+|jj}|s|||j<|Scc}}w)N annotationobjectr r) r~ collections defaultdictmaprritemsrBtagrOtextrL)rvoc_dictchildrendef_dicdcindvrs r^rzVOCDetection.parse_voc_xmls#%: &1&=&=d&CG,44h? + hhj+FCCL''*+ +xx<'%,X%6$7!QXQ^Q^Q`"avsA3A! !#B"abH 9999??$D%)" #bs* DN)rerfrgr|r+r,r-r}r~rArrjrrr{ staticmethod ET_Elementdictrrrdr^rrs{&KK T#YsCx J4S>rdr)rrDpathlibrtypingrrrrxml.etree.ElementTreerrdefusedxml.ElementTreer r ImportErrorPILr utilsr r visionrrErFrCr*rmrrrdr^rsi 11788?!\11GGLLi8  \11GGLL[)D  \11GGLLi8  \11GGLLi8  \11GGLLi8  \11GGLLi8  X-1GGLLi8 K+\6 }6 r)h)X<8<w8778sE EE