pytorch/vision

Accounting for class_id in cityscapes dataset

Open

#1 064 ouverte le 28 juin 2019

Voir sur GitHub
 (6 commentaires) (0 réactions) (0 assignés)Python (6 858 forks)batch import
enhancementhelp wantedmodule: datasetsneeds discussion

Métriques du dépôt

Stars
 (15 050 stars)
Métriques de merge PR
 (Merge moyen 12j 8h) (14 PRs mergées en 30 j)

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

Guide contributeur