pytorch/vision

[RFC] Extend the scope of ImageFolder for segmentation data

Open

#424 aperta il 16 feb 2018

Vedi su GitHub
 (4 commenti) (0 reazioni) (0 assegnatari)Python (6858 fork)batch import
enhancementhelp wanted

Metriche repository

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

Descrizione

I am starting to work on segmentation problem.

ImageFolder is great but seems to be dedicated for classification.

Would it make sense to modify it such that we can load input/target which can be both list of images as in the case of CamVid dataset? It is also true that segmentation can also be organized with a list of images and a list of annotations which does not fit the previous requirement.

So I was wondering what the core devs are thinking about it.

Guida contributor