使用Theano实现Conv3D的建议

我正在尝试使用Theano运行一个3D卷积神经网络,但我对theano.tensor.nnet.Conv3d函数的使用不是很确定。

我习惯使用lasagne,但由于目前无法使用GPU,我无法使用lasagne.layers.dnn.Conv3DDNNLayer函数。

有谁能就输入和输出方面给我一些建议,告诉我如何使用Theano函数?我有N x 9 x 9 x 9格式的数据,通道数为1,并且已经初始化了Theano张量。我还像这样创建了一个输入层,使用了lasagne:

input = lasagne.layers.InputLayer((None, 1, 9, 9, 9), input_var=input_var)

任何建议都非常欢迎!谢谢。


回答:

目前Lasagne中除了cuDNN之外没有支持3D卷积。然而,我有一个实现了它的Lasagne分支:https://github.com/gyglim/Lasagne (Conv3DLayer)

它的输出与Conv3DDNNLayer的输出相同。

关于将其添加到核心Lasagne的讨论也在进行中(参见https://github.com/Lasagne/Lasagne/issues/27),但尚未实现。

希望这对你有帮助。

祝好,Michael

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注