j ! d dl Z d dlmZmZ d dlmZ d dlmZmZm Z m
Z
m Z m Z m
Z
d dlZd dlmZ ddlmZmZmZmZ ddlmZ G d d
e ZdS ) N)abspath
expanduser)Path)AnyCallableDictListOptionalTupleUnion)Image ) download_and_extract_archive download_file_from_google_driveextract_archiveverify_str_arg)
VisionDatasetc
e Zd ZdZdZg dZdZ ddeee f d ed
e
e d e
e d e d
df fd
Z
ded
eeef fdZd
efdZd
efdZddZddZd
e fdZddZ xZS ) WIDERFaceuu `WIDERFace `_ Dataset.
Args:
root (str or ``pathlib.Path``): 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.
.. warning::
To download the dataset `gdown `_ is required.
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.ziptrainNFrootsplit transformtarget_transformdownloadreturnc t t j || j || t
|dd | _ |r| | st d g | _ | j dv r| d S |
d S )N)r r! r" r )r valtestzTDataset 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 OC:\Projects\Historical\venv\Lib\site-packages\torchvision/datasets/widerface.pyr) zWIDERFace.__init__: s dD$455]m
$E74LMM
MMOOO$$&& w uvv vNP
:) ) ) 1 1 3 3 3 3 3 , , . . . . . indexc t j | j | d }| j | | }| j dk rdn| j | d }| 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
openr0 r! r r" )r3 r7 imgtargets r5 __getitem__zWIDERFace.__getitem__T s j u-j9:: > %..%%C v--4=3G
3V ,**622FF{r6 c * t | j S )N)lenr0 )r3 s r5 __len__zWIDERFace.__len__j s 4=!!!r6 c R dg} d | j di | j S )NzSplit: {split}
)r, format__dict__)r3 liness r5
extra_reprzWIDERFace.extra_reprm s1 !"&tyy &77 777r6 c | j dk rdnd}t j | j d| }t | 5 }| }d\ }}}d\ }} g }
|D ]} | } |rPt j | j d| j z d| } t t | } d }d
}i|rt | }d }d
}|r]| d z
} | d }
d
|
D }|
| | |k rd }d
}t j
|
}| j | |d d ddf |d d df |d d df |d d df |d d df |d d df |d d df dd d} |
t# d| d d d d S # 1 swxY w Y d S )Nr zwider_face_train_bbx_gt.txtzwider_face_val_bbx_gt.txtwider_face_split)TFF)r r WIDER_imagesFTr c , g | ]}t | S rD )int).0xs r5
z>WIDERFace.parse_train_val_annotations_file.. s ">">">a3q66">">">r6 r )bboxblur
expression illumination occlusionposeinvalid)r9 r: z Error parsing annotation file )r r* r+ r, r r; readlinesrstripr r rO appendtorchtensorr0 cloneclearr/ )r3 filenamefilepathfrG file_name_linenum_boxes_linebox_annotation_line num_boxes box_counterlabelsliner9
line_split line_values
labels_tensors r5 r1 z*WIDERFace.parse_train_val_annotations_fileq s 48J'4I4I00Oj7<< +=xHH
(^^ * TqKKMMEBT ?NN,?%) "I{F %
T %
T{{}}! #T !w||DIx$*7LhX\ ] ]H &z(';'; < ">:">">">KMM+..."i//.3+)-(- V(<(<
,,,4,9!!!QqS&,A,G,G,I,I,9!!!Q$,?,E,E,G,G2?12E2K2K2M2M4A!!!Q$4G4M4M4O4O1>qqq!t1D1J1J1L1L,9!!!Q$,?,E,E,G,G/>-00
uu
tr6 c > | rd S | j D ]Q\ }}}t || j || t j | j | }t | Rt | j d | j | j d d S )Nr r )url
download_rootr~ )
r. rw r r r* r+ r, r r ry )r3 file_idr~ rg rh s r5 r# zWIDERFace.download s " " F )- & & $Wc8 +GTY# N N Nw||DIx88H H % % % % %%a( tG\]^G_
r6 )r NNF)r$ N)__name__
__module__ __qualname____doc__r- rw ry r strr r
r boolr) rO r r r> rA rH r1 r2 r. r#
__classcell__)r4 s @r5 r r s < K I (,/3
/ /CI / / H% /
#8, /
/
/ / / / / /4 sCx ," " " " "8C 8 8 8 8.T .T .T .T` = = = = $
r6 r )r* os.pathr r pathlibr typingr r r r r
r r rc PILr
utilsr r r r visionr r rD r6 r5 r s ' ' ' ' ' ' ' ' D D D D D D D D D D D D D D D D D D q q q q q q q q q q q q ! ! ! ! ! !v
v
v
v
v
v
v
v
v
v
r6