簡體   English   中英

使用 Conv3D 的 theano 實現的建議

[英]Advice using the theano implementation of Conv3D

我正在嘗試使用 theano 運行 3D 卷積神經網絡,但是我不完全確定函數theano.tensor.nnet.Conv3d的用法。

我習慣使用lasagne.layers.dnn.Conv3DDNNLayer ,但是由於此時無法訪問 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)

非常歡迎任何建議! 謝謝。

目前除了 cuDNN 之外,Lasagne 中沒有 3D conv 支持。 但是我有實現它的千層面的分支: https : //github.com/gyglim/Lasagne (Conv3DLayer)

它的輸出與 Conv3DDNNLayer 的輸出相同。

還有關於將它添加到核心千層面的討論(參見https://github.com/Lasagne/Lasagne/issues/27 ),但還沒有發生。

希望這可以幫助。

干杯,邁克爾

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM