Di r d dl Z d dlZd dlmZmZmZmZ d dlZddlm Z ddl
m Z m Z ddl
mZ G d de Zy) N)AnyCallableOptionalUnion ) make_dataset) download_and_extract_archiveverify_str_arg)
VisionDatasetc
e Zd ZdZ ddeeej f dedee dee de
ddf fd
Z de fd
Z
d e deeef fd Zde
fd
ZddZ xZS )GTSRBa `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 | ||| t |dd | _ t j
| dz | _ | j dz | j dk( rdndz | _ |r| j | j s t d | j dk( r!t t | j d
}nmt | j d z 5 }t j |d
d D cg c]+ }t | j |d z t! |d f- }}d d d | _ || _ || _ y c c}w # 1 sw Y $xY w)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
\/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/torchvision/datasets/gtsrb.pyr zGTSRB.__init__ sJ EUV$UG5FG #LL . 8 '9O:Uh i
MMO!!# \] ] ;;' !"3t':':#; RGd''*==>
( #~~h#X\ ] ,,s:>?S^ATU
" 0
s E;0E +E EEc , t | j S N)lenr- r. s r3 __len__z
GTSRB.__len__> s 4==!! indexc | j | \ }}t j j | j d }| j
| j | }| j | j
| }||fS )NRGB)r- PILImager) convertr r )r. r: pathtargetsamples r3 __getitem__zGTSRB.__getitem__A so }}U+ f %--e4 >> %^^F+F ,**62Fv~ r9 c 6 | j j S r5 )r% is_dirr7 s r3 r& zGTSRB._check_existsN s ""))++r9 c " | j ry d}| j dk( r%t | dt | j d y t | dt | j d t | dt | j d y )
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 r3 r zGTSRB.downloadQ s Z ;;' ! (* 45 !$"3"3 46
)* 78 !$"3"3 46
)* 34 !$"3"3 46
r9 )r NNF)r N)__name__
__module__ __qualname____doc__r r( r" r# r r boolr r, r8 tupler rC r& r
__classcell__)r2 s @r3 r
r
s (,/3
"1C %&"1 "1 H% "1
#8, "1
"1
"1H" " sCx ,t ,r9 r
)r* r" typingr r r r r= folderr utilsr r
visionr r
r9 r3 rY s+
1 1
? ![M [r9