Question

我需要增加多声道图像，并希望使用ImageDataGenerator 不幸的是，它仅支持1,3和4通道图像，我需要更多。是否可以直接编辑site-packages / Keras / preprocessing / image.py添加必要数量的频道？

    if x.shape[self.channel_axis] not in {1, 3, 4, XXX}:
        raise ValueError(
            'Expected input to be images (as Numpy array) '
            'following the dimension ordering convention "' + self.dim_ordering + '" '
            '(channels on axis ' + str(self.channel_axis) + '), i.e. expected '
            'either 1, 3 or 4 channels on axis ' + str(self.channel_axis) + '. '
            'However, it was passed an array with shape ' + str(x.shape) +
            ' (' + str(x.shape[self.channel_axis]) + ' channels).')

其中XXX - 是我需要的多个频道。这会破坏什么吗？谢谢！

Answer 1

无论如何我试过了它似乎没有太多的副作用。不过，我没有探究每一个可能的问题。因此，如果你有多通道数据（如卫星图像等），你可以试试这个黑客。有几个地方需要增加扩充代码：）

Hacky在Keras中增强多声道图像的方式

1 个答案: