pytorch/vision

[RFC] Extend the scope of ImageFolder for segmentation data

Open

#424 aberto em 16 de fev. de 2018

Ver no GitHub
 (4 comments) (0 reactions) (0 assignees)Python (6.858 forks)batch import
enhancementhelp wanted

Métricas do repositório

Stars
 (15.050 stars)
Métricas de merge de PR
 (Mesclagem média 12d 8h) (14 fundiu PRs em 30d)

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.

Guia do colaborador