Ph#
d dl Z d dlZd dlmZ ddlmZ d dlmZ d dl m Z m
Z
mZmZmZmZ d dlmZ ddlmZmZ d d
d ej. j1 d d
ddddej. j1 dd d ddddej. j1 d d ddddej. j1 d d dd d
d ej. j1 d d dd d d!ej. j1 d d" dd#d$d%ej. j1 d d" dd&Z G d' d(e Z G d) d*e Z G d+ d,e Z y# e
$ r
d dlmZ Y 2w xY w)- 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.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-testc e Zd ZU eed< eed< eed< ddededededee d ee d
ee f fd
Zd e fd
Z
xZ S )_VOCBase _SPLITS_DIR _TARGET_DIR_TARGET_FILE_EXTrootyear image_setdownload transformtarget_transform
transformsc
t | |||| t |dt dd D cg c]
}t | c} | _ g d} |dk( r| j
d t |d| | _ |dk( r|dk( rd n|}
t |
} | d
| _ | d | _
| d | _ | d
} t j j | j | }
|r7t! | j | j | j | j t j j# |
s t% d t j j |
d| j& }t j j ||j) d dz }t+ t j j | 5 }|j- D cg c] }|j/ }}d d d t j j |
d }D cg c]% }t j j ||dz ' c}| _ t j j |
| j2 }|D cg c]/ }t j j ||| j4 z 1 c}| _ t9 | j0 t9 | j6 k( sJ y c c}w c c}w # 1 sw Y xY wc c}w c c}w )Nr0 i i ) valid_values)traintrainvalvalr( testr1 r) r r r r )r r zHDataset not found or corrupted. You can use download=True to download it ImageSets
z.txt
JPEGImagesz.jpg) super__init__r rangestrr0 appendr1 DATASET_YEAR_DICTr r r ospathjoinr/ r isdir RuntimeErrorr, rstripopen readlinesstripimagesr- r. targetslen)selfr/ r0 r1 r2 r3 r4 r5 yrvalid_image_setskeydataset_year_dictr voc_root
splits_dirsplit_ffx
file_names image_dir
target_dir __class__s cC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/datasets/voc.pyr@ z_VOCBase.__init__D sa z96FG"4QVW[]aQb>cQb2s2wQb>cd 7 6> # #F +' ;@PQ !V^ V0Ck -c 2$U+)*5
$U+$Z077<< 84 (499t}}Z^ZbZb cww}}X& ij jWW\\(K9I9IJ
'',,z9+;+;D+AF+JK
"'',,w'
(A-.[[];]!'')]J; ) GGLL<8 DNOJqrww||Iq6z:JO WW\\(D,<,<=
U_`U_PQ ZT5J5J1JKU_` 4;; 3t||#4444E ?d4 < )
( P as/ K" K,K'*K,*K984K>'K,,K6returnc , t | j S N)rP rN rQ s r_ __len__z_VOCBase.__len__t s 4;; )r# r8 FNNN) __name__
__module__ __qualname__rB __annotations__boolr r r@ intrd
__classcell__)r^ s @r_ r+ r+ ? s
(,/3)-.5.5 .5 .5
.5 H%
.5 #8,.5 X&.5` re r+ c R e Zd ZdZdZdZdZedee fd Z
de de e
e
f fdZy )
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.
SegmentationSegmentationClassz.pngr` c | j S rb rO rc s r_ maskszVOCSegmentation.masks ||re indexc t j | j | j d }t j | j | }| j
| j || \ }}||fS )z
Args:
index (int): Index
Returns:
tuple: (image, target) where target is the image segmentation.
RGB)r
rK rN convertrs r5 rQ ru imgtargets r_ __getitem__zVOCSegmentation.__getitem__ se jjU+,44U;DJJu -. ?? & //#v6KCF{re N)rf rg rh __doc__r, r- r. propertyr
rB rs rk r r r| re r_ rn rn x sQ $ !K%K
tCy sCx re rn c v e Zd ZdZdZdZdZedee fd Z
de de e
e
f fdZed edee e
f fd
Zy ) 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.
Main Annotationsz.xmlr` c | j S rb rr rc s r_ annotationszVOCDetection.annotations rt re ru c t j | j | j d }| 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.
rw ) r
rK rN rx
parse_voc_xmlET_parser getrootr5 ry s r_ r| zVOCDetection.__getitem__ sw jjU+,44U;##HT-=-=e-D$E$M$M$OP ?? & //#v6KCF{re nodec i }t | }|rt j t }t t j
| D ]. }|j
D ] \ }}|| j | 0 | j dk( r |d g|d< | j |j
D ci c] \ }}|t | dk( r|d n| c}}i}| j r+| j j }|s||| j < |S c c}}w )N
annotationobjectr r ) list collections defaultdictmapr r itemsrC tagrP textrM )r voc_dictchildrendef_dicdcindvr s r_ r zVOCDetection.parse_voc_xml s #%: &1&=&=d&CG,44h? hhjFCCL''* ) @ xx<'%,X%6$7! QXQ^Q^Q`"aQ`vsA3A! !#BQ`"abH 9999??$D%)" #bs * DN)rf rg rh r} r, r- r. r~ r
rB r rk r r r| staticmethod
ET_Elementr r r re r_ r r s{ & K K
T#Y sCx J 4S> re r ) r rE xml.etree.ElementTreer r visionr defusedxml.ElementTreer r ImportErrortypingr r r r
r r PILr
utilsr r rF rG rD r+ rn r r re r_ r sf 7 !88 > = ? \11GGLLi8
\11GGLL[)D
\11GGLLi8
\11GGLLi8
\11GGLLi8
\11GGLLi8
X-1GGLLi8 K+ \6 } 6 r)h )X<8 <w 8778s E EE