2016-03-20 15 views
1

Ich versuche, ein 3D-Faltungs neuronalen Netzwerk mit Theano, aber ich bin mir nicht ganz sicher über die Verwendung der Funktion theano.tensor.nnet.Conv3d.Beratung mit der Theano-Implementierung von Conv3D

Ich bin es gewohnt, Lasagne zu benutzen, aber da ich zu diesem Zeitpunkt keinen Zugriff auf eine GPU habe, kann ich die lasagne.layers.dnn.Conv3DDNNLayer Funktion nicht benutzen.

Kann mir jemand hinsichtlich der Ein- und Ausgänge mitteilen, was ich tun muss, um die theano-Funktion nutzen zu können? Ich habe Daten in der Form N x 9 x 9 x 9 mit 1 Kanal und habe die Theano-Tensoren initialisiert. Ich habe auch eine Eingangsschicht mit Lasagne wie folgt erstellt:

Jeder Rat ist sehr willkommen! Vielen Dank.

Antwort

1

Derzeit gibt es in Lasagne keine 3D-Conv-Unterstützung außer cuDNN. aber ich habe Gabel von Lasagne, die es implementiert: https://github.com/gyglim/Lasagne (Conv3DLayer)

Seine Ausgänge sind identisch mit denen von Conv3DDNNLayer.

Es gibt auch Diskussionen über das Hinzufügen der Kernlasagne (siehe https://github.com/Lasagne/Lasagne/issues/27), aber es ist noch nicht passiert.

Hoffe, das hilft.

Cheers, Michael

+0

Dank konnte ich Zugriff auf eine CUDA GPU bekommen, so konnten dort die Lasagne Version laufen! Ich denke, es wäre ohne es sowieso viel zu langsam gewesen :) – JB1