Keras 用於帶有二進制圖像的 3D CNN？

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。 您的 model 將過度擬合數據。 改用 2D CNN