首页 文章

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

提问于
浏览
0

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

Tensorflow中有没有办法将这样的图像赋予conv2d图层并屏蔽或将卷积限制为仅相关的像素?

1 回答

  • 2

    如果黑色转换为0,那么您不需要做任何事情 . 卷积会将0乘以它所具有的任何权重,因此它不会对结果产生影响 . 如果不是,您可以将数据与二进制掩码相乘以使它们为0 .

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

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

相关问题