o
`)im( ã @ sŒ d dl Z d dlZd dlmZ d dlmZ d dlmZmZm Z m
Z
m Z m Z m
Z
d dlmZ ddlmZmZmZ ddlmZ G d d
„ d
eƒZdS ) é N)Ú
namedtuple)ÚPath)ÚAnyÚCallableÚDictÚListÚOptionalÚTupleÚUnion)ÚImageé )Úextract_archiveÚiterable_to_strÚverify_str_arg)Ú
VisionDatasetc sê e Zd ZdZedg d¢ƒZg eddddddd d
ƒ‘ed d ddddd d
ƒ‘ed
dddddd d
ƒ‘eddddddd d
ƒ‘eddddddd d
ƒ‘eddddddd dƒ‘eddddddd dƒ‘eddddd ddd ƒ‘ed d d dd ddd ƒ‘ed d!ddd dd d"ƒ‘ed#d$ddd dd d%ƒ‘ed&d'dd(dddd)ƒ‘ed*d+dd(dddd,ƒ‘ed-d.dd(dddd/ƒ‘ed0d1dd(ddd d2ƒ‘ed3d4dd(ddd d5ƒ‘ed6d7dd(ddd d8ƒ‘ed9d:dd;dddd<ƒ‘ed=d>dd;ddd d<ƒ‘ed?d@dd;ddddAƒ‘edBdCdd;ddddDƒ‘edEdFd dGddddHƒ‘edIdJd!dGddddKƒ‘edLdMd$dLddddNƒ‘edOdPd'dQdd ddRƒ‘edSdTd+dQdd ddUƒ‘edVdWd.dXdd ddYƒ‘edZd[d1dXdd dd\ƒ‘ed]d^d4dXdd dd_ƒ‘ed`daddXdd d dbƒ‘edcddddXdd d deƒ‘edfdgd7dXdd ddhƒ‘edidjd:dXdd ddkƒ‘edldmd>dXdd ddnƒ‘edodpdpdXddd dYƒ‘Z f q r s s sdŠdteee f duedvedwee
e ef dxe e dye e dze e d{dsf‡ fd|d}„
Z
d~ed{eeef fdd€„Zd{efdd‚„Zd{efdƒd„„Zd…ed{eeef fd†d‡„Zdvedwed{efdˆd‰„Z‡ ZS )‹Ú
Cityscapesaw `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)ÚnameÚidZtrain_idÚcategoryZ category_idZ
has_instancesZignore_in_evalÚcolorZ unlabeledr éÿ ÚvoidFT)r r r z ego vehicler zrectification borderé z
out of roié Zstaticé Zdynamicé )éo éJ r Zgroundé )éQ r r Zroadé Zflat)é€ é@ r" Zsidewalké )éô é# éè Úparkingé )éú éª é z
rail tracké
)éæ é– éŒ Zbuildingé Ú construction)éF r3 r3 Zwallé )éf r5 éœ Zfenceé
)é¾ é™ r9 z
guard railé )é´ é¥ r; Zbridgeé )r/ éd r> Ztunnelé )r/ éx éZ Zpoleé Úobject)r9 r9 r9 Z polegroupé z
traffic lighté )r* r+ é z traffic signé )éÜ rH r Z
vegetationé Znature)ék éŽ r& Zterrainé )é˜ éû rM Zskyé )r3 é‚ r; Zpersoné Zhuman)rH rG é< Zrideré )r r r Úcaré Zvehicle)r r rK Útrucké )r r r3 Úbusé )r rR r> Zcaravané )r r rA ÚtrailerrF )r r én Útrainé )r éP r> Ú
motorcycleé )r r r. Úbicycleé! )éw r1 ra z
license plateéÿÿÿÿÚfineÚinstanceNÚrootÚsplitÚmodeÚ target_typeÚ transformÚtarget_transformÚ
transformsÚreturnc
s@ t ƒ ||||¡ |dkrdnd| _tj | jd|¡| _tj | j| j|¡| _|| _ || _
g | _ g | _ t
|ddƒ |dkr@d}nd}d } | ||t|ƒ¡} t
|d
|| ƒ t|tƒs]|g| _ d d „ | j D ƒ tj | j¡rstj | j¡sÊ|d
kr€tj | jd¡}
ntj | jd¡}
| jdkrštj | j| j› d¡} n| jdkr«tj | j| j› d¡} tj |
¡rÆtj | ¡rÆt|
| jd t| | jd ntdƒ‚t | j¡D ]M} tj | j| ¡}
tj | j| ¡}t |
¡D ]5}g }| j D ] }d |
d¡d | | j|¡¡}| tj ||¡¡ qî| j tj |
|¡¡ | j |¡ qçqÐd S )Nrf ZgtFineZgtCoarseZ leftImg8bitrj )rf Zcoarse)r] ÚtestÚval)r] Ú train_extrarq zOUnknown value '{}' for argument split if mode is '{}'. Valid values are {{{}}}.ri c S s g | ]}t |d dƒ‘qS )rk )rg ÚsemanticÚpolygonr )r )Ú.0Úvalue© rw úk/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/torchvision/datasets/cityscapes.pyÚ
‡ s
ÿÿz'Cityscapes.__init__..rr zleftImg8bit_trainextra.zipz leftImg8bit_trainvaltest.zipz_trainvaltest.zipz.zip)Ú from_pathZto_pathz‹Dataset not found or incomplete. Please make sure all required folders for the specified "split" and "mode" are inside the "root" directoryz{}_{}Z _leftImg8bitr )ÚsuperÚ__init__rj ÚosÚpathÚjoinrh Z
images_dirZ targets_dirrk ri ÚimagesÚtargetsr Úformatr Ú
isinstanceÚlistÚisdirÚisfiler
Ú RuntimeErrorÚlistdirÚ_get_target_suffixÚappend)Úselfrh ri rj rk rl rm rn Z valid_modesÚmsgZ
image_dir_zipZtarget_dir_zipZcityZimg_dirÚ
target_dirÚ file_nameZ target_typesÚtÚ target_name©Ú __class__rw rx r| i s`
þ
ÿ
ÿ÷ýzCityscapes.__init__Úindexc C s¨ t | j| ¡ d¡}g }t| jƒD ]"\}}|dkr%| | j| | ¡}n
t | j| | ¡}| |¡ qt |ƒdkr?t
|ƒn|d }| j durP| ||¡\}}||fS )a.
Args:
index (int): Index
Returns:
tuple: (image, target) where target is a tuple of all target types if target_type is a list with more
than one item. Otherwise, target is a json object if target_type="polygon", else the image segmentation.
ÚRGBrt r r N) r Úopenr€ ÚconvertÚ enumeraterk Ú
_load_jsonr rŠ ÚlenÚtuplern )r‹ r“ Úimager Úir Útargetrw rw rx Ú __getitem__¯ s
zCityscapes.__getitem__c C s
t | jƒS ©N)r™ r€ )r‹ rw rw rx Ú__len__Ê s
zCityscapes.__len__c C s g d¢}d |¡jdi | j¤ŽS )N)zSplit: {split}z Mode: {mode}zType: {target_type}Ú
rw )r r‚ Ú__dict__)r‹ Úlinesrw rw rx Ú
extra_reprÍ s zCityscapes.extra_reprr~ c C s8 t |ƒ}t |¡}W d ƒ |S 1 sw Y |S rŸ )r• ÚjsonÚload)r‹ r~ ÚfileÚdatarw rw rx r˜ Ñ s
ÿþzCityscapes._load_jsonc C s@ |dkr |› dS |dkr|› dS |dkr|› dS |› dS )Nrg z_instanceIds.pngrs z
_labelIds.pngr z
_color.pngz_polygons.jsonrw )r‹ rj rk rw rw rx r‰ Ö s
z Cityscapes._get_target_suffix)r] rf rg NNN)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r Úclassesr
Ústrr r r r r| Úintr r rž r r¤ r r˜ r‰ Ú
__classcell__rw rw r‘ rx r
sÒ 0þÿþýüûúùø ÷
ö õ ô
óòñðïîíìëêéèçæå ä ã â á à!ß"Þ#Ý)ø
þýüûúùø ÷F r )r¥ r} Ú collectionsr Úpathlibr Útypingr r r r r r r
ZPILr Úutilsr
r r Zvisionr r rw rw rw rx Ú s $