我有一个包含11 * 11 * 21 3D数据的文件的文件夹,我正在尝试使用3D卷积神经网络进行二进制分类。我只是在胶子教程中找到了2D卷积NN的教程,但不确定要更改哪些内容以适应3D数据。我正在按照本教程进行操作,但是使用自己的数据,我试图将图层更改为3D,但是我仍然需要更改其他内容。 https://gluon.mxnet.io/chapter04_convolutional-neural-networks/cnn-gluon.html
我对convnet和3D convnet还是很陌生,因此任何帮助将不胜感激。
答案 0 :(得分:1)
对于像您提供的示例教程一样简单的事情,只需将Conv2D
更改为Conv3D
,将MaxPool2D
更改为MaxPool3D
,您的数据就必须放在(N, C, D, H, W)
版式,而不是(N, C, H, W)
版式。