EiZ( S SK r S SKrS SKJr S SKJr S SKJrJrJ r J
r
S SK J r SSK
JrJrJr SSKJr " S S
\5 rg) N)
namedtuple)Path)AnyCallableOptionalUnion)Image )extract_archiveiterable_to_strverify_str_arg)
VisionDatasetc * ^ \ rS rSrSr\" S/ SQ5 r/ \" SSSSSS S
S 5 P\" S S
SSSS S
S 5 P\" SSSSSS S
S 5 P\" SSSSSS S
S 5 P\" SSSSSS S
S 5 P\" SSSSSS S
S5 P\" SSSSSS S
S5 P\" SSSS S
S S S 5 P\" S S S
S S
S S S 5 P\" S!S"SS S
S S
S#5 P\" S$S%SS S
S S
S&5 P\" S'S(SS)SS S S*5 P\" S+S,SS)SS S S-5 P\" S.SSS)SS S S/5 P\" S0S1SS)SS S
S25 P\" S3S4SS)SS S
S55 P\" S6S7SS)SS S
S85 P\" S9S:SS;SS S S<5 P\" S=S>SS;SS S
S<5 P\" S?S@SS;SS S SA5 P\" SBSCSS;SS S SD5 P\" SESFS SGSS S SH5 P\" SISJS"SGSS S SK5 P\" SLSMS%SLSS S SN5 P\" SOSPS(SQSS
S SR5 P\" SSSTS,SQSS
S SU5 P\" SVSWSSXSS
S SY5 P\" SZS[S1SXSS
S S\5 P\" S]S^S4SXSS
S S_5 P\" S`SaSSXSS
S
Sb5 P\" ScSdSSXSS
S
Se5 P\" SfSgS7SXSS
S Sh5 P\" SiSjS:SXSS
S Sk5 P\" SlSmS>SXSS
S Sn5 P\" SoSpSpSXSS S
SY5 Pr SSr\\ \
4 Ss\ St\ Su\\ \ \ 4 Sv\ \
Sw\ \
Sx\ \
SySq4U 4Sz jjjrS{\Sy\\\4 4S| jrSy\4S} jrSy\ 4S~ jrS\ Sy\\ \4 4S jrSt\ Su\ Sy\ 4S jrSrU =r$ )
Cityscapes
a `Cityscapes `_ Dataset.
Args:
root (str or ``pathlib.Path``): Root directory of dataset where directory ``leftImg8bit``
and ``gtFine`` or ``gtCoarse`` are located.
split (string, optional): The image split to use, ``train``, ``test`` or ``val`` if mode="fine"
otherwise ``train``, ``train_extra`` or ``val``
mode (string, optional): The quality mode to use, ``fine`` or ``coarse``
target_type (string or list, optional): Type of target to use, ``instance``, ``semantic``, ``polygon``
or ``color``. Can also be a list to output a tuple with all specified target types.
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.
transforms (callable, optional): A function/transform that takes input sample and its target as entry
and returns a transformed version.
Examples:
Get semantic segmentation target
.. code-block:: python
dataset = Cityscapes('./data/cityscapes', split='train', mode='fine',
target_type='semantic')
img, smnt = dataset[0]
Get multiple targets
.. code-block:: python
dataset = Cityscapes('./data/cityscapes', split='train', mode='fine',
target_type=['instance', 'color', 'polygon'])
img, (inst, col, poly) = dataset[0]
Validate on the "coarse" set
.. code-block:: python
dataset = Cityscapes('./data/cityscapes', split='val', mode='coarse',
target_type='semantic')
img, smnt = dataset[0]
CityscapesClass)nameidtrain_idcategory category_id
has_instancesignore_in_evalcolor unlabeledr voidFT)r r r z ego vehicler
zrectification border z
out of roi static dynamic )o J r ground )Q r r( road flat) @ r, sidewalk ) # parking ) z
rail track
) building construction)F r? r? wall )f rB fence) rF z
guard rail ) rH bridge )r: d rL tunnel )r: x Z pole object)rF rF rF polegroup z
traffic light )r5 r6 z traffic sign ) rY r
vegetation nature)k r1 terrain ) ra sky )r? rH person human)rY rX < rider )r r r car vehicle)r r r^ truck )r r r? bus )r ri rL caravan )r r rP trailerrW )r r n train )r P rL
motorcycle )r r r9 bicycle! )w r= r{ z
license plateNrootsplitmode target_type transformtarget_transform
transformsreturnc
> [ TU ] XXV5 US:X a SOSU l [ R R U R SU5 U l [ R R U R U R U5 U l X@l X l
/ U l / U l [ USS5 US:X a SnOSnS n U R X#[ U5 5 n [ US
X5 [! U[" 5 ( d U/U l U R V
s/ s H n
[ U
S S 5 PM n
[ R R% U R 5 ( a/ [ R R% U R 5 ( Gdd US
:X a+ [ R R U R S5 n O*[ R R U R S5 n U R S:X a8 [ R R U R U R S35 n OGU R S:X a7 [ R R U R U R S35 n [ R R' U 5 ( aK [ R R' W 5 ( a' [) XR S9 [) XR S9 O [+ S5 e[ R, " U R 5 GHK n
[ R R U R U
5 n[ R R U R U
5 n[ R, " U5 H n/ nU R Hp nSR UR S5 S U R/ U R U5 5 nUR1 [ R R UU5 5 Mr U R R1 [ R R UU5 5 U R R1 U5 M GMN g s sn
f )NfinegtFinegtCoarse leftImg8bitr )r coarse)rw testval)rw train_extrar zOUnknown value '{}' for argument split if mode is '{}'. Valid values are {{{}}}.r r )instancesemanticpolygonr r zleftImg8bit_trainextra.zipz leftImg8bit_trainvaltest.zipz_trainvaltest.zipz.zip) from_pathto_pathzDataset not found or incomplete. Please make sure all required folders for the specified "split" and "mode" are inside the "root" directoryz{}_{} _leftImg8bitr )super__init__r ospathjoinr
images_dir targets_dirr r imagestargetsr
formatr
isinstancelistisdirisfiler RuntimeErrorlistdir_get_target_suffixappend)selfr r r r r r r valid_modesmsgvalue
image_dir_ziptarget_dir_zipcityimg_dir
target_dir file_name target_typest target_name __class__s \/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torchvision/datasets/cityscapes.pyr Cityscapes.__init__i s 9G $HJ '',,tyy-G 77<< 499eD&
tV%78 6>2K9K_jjok&BCug{8+t,, +}D ))
)
5-1] ^)
ww}}T__--RWW]]4CSCS5T5T
% " TYY8T U
" TYY8V W
yyH$!# diiDII;FW9X!Yj(!# diiDII;d9K!Lww~~m,, 1O1O -K .))L"T
JJt/Dggll4??D9G d&6&6=J ZZ 0 ! ))A")..!7:D* *r )r r r r r r r )rw r r NNN)__name__
__module__ __qualname____firstlineno____doc__r r classesr strr r r r r intr r r r r dictr r __static_attributes__
__classcell__)r s @r r r
s) -` !iO
$ QVQtYO$
q#vq%yQ$ .35$PYZ$ afaiP $
!S&!UD)L $ 1c61eT<P
$ !S&!UD+N$ 1fa~N$
Aq&!UE>R$ 1c61eT?S$ b#vq%W$
B>1eULY$ A~q%X$ Q5% Y $ b#~q%_ $ "c>1eT?[!$" "c>1eT>Z#$$ AxE5/R%$& Rh5$X'$( Q!UE>Z)$* AxE5-X+$, b!Xq%W-$. 2q(Aue_U/$0 r2ua~N1$2 "b'1dE=Q3$4 R!T5+N5$6 r2y!T5+N7$8 RAtUJO9$: r2y!T5,O;$< 2sIq$jQ=$> 2sIq$kR?$@ RAtULQA$B b"iD%UC$D 2r9aumTE$F RAudKXG$GR -7(,/3)-D2CI D2 D2 D2
49c>* D2 H%
D2 #8,D2 X&D2
D2 D2L sCx 6 8C 8s tCH~
+s + + + +r r )r r collectionsr pathlibr typingr r r r PILr utilsr r r
visionr r r r r r s1 " 1 1 C C !Q+ Q+r