pytorch/vision
[RFC] Extend the scope of ImageFolder for segmentation data
Offen
#424 geöffnet am 16.02.2018
enhancementhelp wanted
Repository-Metriken
- Stars
- (15.050 Sterne)
- PR-Merge-Metriken
- (PR-Metriken ausstehend)
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.