pytorch/vision

[RFC] Extend the scope of ImageFolder for segmentation data

Ouverte

#424 ouverte le 16 févr. 2018

 (4 commentaires) (0 réaction) (0 personne assignée)Python (6 858 forks)batch import
enhancementhelp wanted

Métriques du dépôt

Stars
 (15 050 étoiles)
Métriques de merge PR
 (Métriques PR en attente)

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