卷积神经网络和3D图像

Question

我有兴趣将CNN应用于3D图像(即医学数据).TensorFlow是否已经包含此功能？

Answer 1

TensorFlow现在支持主分支中的3D卷积和3D池.

您可以将它们与5D张量一起用作输入形状:[batch_size, depth, height, width, channels].

Answer 2

不，当前的实现是针对 2D 图像（如 nn.conv2d 之类的函数）。它们支持多个通道（例如 RGB），并且可以将 3D 图像表示为多通道 2D 图像（每个 z 切片都是一个通道），但这并不总是理想的。此外，为了使用这些方法，您需要大量的图像数据，这在医疗领域通常很难获得。

更新：TensorFlow 和 Theano（随后的 Keras、Lasagne 等）现在都支持如上所述的 3D 操作。值得注意的是，与类似的 2D 操作相比，3D 操作的计算量和内存消耗要大得多。