pytorch/vision

[RFC] Extend the scope of ImageFolder for segmentation data

Open

#424 geöffnet am 16. Feb. 2018

Auf GitHub ansehen
 (4 Kommentare) (0 Reaktionen) (0 zugewiesene Personen)Python (6.858 Forks)batch import
enhancementhelp wanted

Repository-Metriken

Stars
 (15.050 Stars)
PR-Merge-Metriken
 (Durchschn. Merge 12T 8h) (14 gemergte PRs in 30 T)

Beschreibung

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.

Contributor Guide