PyC中定义的层模块在哪里

2024-05-15 04:30:34 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在修改一个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和{}是defined here,但没有提到类型(例如layers.Convolution,等等)。在

我之所以想弄清楚这一点,是因为pycaffe教程中遗漏了其他prototxt参数,我希望在生成prototxt时能够从Python中定义这些参数。其中包括blob_lr和{}。在


Tags: 函数importoutput定义layersgroup教程num
1条回答
网友
1楼 · 发布于 2024-05-15 04:30:34

您可以像这样添加blob_lrphase

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中看到更多示例。在

相关问题 更多 >

    热门问题