kh" ddlZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZddlmZmZdd lmZd d d ej4j7d dddddej4j7dd dddddej4j7d dddddej4j7d dddd dej4j7d ddd d!d"ej4j7d d#dd$d%d&ej4j7d d#dd'ZGd(d)eZGd*d+eZGd,d-eZy#e$r ddl mZY(wxYw).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-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)Nr2ii) valid_values)traintrainvalvalr*testr3r+rrrr)rrzHDataset not found or corrupted. You can use download=True to download it ImageSets z.txt JPEGImagesz.jpg)super__init__rrangestrr2appendr3DATASET_YEAR_DICTrrrospathjoinr1risdir RuntimeErrorr.rstripopen readlinesstripimagesr/r0targetslen)selfr1r2r3r4r5r6r7yrvalid_image_setskeydataset_year_dictrvoc_root splits_dirsplit_ffx file_names image_dir target_dir __class__s T/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/datasets/voc.pyrBz_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)rRrPrSs ra__len__z_VOCBase.__len__ts4;;)r%r:FNNN)__name__ __module__ __qualname__rD__annotations__r rboolrrrBintrf __classcell__)r`s@rar-r-?s  (,/3)-.5CI.5.5 .5  .5 H% .5#8,.5X&.5`  rgr-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.pngrbc|jSrdrQres ramaskszVOCSegmentation.masks ||rgindexctj|j|jd}tj|j|}|j |j ||\}}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is the image segmentation. RGB)r rMrPconvertrur7rSrwimgtargets ra __getitem__zVOCSegmentation.__getitem__sejjU+,44U;DJJu-. ?? &//#v6KCF{rgN)rhrirj__doc__r.r/r0propertyrrDrurmr rr~rgrarprpxsQ$!K%K tCysCxrgrpcveZdZdZdZdZdZedee 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.xmlrbc|jSrdrtres ra annotationszVOCDetection.annotationsrvrgrwctj|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. ry) r rMrPrz parse_voc_xmlET_parsergetrootr7r{s rar~zVOCDetection.__getitem__swjjU+,44U;##HT-=-=e-D$E$M$M$OP ?? &//#v6KCF{rgnodec 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 annotationobjectrr) list collections defaultdictmaprritemsrEtagrRtextrO)rvoc_dictchildrendef_dicdcindvrs rarzVOCDetection.parse_voc_xmls#%: &1&=&=d&CG,44h? + hhj+FCCL''*+ +xx<'%,X%6$7!QXQ^Q^Q`"avsA3A! !#B"abH 9999??$D%)" #bs* DN)rhrirjrr.r/r0rrrDrrmr rr~ staticmethod ET_Elementrrrrgrarrs{&KK T#YsCx J4S>rgr) rrGpathlibrtypingrrrrrr r xml.etree.ElementTreer rdefusedxml.ElementTreer r ImportErrorPILr utilsrrvisionrrHrIrFr-rprrrgrarsl DDD788?!\11GGLLi8  \11GGLL[)D  \11GGLLi8  \11GGLLi8  \11GGLLi8  \11GGLLi8  X-1GGLLi8 K+\6 }6 r)h)X<8<w8778sE EE