Ei1 S SK r S SKJrJr S SKJr S SKJrJrJ r J
r
S SK r S SK J
r
SSKJrJrJrJr SSKJr " S S
\5 rg) N)abspath
expanduser)Path)AnyCallableOptionalUnion)Image ) download_and_extract_archive download_file_from_google_driveextract_archiveverify_str_arg)
VisionDatasetc
^ \ rS rSrSrSr/ SQrSr SS\\ \
4 S\ S \ \ S
\ \ S \
S S4 U 4S
jjjrS\S \\\4 4S jrS \4S jrS \ 4S jrSS jrSS jrS \
4S jrSS jrSrU =r$ ) WIDERFace u `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.zipNrootsplit transformtarget_transformdownloadreturnc r > [ TU ] [ R R XR
5 X4S9 [
USS5 U l U( a U R 5 U R 5 ( d [ S5 e/ U l U R S; a U R 5 g U R 5 g )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 [/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torchvision/datasets/widerface.pyr' WIDERFace.__init__: s d$4$45
$E74LM
MMO$$&& uv vNP
::) ) 1 1 3 , , . indexc [ R " U R U S 5 nU R b U R U5 nU R S:X a SOU R U S nU R
b U R U5 nX#4$ )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 )r1 r6 imgtargets r3 __getitem__WIDERFace.__getitem__T s| jj u-j9: >> %..%C v-4==3G
3V ,**62F{r5 c , [ U R 5 $ )N)lenr. )r1 s r3 __len__WIDERFace.__len__j s 4==!!r5 c ^ S/nSR U5 R " S0 U R D6$ )NzSplit: {split}
)r* format__dict__)r1 liness r3
extra_reprWIDERFace.extra_reprm s+ !"yy &&7 77r5 c U R S:X a SOSn[ R R U R SU5 n[ U5 nUR
5 nSu pVnSu p/ n
U GH n U R 5 n U( aS [ R R U R SU R - SU 5 n [ [ U 5 5 n S nS
nMn U( a [ U 5 nS nS
nM U( Ga4 U S -
n U R S 5 n
U
Vs/ s H n[ U5 PM nnU
R U5 X: a S nS
n[ R " U
5 nU R R W US S 2S
S24 R 5 US S 2S4 R 5 US S 2S4 R 5 US S 2S4 R 5 US S 2S4 R 5 US S 2S4 R 5 US S 2S4 R 5 S.S.5 S
n U
R! 5 GM GM [# SU 35 e S S S 5 g s snf ! , ( d f g = f)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- )r1 filenamefilepathfrH file_name_linenum_boxes_linebox_annotation_line num_boxes box_counterlabelsliner8
line_splitx line_values
labels_tensors r3 r/ *WIDERFace.parse_train_val_annotations_fileq s 48JJ'4I0Oj77<< +=xH
(^qKKMEBT ?N,?%) "IF {{}! !ww||DIIx$**7LhX\ ]H &z('; MM+."/.3+)-(- V(<
,,,4,9!QqS&,A,G,G,I,9!Q$,?,E,E,G2?12E2K2K2M4A!Q$4G4M4M4O1>q!t1D1J1J1L,9!Q$,?,E,E,G/>-00 !*
r5 c P U R 5 ( a g U R HQ u pn[ XR X25 [ R
R
U R U5 n[ U5 MS [ U R S U R U R S S9 g )Nr r )url
download_rootr )
r, rx r
r r( r) r* r r rz )r1 file_idr re rf s r3 r WIDERFace.download s " " '+nn "G( +GYY Nww||DIIx8H H % '5 %%%a( tG\G\]^G_
r5 )r. r )r# NNF)r! N)__name__
__module__ __qualname____firstlineno____doc__r+ rx rz r strr r r boolr' r_ tupler r= rA rI r/ r0 r, r __static_attributes__
__classcell__)r2 s @r3 r r s < KI (,/3
/CI / / H% /
#8, /
/
/ /4 sCx ," "8C 8.T` = $
r5 r )r( os.pathr r pathlibr typingr r r r ra PILr
utilsr r
r r visionr r rE r5 r3 r s1 ' 1 1 q q !v
v
r5