我用的是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)