pytorch/vision

[RFC] Extend the scope of ImageFolder for segmentation data

Aberta

#424 aberto em 16 de fev. de 2018

 (4 comentários) (0 reação) (0 responsável)Python (6.858 forks)batch import
enhancementhelp wanted

Métricas do repositório

Stars
 (15.050 estrelas)
Métricas de merge de PR
 (Métricas PR pendentes)

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