pytorch/vision

[RFC] Extend the scope of ImageFolder for segmentation data

Open

#424 ouverte le 16 févr. 2018

Voir sur GitHub
 (4 commentaires) (0 réactions) (0 assignés)Python (6 858 forks)batch import
enhancementhelp wanted

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

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.

Guide contributeur