如何使用mxnet创建3D卷积神经网络?

时间:2018-08-03 18:41:35

标签: python-2.7 deep-learning conv-neural-network mxnet

我有一个包含11 * 11 * 21 3D数据的文件的文件夹,我正在尝试使用3D卷积神经网络进行二进制分类。我只是在胶子教程中找到了2D卷积NN的教程,但不确定要更改哪些内容以适应3D数据。我正在按照本教程进行操作,但是使用自己的数据,我试图将图层更改为3D,但是我仍然需要更改其他内容。 https://gluon.mxnet.io/chapter04_convolutional-neural-networks/cnn-gluon.html

我对convnet和3D convnet还是很陌生,因此任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:1)

对于像您提供的示例教程一样简单的事情,只需将Conv2D更改为Conv3D,将MaxPool2D更改为MaxPool3D,您的数据就必须放在(N, C, D, H, W)版式,而不是(N, C, H, W)版式。