在Tensorflow中的非矩形图像上转换

Question

我有上三角形半黑的图像数据集，即主对角线以下的所有像素都是黑色。

Tensorflow中是否有办法将这样的图像提供给conv2d图层并屏蔽或将卷积限制为仅相关的像素？

Answer 1

如果黑色翻译为0，那么您不需要做任何事情。卷积将乘以它所具有的任何权重，因此它不会对结果做出贡献。如果不是，您可以将数据与二进制掩码相乘以使它们为0。

对于所有黑色像素，如果您有任何偏见，您仍会获得任何偏见。 您可以将结果与二进制掩码相乘，将您想要填充的区域乘以0。通过这种方式，您还可以决定放弃具有太多黑色单元格的结果，例如对角线周围。

您也可以编写自己的自定义操作来执行您想要的操作。我建议反对它，因为你只获得最多2的加速（其他操作会降低它）。通过在GPU上运行，您可能获得更高的性能。