QЦi v S SK r S SKrS SKJrJrJrJrJr S SKrSSK J
r
SSK J r J
r
SSKJr " S S\5 rg) N)AnyCallableOptionalTupleUnion ) make_dataset) download_and_extract_archiveverify_str_arg)
VisionDatasetc
^ \ rS rSrSr SS\\\R 4 S\S\ \
S\ \
S\ S S4 U 4S
jjjr S \
4S jrS \
S \\\4 4S
jrS \ 4S jrSS jrSrU =r$ )GTSRB a `German Traffic Sign Recognition Benchmark (GTSRB) `_ Dataset.
Args:
root (str or ``pathlib.Path``): Root directory of the dataset.
split (string, optional): The dataset split, supports ``"train"`` (default), or ``"test"``.
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.
Nrootsplit transformtarget_transformdownloadreturnc > [ T U ] XUS9 [ USS5 U l [ R
" U5 S- U l U R S- U R S:X a SOS- U l U( a U R 5 U R 5 ( d [ S 5 eU R S:X a [ [ U R 5 S
S 9nOn[ U R S - 5 n[ R " US
SS9 Vs/ s H, n[ U R US - 5 [! US 5 4PM. nnS S S 5 WU l X0l X@l g s snf ! , ( d f N'= f)N)r r r )traintestgtsrbr r TrainingzFinal_Test/Imagesz;Dataset not found. You can use download=True to download it)z.ppm)
extensionszGT-final_test.csv;T) delimiterskipinitialspaceFilenameClassId)super__init__r _splitpathlibPath _base_folder_target_folderr
_check_exists RuntimeErrorr stropencsv
DictReaderint_samplesr r )
selfr r r r r samplescsv_filerow __class__s
Y/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torchvision/datasets/gtsrb.pyr" GTSRB.__init__ sC EUV$UG5FG #LL . 8 '9O:Uh i
MMO!!## \] ] ;;' !"3t':':#; RGd''*==>( #~~h#X\ ] ] ,,s:>?S^ATU ] ?
" 0 ?>s &E?3E2EE
E#c , [ U R 5 $ N)lenr/ r0 s r5 __len__
GTSRB.__len__> s 4==!! indexc U R U u p#[ R R U5 R S5 nU R
b U R U5 nU R b U R
U5 nXC4$ )NRGB)r/ PILImager+ convertr r )r0 r> pathtargetsamples r5 __getitem__GTSRB.__getitem__A sk }}U+ %--e4 >> %^^F+F ,**62F~ r= c 6 U R R 5 $ r8 )r' is_dirr: s r5 r( GTSRB._check_existsN s ""))++r= c U R 5 ( a g SnU R S:X a" [ U S3[ U R 5 SS9 g [ U S3[ U R 5 SS9 [ U S3[ U R 5 S S9 g )
NzEhttps://sid.erda.dk/public/archives/daaeac0d7ce1152aea9b61d9f1e19370/r zGTSRB-Training_fixed.zip 513f3c79a4c5141765e10e952eaa2478)
download_rootmd5zGTSRB_Final_Test_Images.zip c7e4e6327067d32654124b0fe9e82185zGTSRB_Final_Test_GT.zip fe31e9c9270bbcd7b84b7f21a9d9d9e5)r( r# r
r* r& )r0 base_urls r5 r GTSRB.downloadQ s Z ;;' ! (* 45 !$"3"3 46
)* 78 !$"3"3 46
)* 34 !$"3"3 46
r= )r& r/ r# r' r r )r NNF)r N)__name__
__module__ __qualname____firstlineno____doc__r r* r$ r% r r boolr" r. r; r r rG r( r __static_attributes__
__classcell__)r4 s @r5 r r s (,/3
"1C %&"1 "1 H% "1
#8, "1
"1
"1 "1H" " sCx ,t , r= r )r, r$ typingr r r r r rA folderr utilsr
r visionr r r= r5 ra s+
8 8
? ![M [r=