Pht d dl Z d dlmZmZ d dlmZmZmZmZm Z m
Z
m Z d dl Z d dl
mZ ddlmZmZmZmZ ddlmZ G d d e Zy)
N)abspath
expanduser)AnyCallableDictListOptionalTupleUnion)Image ) download_and_extract_archive download_file_from_google_driveextract_archiveverify_str_arg)
VisionDatasetc
e Zd ZdZdZg dZdZ ddededee d ee d
e
d df fd
Z d
e d e
eef fdZd e fdZd efdZddZddZd e
fdZddZ xZS ) WIDERFaceu `WIDERFace `_ Dataset.
Args:
root (string): Root directory where images and annotations are downloaded to.
Expects the following folder structure if download=False:
.. code::
└── widerface
├── wider_face_split ('wider_face_split.zip' if compressed)
├── WIDER_train ('WIDER_train.zip' if compressed)
├── WIDER_val ('WIDER_val.zip' if compressed)
└── WIDER_test ('WIDER_test.zip' if compressed)
split (string): The dataset split to use. One of {``train``, ``val``, ``test``}.
Defaults to ``train``.
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.
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.
widerface))!15hGDLhsx8bLgLcIRD5DhYt5iBxnjNF1M 3fedf70df600953d25982bcd13d91ba2zWIDER_train.zip)!1GUCogbp16PMGa39thoMMeWxp7Rp5oM8Q dfa7d7e790efa35df3788964cf0bbaeaz
WIDER_val.zip)!1HIfDbVEWKmsYKJZm4lchTBDLW5N7dY5T e5d8f4248ed24c334bbd12f49c29dd40zWIDER_test.zip)zLhttp://shuoyang1213.me/WIDERFACE/support/bbx_annotation/wider_face_split.zip 0e3767bcf0e326556d407bf5bff5d27czwider_face_split.zipNrootsplit transformtarget_transformdownloadreturnc d t | t j j || j
|| t
|dd | _ |r| j | j s t d g | _ | j dv r| j y | j y )N)r r r r )trainvaltestzTDataset not found or corrupted. You can use download=True to download and prepare it)r$ r% )super__init__ospathjoin BASE_FOLDERr r r! _check_integrity RuntimeErrorimg_info parse_train_val_annotations_fileparse_test_annotations_file)selfr r r r r! __class__s iC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/datasets/widerface.pyr( zWIDERFace.__init__4 s dD$4$45]m
$E74LM
MMO$$& uv vNP
::) ) 1 1 3 , , . indexc t j | j | d }| j | j | }| j dk( rdn| j | d }| j
| j | }||fS )z
Args:
index (int): Index
Returns:
tuple: (image, target) where target is a dict of annotations for all faces in the image.
target=None for the test split.
img_pathNr& annotations)r openr/ r r r )r2 r6 imgtargets r4 __getitem__zWIDERFace.__getitem__N s~ jj u-j9: >> %..%C v-4==3G
3V ,**62FF{r5 c , t | j S )N)lenr/ )r2 s r4 __len__zWIDERFace.__len__d s 4==!!r5 c ^ dg} dj | j di | j S )NzSplit: {split}
)r+ format__dict__)r2 liness r4
extra_reprzWIDERFace.extra_reprg s- !"&tyy &&7 77r5 c | | j dk( rdnd}t j j | j d| }t | 5 }|j
}d\ }}}d\ }} g }
|D ] } | j } |rRt j j | j d| j z d| } t t | } d }d
}h|rt | }d }d
}z|r-| d z
} | j d }
|
D cg c]
}t | }}|
j | | |k\ sd }d
}t j |
}| j j |d d d
df j |d d df j |d d df j |d d df j |d d df j |d d df j |d d df j dd d
} |
j! t# d| d d d y c c}w # 1 sw Y y xY w)Nr$ zwider_face_train_bbx_gt.txtzwider_face_val_bbx_gt.txtwider_face_split)TFF)r r WIDER_imagesFTr
r )bboxblur
expression illumination occlusionposeinvalid)r8 r9 z Error parsing annotation file )r r) r* r+ r r: readlinesrstripr r intappendtorchtensorr/ cloneclearr. )r2 filenamefilepathfrF file_name_linenum_boxes_linebox_annotation_line num_boxes box_counterlabelsliner8
line_splitx line_values
labels_tensors r4 r0 z*WIDERFace.parse_train_val_annotations_filek s 48JJ'4I0Oj77<< +=xH
(^qKKMEBT ?NN,?%) "I{F {{}! !ww||DIIx$**7LhX\ ]H &z('; MM+."i/.3+)-(- V(<
,,,4,9!QqS&,A,G,G,I,9!Q$,?,E,E,G2?12E2K2K2M4A!Q$4G4M4M4O1>q!t1D1J1J1L,9!Q$,?,E,E,G/>-0 #,
r5 c d | j r t d y | j D ]S \ }}}t || j || t
j j | j | }t | U t | j d | j | j d y )Nz%Files already downloaded and verifiedr r
)url
download_rootrz ) r- printrs r r r) r* r+ r r ru )r2 file_idrz rb rc s r4 r! zWIDERFace.download s " 9 : )- $Wc8 +GTYY# Nww||DIIx8H H % )7 %%%a( tG\G\]^G_
r5 )r$ NNF)r" N)__name__
__module__ __qualname____doc__r, rs ru strr r boolr( r\ r
r r= r@ rG r0 r1 r- r!
__classcell__)r3 s @r4 r r s 4 KI (,/3
// / H% /
#8, /
/
/4 sCx ," "8C 8.T` = $
r5 r )r) os.pathr r typingr r r r r r
r r^ PILr utilsr r r r visionr r rC r5 r4 r s1 ' D D D q q !s
s
r5