pytorch/vision

[RFC] Extend the scope of ImageFolder for segmentation data

開放

#424 建立於 2018年2月16日

 (4 則留言) (0 個反應) (0 位負責人)Python (6,858 個分叉)batch import
enhancementhelp wanted

倉庫指標

星標
 (15,050 顆星)
PR 合併指標
 (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.

貢獻者指南