caffe下fcn如何训练自己的数据

我用的是voc-fcn8,制作好数据集后 我不知道那个网络层的名字和输出种类怎么改 具体改哪里 改哪些层?求告知?另外那个我gpu溢出 那个bachsize在哪里 怎么改?

http://blog.csdn.net/u013059662/article/details/51854147

layer的名字在定义网络时确定, 例如:

net.conv1_1 = Convolution(...)

这个层在生成配置文件时它的name就叫conv1_1.

fcn segmentation 那篇paper提供的代码里, batch_size为1, 在自定义的数据层里写的. reshape方法里.
voc_layers.py:

    def reshape(self, bottom, top):
        # load image + label image pair
        self.data = self.load_image(self.indices[self.idx])
        self.label = self.load_label(self.indices[self.idx])
        # reshape tops to fit (leading 1 is for batch dimension)
        top[0].reshape(1, *self.data.shape)
        top[1].reshape(1, *self.label.shape)