带有二进制图像的 3D CNN 的 Keras？

Question

我正在尝试将 3D CNN 用于 3D 二值图像（黑色和白色，所有值为 0 或 1）。因为它是黑白的，所以颜色通道没有“第 4”维，数据形状为 (256, 256, 124)。我将如何在 keras 中构建 CNN？我会继续使用 Conv3d 还是改用 Conv2D？

Answer 1

首先，2D CNN 和 3D CNN 都可以工作。您只需要为您的数据获得正确的形状，但您需要了解 2D CNN 和 3D CNN 之间的区别：

• 在 2D CNN 中，您的过滤器仅在图像的两个维度中移动，这意味着您的过滤器的第三个维度必须与图像相同 124（在 keras 中，您不必手动设置它，框架将其设置为你）。
• 在 3D CNN 中，您允许过滤器在图像的第三维中移动，因此您的过滤器第三维可以小于或等于 124

最后，二值图像太简单了，无法使用 3D Conv。您的模型将过度拟合数据。改用 2D CNN