a
«º”hZ( ã @ s€ d dl Z d dlZd dlmZ d dlmZ d dlmZmZm Z m
Z
d dl m Z ddl
mZmZmZ ddlmZ G d d
„ d
eƒZdS ) é N)Ú
namedtuple)ÚPath)ÚAnyÚCallableÚOptionalÚUnion)ÚImageé )Úextract_archiveÚiterable_to_strÚverify_str_arg)Ú
VisionDatasetc + s‚ e Zd ZdZedg d¢ƒZeddddddd 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ƒg#Zd…eee f eeee
e ef e e e e e e dsdtœ‡ fdudv„
Z
eeeef dwœdxdy„Zedzœd{d|„Zedzœd}d~„Zeeeef dœd€d„Zeeed‚œdƒ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é )éô é# éè Zparkingé )éú éª é z
rail tracké
)éæ é– éŒ Zbuildingé Z construction)éF r. r. Zwallé )éf r0 éœ Zfenceé
)é¾ é™ r4 z
guard railé )é´ é¥ r6 Zbridgeé )r+ éd r9 Ztunnelé )r+ éx éZ Zpoleé Úobject)r4 r4 r4 Z polegroupé z
traffic lighté )r& r' é z traffic signé )éÜ rC r Z
vegetationé Znature)ék éŽ r# Zterrainé )é˜ éû rH Zskyé )r. é‚ r6 Zpersoné Zhuman)rC rB é< Zrideré )r r r Zcaré Zvehicle)r r rF Ztrucké )r r r. Zbusé )r rM r9 Zcaravané )r r r< ÚtrailerrA )r r én Útrainé )r éP r9 Z
motorcycleé )r r r* Zbicycleé! )éw r- rX z
license plateéÿÿÿÿÚfineÚinstanceN)ÚrootÚsplitÚmodeÚ target_typeÚ transformÚtarget_transformÚ
transformsÚreturnc
sR t ƒ ||||¡ |dkr dnd| _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¡rètj | j¡s |d
krtj | 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 ] } tj | j| ¡}
tj | j| ¡}t |
¡D ]n}g }| j D ]:}d |
d¡d | | j|¡¡}| tj ||¡¡ qè| j tj |
|¡¡ | j |¡ qÚq¬d S )Nr\ ZgtFineZgtCoarseZ leftImg8bitr` )r\ Zcoarse)rU ÚtestÚval)rU Ú train_extrarg zOUnknown value '{}' for argument split if mode is '{}'. Valid values are {{{}}}.r_ c S s g | ]}t |d dƒ‘qS )ra )r] ÚsemanticÚpolygonr )r )Ú.0Úvalue© rm úM/var/www/auris/lib/python3.9/site-packages/torchvision/datasets/cityscapes.pyÚ
‡ s ÿz'Cityscapes.__init__..rh 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__r` ÚosÚpathÚjoinr^ Z
images_dirZ targets_dirra r_ ÚimagesÚtargetsr Úformatr Ú
isinstanceÚlistÚisdirÚisfiler
Ú RuntimeErrorÚlistdirÚ_get_target_suffixÚappend)Úselfr^ r_ r` ra rb rc rd Z valid_modesÚmsgZ
image_dir_zipZtarget_dir_zipZcityZimg_dirÚ
target_dirÚ file_nameZ target_typesÚtZ target_name©Ú __class__rm rn rr i s\
þ
ÿ
ÿzCityscapes.__init__)Úindexre c C s¨ t | j| ¡ d¡}g }t| jƒD ]D\}}|dkrJ| | j| | ¡}nt | j| | ¡}| |¡ q$t |ƒdkr~t
|ƒn|d }| j dur | ||¡\}}||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.
ZRGBrj r r N) r Úopenrv ÚconvertÚ enumeratera Ú
_load_jsonrw r€ ÚlenÚtuplerd )r rˆ Úimagerw Úir… Útargetrm rm rn Ú __getitem__¯ s
zCityscapes.__getitem__)re c C s
t | jƒS ©N)r rv )r rm rm rn Ú__len__Ê s zCityscapes.__len__c C s g d¢}d |¡jf i | j¤ŽS )N)zSplit: {split}z Mode: {mode}zType: {target_type}Ú
)ru rx Ú__dict__)r Úlinesrm rm rn Ú
extra_reprÍ s zCityscapes.extra_repr)rt re c C s6 t |ƒ}t |¡}W d ƒ n1 s(0 Y |S r“ )r‰ ÚjsonÚload)r rt ÚfileÚdatarm rm rn rŒ Ñ s
(zCityscapes._load_json)r` ra re c C sD |dkr|› dS |dkr$|› dS |dkr6|› dS |› dS d S )Nr] z_instanceIds.pngri z
_labelIds.pngr z
_color.pngz_polygons.jsonrm )r r` ra rm rm rn r Ö s
z Cityscapes._get_target_suffix)rU r\ r] NNN)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r Úclassesr Ústrr rz r r rr ÚintrŽ r r’ r” r˜ ÚdictrŒ r Ú
__classcell__rm rm r† rn r
s| 0þÝ) ø
÷Fr )r™ rs Ú collectionsr Úpathlibr Útypingr r r r ZPILr Úutilsr
r r Zvisionr
r rm rm rm rn Ú s