我正在尝试使用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