pytorch/vision

Accounting for class_id in cityscapes dataset

Open

#1.064 aberto em 28 de jun. de 2019

Ver no GitHub
 (6 comments) (0 reactions) (0 assignees)Python (6.858 forks)batch import
enhancementhelp wantedmodule: datasetsneeds discussion

Métricas do repositório

Stars
 (15.050 stars)
Métricas de merge de PR
 (Mesclagem média 12d 8h) (14 fundiu PRs em 30d)

Description

Hi,

In the dataset class for Cityscapes, the classes with their meta-data is mentioned as a namedtuple. However, this information is not used anywhere. Especially for semantic segmenation, the train_id should be used to convert the pixel value (of the label image) to the training label.

Is this planned to be added anytime soon?

Edit: There seems to have been a commit that does something like this and that hasn't been merged. https://github.com/pytorch/vision/pull/1025/commits/39a7a2fb6015792b6c32fd7ee3eeac9774f712a4

Guia do colaborador