o
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
eZdS ) N)
namedtuple)Path)AnyCallableOptionalUnion)Image )extract_archiveiterable_to_strverify_str_arg)
VisionDatasetc s e Zd ZdZedg dZg eddddddd d
ed d ddddd d
ed
dddddd d
eddddddd d
eddddddd d
eddddddd deddddddd deddddd 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 d2ed3d4dd(ddd d5ed6d7dd(ddd d8ed9d:dd;dddd<ed=d>dd;ddd d<ed?d@dd;ddddAedBdCdd;ddddDedEdFd dGddddHedIdJd!dGddddKedLdMd$dLddddNedOdPd'dQdd ddRedSdTd+dQdd ddUedVdWd.dXdd ddYedZd[d1dXdd dd\ed]d^d4dXdd dd_ed`daddXdd d dbedcddddXdd d deedfdgd7dXdd ddhedidjd:dXdd ddkedldmd>dXdd ddnedodpdpdXddd dYZ f q r s s sddteee f duedvedwee
e ef dxe e dye e dze e d{dsf fd|d}
Z
d~ed{eeef fddZd{efddZd{efddZded{eeef fddZdvedwed{efddZ 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)nameidZtrain_idcategoryZ category_idZ
has_instancesZignore_in_evalcolorZ 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 r0 r0 Zwall )f r2 Zfence
) r6 z
guard rail ) r8 Zbridge )r, d r; Ztunnel )r, x Z Zpole object)r6 r6 r6 Z polegroup z
traffic light )r' r( z traffic sign ) rE r Z
vegetation Znature)k r# Zterrain ) rJ Zsky )r0 r8 Zperson Zhuman)rE rD < Zrider )r r r car Zvehicle)r r rH truck )r r r0 bus )r rO r; Zcaravan )r r r> trailerrC )r r n train )r P r;
motorcycle )r r r+ bicycle! )w r. r^ z
license platefineinstanceNrootsplitmode target_type transformtarget_transform
transformsreturnc
s@ t |||| |dkrdnd| _tj| jd|| _tj| j| j|| _|| _ || _
g | _ g | _ t
|dd |dkr@d}nd}d } | ||t|} t
|d
|| t|ts]|g| _ d d | j D tj| jrstj| js|d
krtj| jd}
ntj| jd}
| jdkrtj| j| j d} n| jdkrtj| j| j d} tj|
rtj| rt|
| jd t| | jd ntdt| jD ]M} tj| j| }
tj| j| }t|
D ]5}g }| j D ] }d|
dd | | j|}|tj|| q| j tj|
| | j | qqd S )Nrc ZgtFineZgtCoarseZ leftImg8bitrg )rc Zcoarse)rZ testval)rZ train_extrarn zOUnknown value '{}' for argument split if mode is '{}'. Valid values are {{{}}}.rf c S s g | ]}t |d dqS )rh )rd semanticpolygonr )r ).0value rt e/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/torchvision/datasets/cityscapes.py
s
z'Cityscapes.__init__..ro zleftImg8bit_trainextra.zipz leftImg8bit_trainvaltest.zipz_trainvaltest.zipz.zip) from_pathZto_pathzDataset 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__rg ospathjoinre Z
images_dirZ targets_dirrh rf imagestargetsr formatr
isinstancelistisdirisfiler
RuntimeErrorlistdir_get_target_suffixappend)selfre rf rg rh ri rj rk Z valid_modesmsgZ
image_dir_zipZtarget_dir_zipZcityZimg_dir
target_dir file_nameZ target_typest target_name __class__rt ru ry i s`
zCityscapes.__init__indexc C s t | j| d}g }t| jD ]"\}}|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.
RGBrq r r N) r openr} convert enumeraterh
_load_jsonr~ r lentuplerk )r r imager~ ir targetrt rt ru __getitem__ s
zCityscapes.__getitem__c C s
t | jS N)r r} )r rt rt ru __len__ s
zCityscapes.__len__c C s g d}d |jdi | jS )N)zSplit: {split}z Mode: {mode}zType: {target_type}
rt )r| r __dict__)r linesrt rt ru
extra_repr s zCityscapes.extra_reprr{ c C s8 t |}t|}W d |S 1 sw Y |S r )r jsonload)r r{ filedatart rt ru r s
zCityscapes._load_jsonc C s@ |dkr | dS |dkr| dS |dkr| dS | dS )Nrd z_instanceIds.pngrp z
_labelIds.pngr z
_color.pngz_polygons.jsonrt )r rg rh rt rt ru r s
z Cityscapes._get_target_suffix)rZ rc rd NNN)__name__
__module__ __qualname____doc__r r classesr strr r r r ry intr r r r r dictr r
__classcell__rt rt r ru r
s 0
!"#)
F r )r rz collectionsr pathlibr typingr r r r ZPILr utilsr
r r Zvisionr
r rt rt rt ru s