chainer / chainercv

ChainerCV: a Library for Deep Learning in Computer Vision

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DirectoryParsingLabelDataset fails to read images with an alpha channel

yoshnary opened this issue · comments

When I tried the ImageNet training example, it failed due to DirectoryParsingLabelDataset. ImageNet 2012 dataset has an image with an alpha channel, and DirectoryParsingLabelDataset cannot read it. I dealt with it by adding alpha='ignore' to the following function call.

return read_image(self.img_paths[i], color=self.color)

Could you fix it in a proper way?