pytorch/vision

[RFC] Extend the scope of ImageFolder for segmentation data

Open

#424 建立於 2018年2月16日

在 GitHub 查看
 (4 留言) (0 反應) (0 負責人)Python (6,858 fork)batch import
enhancementhelp wanted

倉庫指標

Star
 (15,050 star)
PR 合併指標
 (平均合併 12天 8小時) (30 天內合併 14 個 PR)

描述

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.

貢獻者指南