使用Conv3D的theano实现的建议

时间:2016-03-20 17:41:34

标签: python machine-learning computer-vision theano convolution

我正在尝试使用theano运行3D卷积神经网络,但我并不完全确定函数theano.tensor.nnet.Conv3d的用法。

我习惯使用烤宽面条,但由于此时无法访问GPU,我无法使用lasagne.layers.dnn.Conv3DDNNLayer功能。

是否有人能够在输入和输出方面告诉我我需要做些什么才能使用theano功能?我的数据形式为N x 9 x 9 x 9,带有1个通道,并初始化了theano张量。我还创建了一个带宽容的输入层,如下所示:

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

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

1 个答案:

答案 0 :(得分:1)

除了cuDNN之外,目前在Lasagne中没有3D conv支持。 但是我有实现它的烤宽面条叉: https://github.com/gyglim/Lasagne(Conv3DLayer)

其输出与Conv3DDNNLayer的输出相同。

还有关于将其添加到核心千层面的讨论(见https://github.com/Lasagne/Lasagne/issues/27),但尚未发生。

希望这有帮助。

干杯, 迈克尔