Conv2D 到 Conv3D

Conv2D to Conv3D

我有 3D 医学图像,想知道我是否有使用 Conv2D 的 CNN,我可以将 Conv2D 更改为 Conv3D 吗?如果不是,我需要更改什么?

是的,你可以,但有一些事情需要改变。

您的内核现在需要是 3D 的,因此参数 kernel_size 必须是一个 3 整数元组。 strides 也一样。请注意,如果您要修改的 CNN 设计用于在彩色图像上进行训练,则它可能已经是 3D 的(例如 60、60、3)。唯一的区别是,您希望神经网络不仅要检测 3 个独立的 60x60 windows 中的特征,还要检测三个 windows 中的特征。换句话说,不是 2D 的 3 倍,而是 3D。

tl;dr 可以,只需更改 kernel_sizestrideskeras.layers.Conv3D 的默认值无论如何都会相应调整。