+
j1 ^ RI t ^ RIHtHt ^ RIHt ^ RIHtHtH t H
t
^ RI t ^ RI H
t
^RIHtHtHtHt ^RIHt ! R R ]4 tR# )
N)abspath
expanduser)Path)AnyCallableOptionalUnion)Image) download_and_extract_archive download_file_from_google_driveextract_archiveverify_str_arg)
VisionDatasetc a a ] tR t^t oRtRt. ROtRtRV3R lV 3R llltV3R lR lt V3R lR lt
V3R lR
lt V3R lR lt V3R
lR lt
V3R lR ltV3R lR ltRtVtV ;t# ) WIDERFaceu `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.
widerfacec r < V ^8 d QhRS[ S[S[3, RS[RS[S[, RS[S[, RS[RR/# ) rootsplit transformtarget_transformdownloadreturnN)r strr r r bool)format
__classdict__s "_/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchvision/datasets/widerface.py __annotate__WIDERFace.__annotate__: sZ / /CI / / H% /
#8, /
/
/ c < \ SV ` \ P P WP
4 W4R 7 \
VRR4 V n V' d V P 4 V P 4 ' g \ R4 h. V n V P R9 d V P 4 R# V P 4 R# ))r r r r zTDataset not found or corrupted. You can use download=True to download and prepare itN)trainvaltest)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 &&&&&&r r( WIDERFace.__init__: s d$4$45
$E74LM
MMO$$&& uv vNP
::) ) 1 1 3 , , .r" c < < V ^8 d QhRS[ RS[S[S[3, /# )r indexr )inttupler )r r s "r r r! T s# sCx r" c : \ P ! V P V, R, 4 pV P e V P V4 pV P R8X d RMV P V, R, pV P
e V P V4 pW#3# )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 && r __getitem__WIDERFace.__getitem__T s| jj u-j9: >> %..%C v-4==3G
3V ,**62F{r" c < V ^8 d QhRS[ /# r r )r7 )r r s "r r r! j s " " "r" c , \ V P 4 # )N)lenr/ )r2 s &r __len__WIDERFace.__len__j s 4==!!r" c < V ^8 d QhRS[ /# rB )r )r r s "r r r! m s 8 8C 8r" c ^ R .pRP V4 P ! R/ V P B # )zSplit: {split}
)r+ r __dict__)r2 liness & r
extra_reprWIDERFace.extra_reprm s+ !"yy &&7 77r" c < V ^8 d QhRR/# r r NrJ )r r s "r r r! q s .T .T$ .Tr" c V P R 8X d RMRp\ P P V P RV4 p\ V4 ;_uu_ 4 pVP
4 pRRRrvp^ ^ r. p
V EF p V P 4 p V' dX \ P P V P RV P , RV 4 p \ \ V 4 4 p RpRpKs V' d \ V 4 pRpRpK V' Ed> V ^,
p V P R4 p
V
Uu. uF p\ V4 NK ppV
P V4 W8 d RpRp\ P ! V
4 pV P P R X R
R VR, P 4 R VR, P 4 R
VR, P 4 RVR, P 4 RVR, P 4 RVR, P 4 RVR, P 4 //4 ^ p V
P! 4 EK EK \# RV 24 h RRR4 R# u upi + ' g i R# ; i)r$ zwider_face_train_bbx_gt.txtzwider_face_val_bbx_gt.txtwider_face_splitTFWIDER_images r: r; bboxblur
expression illumination occlusionposeinvalidz Error parsing annotation file N)NNN:r N)r] r^ )r] )r] )r] )r] )r] )r r) r* r+ r r< readlinesrstripr r r7 appendtorchtensorr/ cloneclearr. )r2 filenamefilepathfrL file_name_linenum_boxes_linebox_annotation_line num_boxes box_counterlabelsliner:
line_splitx line_values
labels_tensors & r r0 *WIDERFace.parse_train_val_annotations_fileq s 48JJ'4I0Oj77<< +=xH
(^^qKKMEBFu,?N%&{F {{}! !ww||DIIx$**7LhX\ ]H &z('; MM+."/.3+)-(- V(<
,, *H -$*M&,A,G,G,I$*M$,?,E,E,G$0-2E2K2K2M$2M$4G4M4M4O$/ t1D1J1J1L$*M$,?,E,E,G$-}T/B/H/H/J0"
'( ' 0* ')Gz'RSSK ^& #?' ^^s, BI",I" I"$I
8DI" I"" I3 c < V ^8 d QhRR/# rP rJ )r r s "r r r! s = =T =r" c \ P P V P R R4 p\ \ V4 4 p\
V4 ;_uu_ 4 pVP 4 pV Fp pVP 4 p\ P P V P RRV4 p\ \ V4 4 pV P P RV/4 Kr RRR4 R# + ' g i R# ; i)rR z wider_face_test_filelist.txt
WIDER_testrT r: N) r) r* r+ r r r r< rd re r/ rf )r2 rl rm rL rt r: s & r r1 %WIDERFace.parse_test_annotations_file s 77<< +=?]^:h/0
(^^qKKME {{} 77<< <4P":h#78
$$j(%;< ^^^s
BC$$ C5 c < V ^8 d QhRS[ /# rB )r )r r s "r r r! s $ r" c n V P P 4 pVP V P 4 V Fy w r#p\ P
P
V4 w rV\ P
P V P V4 p\ P
P V4 ' d Kx R # R# )FT)
FILE_LISTcopyrf ANNOTATIONS_FILEr) r* splitextr+ r exists)r2 all_files_md5rk fileext
extracted_dirs & r r- WIDERFace._check_integrity s NN'') ../ ) AH((2ID GGLLD 9M77>>-00 !*
r" c < V ^8 d QhRR/# rP rJ )r r s "r r r! s
$
r" c l V P 4 ' d R # V P FQ w rp\ WP W24 \ P
P
V P V4 p\ V4 KS \ V P ^ , V P V P ^, R7 R # )N)url
download_rootr )
r- r r r r) r* r+ r
r r )r2 file_idr rk rl s & r r WIDERFace.download s " " '+nn "G( +GYY Nww||DIIx8H H % '5 %%%a( tG\G\]^G_
r" )r/ r ))!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.zip)r$ NNF)__name__
__module__ __qualname____firstlineno____doc__r, r r r( r? rE rM r0 r1 r- r __static_attributes____classdictcell__
__classcell__)r3 r s @@r r r sx < KI / /4 ," "8 8.T .T` = =
r" r )r) os.pathr r pathlibr typingr r r r rg PILr
utilsr r r
r visionr r rJ r" r r s1 ' 1 1 q q !v
v
r"