Ei r S SK r S SKrS SKJrJrJrJr S SKrSSKJ r SSK
J r J r SSK
Jr " S S\5 rg) N)AnyCallableOptionalUnion ) 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
W/home/mande/repo/quber/.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. r/ s r4 __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 )r/ r= pathtargetsamples r4 __getitem__GTSRB.__getitem__A sk }}U+ %--e4 >> %^^F+F ,**62F~ r< c 6 U R R 5 $ r7 )r& is_dirr9 s r4 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% )r/ base_urls r4 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: tupler rF r' r __static_attributes__
__classcell__)r3 s @r4 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@ folderr utilsr r
visionr r
r< r4 ra s+
1 1
? ![M [r<