pytorch/vision

Accounting for class_id in cityscapes dataset

Open

#1064 aperta il 28 giu 2019

Vedi su GitHub
 (6 commenti) (0 reazioni) (0 assegnatari)Python (6858 fork)batch import
enhancementhelp wantedmodule: datasetsneeds discussion

Metriche repository

Star
 (15.050 star)
Metriche merge PR
 (Merge medio 12g 8h) (14 PR mergiate in 30 g)

Descrizione

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

Guida contributor