PyCaffe中定義的圖層模塊在哪裏

Question

我正在修改一個Caffe tutorial來實現一個神經網絡，但我正在努力確定某些pycaffe模塊的位置，以便查看某些函數定義。

例如，本教程中提到：

import caffe 
from caffe import layers a L, params as P 
.... 
L.Convolution(bottom, kernel_size=ks, stride=stride, num_output=nout, pad=pad, group=group) 
L.InnerProduct(bottom, num_output=nout) 
L.ReLU(fc, in_place=True) 
... 

我在哪裏可以找到這些函數定義和我在哪裏可以看到其他類型的層是預先定義的？我看到layers和params是defined here，但沒有提及類型（例如layers.Convolution等）。

我想知道的原因是因爲在pycaffe教程中還有其他prototxt參數，我希望能夠在生成原型文件時從Python中定義這些參數。這些包括，blob_lr和include{phase: TRAIN}。

Answer 1

您可以添加blob_lr和phase這樣的：

import caffe 
from caffe import layers a L, params as P 

ns = caffe.NetSpec() 
ns.conv = L.Convolution(bottom, convolution_param={'kernel_size':ks, 
                'stride':stride, 
                'num_output':nout, 
                'pad':pad, 
                'group':group}, 
           param=[{'lr_mult':1, 'decay_mult':1}, 
             {'lr_mult':2, 'decay_mult':0}], 
           include={'phase': caffe.TRAIN}) 

你可以看到在this answer一些例子。